``` refactor(knowledge-base): 重构RAG向量检索功能的文件关联和异步处理

云

2026-06-09 f818fa5eca223270f0656e0455c338ec1f2fac88

 src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java

@@ -4,14 +4,13 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.approve.dto.KnowledgeBaseVectorVO;
import com.ruoyi.approve.pojo.KnowledgeBase;
import com.ruoyi.approve.pojo.KnowledgeBaseVector;
import com.ruoyi.approve.service.KnowledgeBaseService;
import com.ruoyi.approve.service.KnowledgeBaseVectorService;
import com.ruoyi.basic.dto.StorageAttachmentDTO;
import com.ruoyi.basic.dto.StorageBlobDTO;
import com.ruoyi.basic.pojo.StorageAttachment;
import com.ruoyi.basic.pojo.StorageBlob;
import com.ruoyi.basic.service.StorageAttachmentService;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.basic.service.StorageBlobService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.web.domain.AjaxResult;
import io.swagger.v3.oas.annotations.Operation;
@@ -31,6 +30,7 @@
    private KnowledgeBaseService knowledgeBaseService;
    private KnowledgeBaseVectorService knowledgeBaseVectorService;
    private StorageAttachmentService storageAttachmentService;
    private StorageBlobService storageBlobService;

    /**
     * 获取列表
@@ -128,18 +128,10 @@
        // 创建向量记录并触发向量化
        for (Long blobId : dto.getStorageBlobIds()) {
            // 获取文件信息
            var blob = storageAttachmentService.getBaseMapper()
                    .selectOne(com.baomidou.mybatisplus.core.toolkit.Wrappers.<StorageAttachment>lambdaQuery()
                            .eq(StorageAttachment::getStorageBlobId, blobId)
                            .eq(StorageAttachment::getRecordType, "knowledge_base")
                            .eq(StorageAttachment::getRecordId, dto.getKnowledgeBaseId())
                            .last("limit 1"));

            StorageBlob blob = storageBlobService.getById(blobId);
            if (blob != null) {
                // 获取文件名，需要从 storage_blob 表获取
                // 这里简化处理，实际需要查询 storage_blob 表
                String fileName = "file_" + blobId;
                String fileType = "unknown";
                String fileName = blob.getOriginalFilename();
                String fileType = getFileExtension(fileName);

                knowledgeBaseVectorService.createVectorRecord(
                        dto.getKnowledgeBaseId(),
@@ -153,6 +145,13 @@
        return AjaxResult.success();
    }

    private String getFileExtension(String fileName) {
        if (fileName == null || !fileName.contains(".")) {
            return "unknown";
        }
        return fileName.substring(fileName.lastIndexOf('.') + 1).toLowerCase();
    }

    /**
     * 删除知识库文件
     */