xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-17 13:53:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:pdf文件解析图片和分析图片，上传向量数据库都修改成 成异步处理

Browse Source
This commit is contained in:
zhouweiyi
2025-05-16 13:50:12 +08:00
parent f5fa7d5da8
commit 931af963b1
2 changed files with 19 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/DealFileService.java
View File

@@ -302,11 +302,6 @@ public class DealFileService {
.eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_20) .eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_20)
.eq(KnowledgeAttachPic::getId, picItem.getId())); .eq(KnowledgeAttachPic::getId, picItem.getId()));
throw new RuntimeException(e); throw new RuntimeException(e);
} finally {
//将图片分析失败的数据 重新设置状态
picMapper.update(new LambdaUpdateWrapper<KnowledgeAttachPic>()
.set(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_10)
.eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_40));
} }
} }

29
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java
View File

@@ -317,7 +317,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
} }
/** /**
* 检查用户是否有删除知识库权限 * 检查用户是否有删除知识库权限
* *
@@ -331,8 +330,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
} }
} }
} }
/** /**
* 第一步 定时 拆解PDF文件中的图片 * 第一步 定时 拆解PDF文件中的图片
*/ */
@@ -352,8 +349,6 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
} }
} }
} }
/** /**
* 第二步 定时 解析图片内容 * 第二步 定时 解析图片内容
*/ */
@@ -363,6 +358,7 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
List<KnowledgeAttachPic> knowledgeAttachPics = picMapper.selectList( List<KnowledgeAttachPic> knowledgeAttachPics = picMapper.selectList(
new LambdaQueryWrapper<KnowledgeAttachPic>() new LambdaQueryWrapper<KnowledgeAttachPic>()
.eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_10) .eq(KnowledgeAttachPic::getPicAnysStatus, DealStatus.STATUS_10)
.last("LIMIT 20")
); );
if (ObjectUtil.isNotEmpty(knowledgeAttachPics)) { if (ObjectUtil.isNotEmpty(knowledgeAttachPics)) {
for (KnowledgeAttachPic picItem : 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));
}
} }