From 47806d9e390ee00e1d29ad1da8c1aa908882a758 Mon Sep 17 00:00:00 2001
From: yuan <123@>
Date: 星期二, 11 八月 2026 14:43:59 +0800
Subject: [PATCH] 用河南鹤壁代码覆盖山西长治
---
src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java | 146 +++++++-----------------------------------------
1 files changed, 22 insertions(+), 124 deletions(-)
diff --git a/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java b/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java
index ba93fb4..4bbb84b 100644
--- a/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java
+++ b/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java
@@ -2,25 +2,18 @@
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
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.StorageBlob;
-import com.ruoyi.basic.service.StorageAttachmentService;
-import com.ruoyi.basic.service.StorageBlobService;
import com.ruoyi.common.utils.poi.ExcelUtil;
+import com.ruoyi.framework.aspectj.lang.annotation.Log;
+import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.framework.web.domain.AjaxResult;
-import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
+import io.swagger.v3.oas.annotations.Operation;
import jakarta.servlet.http.HttpServletResponse;
import lombok.AllArgsConstructor;
import org.springframework.web.bind.annotation.*;
-import java.util.ArrayList;
import java.util.List;
@RestController
@@ -29,45 +22,47 @@
@Tag(name = "鐭ヨ瘑搴撶鐞�")
public class KnowledgeBaseController {
private KnowledgeBaseService knowledgeBaseService;
- private KnowledgeBaseVectorService knowledgeBaseVectorService;
- private StorageAttachmentService storageAttachmentService;
- private StorageBlobService storageBlobService;
- /**
+ /**銆�
* 鑾峰彇鍒楄〃
+ * @return
*/
@GetMapping("/getList")
public AjaxResult getList(@RequestParam(defaultValue = "1") long current,
@RequestParam(defaultValue = "10") long size, KnowledgeBase knowledgeBase) {
Page page = new Page(current, size);
- return AjaxResult.success(knowledgeBaseService.listpage(page, knowledgeBase));
+ return AjaxResult.success(knowledgeBaseService.listpage(page,knowledgeBase));
}
-
- /**
- * 鏂板鐭ヨ瘑搴�
+ /**銆�
+ * 澧炴坊
+ * @return
*/
+ @Log(title = "鐭ヨ瘑搴�", businessType = BusinessType.INSERT)
@PostMapping("/add")
- public AjaxResult add(@RequestBody KnowledgeBase knowledgeBase) {
+ public AjaxResult add(@RequestBody KnowledgeBase knowledgeBase){
return AjaxResult.success(knowledgeBaseService.save(knowledgeBase));
}
-
/**
- * 鏇存柊鐭ヨ瘑搴�
+ * 鏇存柊
+ * @return
*/
+ @Log(title = "鐭ヨ瘑搴�", businessType = BusinessType.UPDATE)
@PostMapping("/update")
- public AjaxResult update(@RequestBody KnowledgeBase knowledgeBase) {
+ public AjaxResult update(@RequestBody KnowledgeBase knowledgeBase){
return AjaxResult.success(knowledgeBaseService.updateById(knowledgeBase));
}
-
/**
- * 鍒犻櫎鐭ヨ瘑搴�
+ * 鍒犻櫎
+ * @return
*/
+ @Log(title = "鐭ヨ瘑搴�", businessType = BusinessType.DELETE)
@DeleteMapping("/delete")
- public AjaxResult delete(@RequestBody List<Long> ids) {
- if (CollectionUtils.isEmpty(ids)) return AjaxResult.error("璇蜂紶鍏ヨ鍒犻櫎鐨処D");
+ public AjaxResult delete(@RequestBody List<Long> ids){
+ if(CollectionUtils.isEmpty(ids)) return AjaxResult.error("璇蜂紶鍏ヨ鍒犻櫎鐨処D");
return AjaxResult.success(knowledgeBaseService.removeByIds(ids));
}
+ @Log(title = "鐭ヨ瘑搴�", businessType = BusinessType.EXPORT)
@Operation(summary = "鐭ヨ瘑搴撶鐞嗗鍑�")
@PostMapping("/export")
public void export(HttpServletResponse response) {
@@ -76,101 +71,4 @@
util.exportExcel(response, accountExpenses, "鐭ヨ瘑搴撶鐞嗗鍑�");
}
- /**
- * 鏌ヨ鐭ヨ瘑搴撴枃浠跺悜閲忓寲鐘舵��
- */
- @GetMapping("/vector/status/{knowledgeBaseId}")
- @Operation(summary = "鏌ヨ鐭ヨ瘑搴撴枃浠跺悜閲忓寲鐘舵��")
- public AjaxResult getVectorStatus(@PathVariable Long knowledgeBaseId) {
- List<KnowledgeBaseVectorVO> list = knowledgeBaseVectorService.getVectorStatusByKnowledgeBaseId(knowledgeBaseId);
- return AjaxResult.success(list);
- }
-
- /**
- * 閲嶆柊鍚戦噺鍖栨枃浠�
- */
- @PostMapping("/vector/reprocess/{vectorId}")
- @Operation(summary = "閲嶆柊鍚戦噺鍖栨枃浠�")
- public AjaxResult reprocessVector(@PathVariable Long vectorId) {
- knowledgeBaseVectorService.reprocessVector(vectorId);
- return AjaxResult.success("宸查噸鏂版彁浜ゅ悜閲忓寲浠诲姟");
- }
-
- /**
- * 淇濆瓨鐭ヨ瘑搴撴枃浠跺叧鑱旓紙鏂囦欢涓婁紶鍚庤皟鐢級
- * 涓婁紶娴佺▼锛�
- * 1. 鍏堣皟鐢� /common/upload 涓婁紶鏂囦欢锛岃幏鍙� storageBlobDTOs
- * 2. 鍐嶈皟鐢ㄦ鎺ュ彛鍏宠仈鏂囦欢鍒扮煡璇嗗簱骞惰Е鍙戝悜閲忓寲
- */
- @PostMapping("/file/save")
- @Operation(summary = "淇濆瓨鐭ヨ瘑搴撴枃浠跺叧鑱�")
- public AjaxResult saveKnowledgeBaseFiles(@RequestBody KnowledgeBaseFileDTO dto) {
- if (dto.getKnowledgeBaseId() == null) {
- return AjaxResult.error("鐭ヨ瘑搴揑D涓嶈兘涓虹┖");
- }
- if (CollectionUtils.isEmpty(dto.getStorageBlobIds())) {
- return AjaxResult.error("鏂囦欢ID涓嶈兘涓虹┖");
- }
-
- // 淇濆瓨闄勪欢鍏宠仈
- StorageAttachmentDTO attachmentDTO = new StorageAttachmentDTO();
- attachmentDTO.setRecordType("knowledge_base");
- attachmentDTO.setRecordId(dto.getKnowledgeBaseId());
- attachmentDTO.setApplication("rag_file");
- List<StorageBlobDTO> blobDTOs = new ArrayList<>();
- for (Long blobId : dto.getStorageBlobIds()) {
- StorageBlobDTO blobDTO = new StorageBlobDTO();
- blobDTO.setId(blobId);
- blobDTOs.add(blobDTO);
- }
- attachmentDTO.setStorageBlobDTOs(blobDTOs);
- storageAttachmentService.saveStorageAttachment(attachmentDTO);
-
- // 鍒涘缓鍚戦噺璁板綍骞惰Е鍙戝悜閲忓寲
- for (Long blobId : dto.getStorageBlobIds()) {
- StorageBlob blob = storageBlobService.getById(blobId);
- if (blob != null) {
- String fileName = blob.getOriginalFilename();
- String fileType = getFileExtension(fileName);
-
- knowledgeBaseVectorService.createVectorRecord(
- dto.getKnowledgeBaseId(),
- blobId,
- fileName,
- fileType
- );
- }
- }
-
- return AjaxResult.success();
- }
-
- private String getFileExtension(String fileName) {
- if (fileName == null || !fileName.contains(".")) {
- return "unknown";
- }
- return fileName.substring(fileName.lastIndexOf('.') + 1).toLowerCase();
- }
-
- /**
- * 鍒犻櫎鐭ヨ瘑搴撴枃浠�
- */
- @DeleteMapping("/file/delete")
- @Operation(summary = "鍒犻櫎鐭ヨ瘑搴撴枃浠�")
- public AjaxResult deleteKnowledgeBaseFiles(@RequestBody List<Long> vectorIds) {
- if (CollectionUtils.isEmpty(vectorIds)) {
- return AjaxResult.error("璇烽�夋嫨瑕佸垹闄ょ殑鏂囦欢");
- }
- knowledgeBaseVectorService.deleteVectors(vectorIds);
- return AjaxResult.success();
- }
-
- /**
- * 鐭ヨ瘑搴撴枃浠禗TO
- */
- @lombok.Data
- public static class KnowledgeBaseFileDTO {
- private Long knowledgeBaseId;
- private List<Long> storageBlobIds;
- }
-}
\ No newline at end of file
+}
--
Gitblit v1.9.3