wangys/urbanLifeline
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

知识库文件数量修正

Browse Source
This commit is contained in:
wangys
2025-12-30 20:54:48 +08:00
parent 4e04e4599e
commit 5ce012000b
3 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
urbanLifelineServ/ai/src/main/java/org/xyzh/ai/mapper/TbKnowledgeMapper.java
View File

@@ -32,6 +32,8 @@ public interface TbKnowledgeMapper {
*/ */
int deleteKnowledge(TbKnowledge knowledge); int deleteKnowledge(TbKnowledge knowledge);
int updateKnowledgeFileCount(String knowledgeId, int num);
/** /**
* 根据ID查询知识库 * 根据ID查询知识库
*/ */

4
urbanLifelineServ/ai/src/main/java/org/xyzh/ai/service/impl/KnowledgeServiceImpl.java
View File

@@ -533,6 +533,7 @@ public class KnowledgeServiceImpl implements KnowledgeService {
knowledgeFile.setDifyFileId(difyFileId); knowledgeFile.setDifyFileId(difyFileId);
knowledgeFile.setVersion(1); knowledgeFile.setVersion(1);
knowledgeMapper.updateKnowledgeFileCount(knowledgeId, 1);
int rows = knowledgeFileMapper.insertKnowledgeFile(knowledgeFile); int rows = knowledgeFileMapper.insertKnowledgeFile(knowledgeFile);
if (rows > 0) { if (rows > 0) {
logger.info("保存知识库文件记录成功: knowledgeId={}, fileId={}, difyFileId={}", knowledgeId, fileId, difyFileId); logger.info("保存知识库文件记录成功: knowledgeId={}, fileId={}, difyFileId={}", knowledgeId, fileId, difyFileId);
@@ -722,10 +723,13 @@ public class KnowledgeServiceImpl implements KnowledgeService {
if (!difyDocIds.isEmpty()) { if (!difyDocIds.isEmpty()) {
aiFileUploadService.batchDeleteFilesFromDify(knowledge.getDifyDatasetId(), difyDocIds); aiFileUploadService.batchDeleteFilesFromDify(knowledge.getDifyDatasetId(), difyDocIds);
} }
}else{
return ResultDomain.failure("知识库未关联Dify");
} }
// 3. 软删除本地记录和minio文件 // 3. 软删除本地记录和minio文件
int rows = knowledgeFileMapper.deleteFilesByRootId(fileRootId); int rows = knowledgeFileMapper.deleteFilesByRootId(fileRootId);
knowledgeMapper.updateKnowledgeFileCount(knowledge.getKnowledgeId(), -1);
if (rows > 0) { if (rows > 0) {
logger.info("删除知识库文件成功: fileRootId={}", fileRootId); logger.info("删除知识库文件成功: fileRootId={}", fileRootId);
for (TbKnowledgeFile file : versions) { for (TbKnowledgeFile file : versions) {

6
urbanLifelineServ/ai/src/main/resources/mapper/TbKnowledgeMapper.xml
View File

@@ -112,6 +112,12 @@
WHERE knowledge_id = #{knowledgeId} AND deleted = false WHERE knowledge_id = #{knowledgeId} AND deleted = false
</update> </update>
<update id="updateKnowledgeFileCount">
UPDATE ai.tb_knowledge
SET document_count = document_count + #{fileCount}
WHERE knowledge_id = #{knowledgeId} AND deleted = false
</update>
<select id="selectKnowledgeById" resultMap="BaseResultMap"> <select id="selectKnowledgeById" resultMap="BaseResultMap">
SELECT <include refid="Base_Column_List"/> SELECT <include refid="Base_Column_List"/>
FROM ai.tb_knowledge FROM ai.tb_knowledge