From 931af963b116636f130a7e6edee1005139f884f0 Mon Sep 17 00:00:00 2001
From: zhouweiyi <>
Date: Fri, 16 May 2025 13:50:12 +0800
Subject: [PATCH] =?UTF-8?q?feat:pdf=E6=96=87=E4=BB=B6=E8=A7=A3=E6=9E=90?=
 =?UTF-8?q?=E5=9B=BE=E7=89=87=E5=92=8C=E5=88=86=E6=9E=90=E5=9B=BE=E7=89=87?=
 =?UTF-8?q?=EF=BC=8C=E4=B8=8A=E4=BC=A0=E5=90=91=E9=87=8F=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E9=83=BD=E4=BF=AE=E6=94=B9=E6=88=90=20=E6=88=90?=
 =?UTF-8?q?=E5=BC=82=E6=AD=A5=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/knowledge/DealFileService.java    |  5 ----
 .../knowledge/KnowledgeInfoServiceImpl.java   | 29 ++++++++++++-------
 2 files changed, 19 insertions(+), 15 deletions(-)

diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/DealFileService.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/DealFileService.java
index 882fce3a..2951c20f 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/DealFileService.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/DealFileService.java
@@ -302,11 +302,6 @@ public class DealFileService {
           .eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_20)
           .eq(KnowledgeAttachPic::getId, picItem.getId()));
       throw new RuntimeException(e);
-    } finally {
-      //将图片分析失败的数据 重新设置状态
-      picMapper.update(new LambdaUpdateWrapper<KnowledgeAttachPic>()
-          .set(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_10)
-          .eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_40));
     }
   }
 
diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java
index fb237b2d..c34b4af4 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java
@@ -317,7 +317,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
 
   }
 
-
   /**
    * 检查用户是否有删除知识库权限
    *
@@ -331,8 +330,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
       }
     }
   }
-
-
   /**
    * 第一步 定时 拆解PDF文件中的图片
    */
@@ -352,8 +349,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
       }
     }
   }
-
-
   /**
    * 第二步 定时 解析图片内容
    */
@@ -363,6 +358,7 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
     List<KnowledgeAttachPic> knowledgeAttachPics = picMapper.selectList(
         new LambdaQueryWrapper<KnowledgeAttachPic>()
             .eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_10)
+            .last("LIMIT 20")
     );
     if (ObjectUtil.isNotEmpty(knowledgeAttachPics)) {
       for (KnowledgeAttachPic picItem : knowledgeAttachPics) {
@@ -370,8 +366,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
       }
     }
   }
-
-
   /**
    * 第三步 定时 处理 附件上传后上传向量数据库
    */
@@ -391,7 +385,22 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
       }
     }
   }
-
-
-
+  /**
+   * 第四步 定时 处理 失败数据
+   */
+  @Scheduled(fixedDelay = 30 * 60 * 1000)
+  public void dealKnowledge40Status() throws Exception {
+      //拆解PDF失败 重新设置状态
+      attachMapper.update(new LambdaUpdateWrapper<KnowledgeAttach>()
+          .set(KnowledgeAttach::getPicStatus, DealStatus.STATUS_10)
+          .eq(KnowledgeAttach::getPicStatus, DealStatus.STATUS_40));
+      //将图片分析失败的数据 重新设置状态
+      picMapper.update(new LambdaUpdateWrapper<KnowledgeAttachPic>()
+          .set(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_10)
+          .eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_40));
+      //上传向量库失败 重新设置状态
+      attachMapper.update(new LambdaUpdateWrapper<KnowledgeAttach>()
+          .set(KnowledgeAttach::getVectorStatus, DealStatus.STATUS_10)
+          .eq(KnowledgeAttach::getVectorStatus, DealStatus.STATUS_40));
+  }
 }