From dcfbf44fcde3b8f4551de9ed45bb9ab8ffb6c6db Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Tue, 30 Sep 2025 13:37:56 +0800 Subject: [PATCH] feat(ai): 更新知识文件时自动设置更新时间 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java | 17 +++++++++++------ 1 files changed, 11 insertions(+), 6 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java index 7d7e8f7..221403a 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java @@ -122,6 +122,7 @@ Wrappers.lambdaUpdate(AiKnowledgeFile.class) .set(AiKnowledgeFile::getFileId, fileId) .set(AiKnowledgeFile::getJobId, jobId) + .set(AiKnowledgeFile::getUpdatedTime, new Date()) .eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId()) ); @@ -138,10 +139,11 @@ if (state == 1){ state = KnowledgeBaseUtil.getFileJobStatus(aiKnowledgeFile.getFileId()); - if ( state == 1) { + if ( state == 2) { aiKnowledgeFileMapper.update(null, Wrappers.lambdaUpdate(AiKnowledgeFile.class) - .set(AiKnowledgeFile::getState, 2) + .set(AiKnowledgeFile::getState, state) + .set(AiKnowledgeFile::getUpdatedTime, new Date()) .eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId()) ); } @@ -157,11 +159,12 @@ }else{ knowledgeId = KnowledgeBaseUtil.DEFAULT_KNOWLEDGE_ID; } - state = KnowledgeBaseUtil.getIndexKnowledgeJobStatus(aiKnowledgeFile.getFileId(),knowledgeId); - if ( state == 1) { + state = KnowledgeBaseUtil.getIndexKnowledgeJobStatus(aiKnowledgeFile.getJobId(),knowledgeId); + if ( state == 3) { aiKnowledgeFileMapper.update(null, Wrappers.lambdaUpdate(AiKnowledgeFile.class) - .set(AiKnowledgeFile::getState, 3) + .set(AiKnowledgeFile::getState, state) + .set(AiKnowledgeFile::getUpdatedTime, new Date()) .eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId()) ); } @@ -181,12 +184,14 @@ Path filePath = Paths.get(aiKnowledgeFile.getSavePath()); boolean deleted = Files.deleteIfExists(filePath); if (!deleted) { - throw new FebsException("文件删除成功"); + throw new FebsException("文件删除失败"); } } catch (Exception e) { throw new FebsException("删除文件时发生错误: " + e.getMessage()); } + aiKnowledgeFileMapper.deleteById(id); + //知识库删除 String knowledgeId = null; if (StrUtil.isNotEmpty(aiKnowledgeFile.getCompanyId())){ -- Gitblit v1.9.1