知识库文件数量修正

urbanLifelineServ/ai/src/main/java/org/xyzh/ai/mapper/TbKnowledgeMapper.java

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

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

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

urbanLifelineServ/ai/src/main/resources/mapper/TbKnowledgeMapper.xml

@@ -112,6 +112,12 @@
         WHERE knowledge_id = #{knowledgeId} AND deleted = false
     </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 <include refid="Base_Column_List"/>
         FROM ai.tb_knowledge