Bruce/common_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add enum dto APIs and batch save

Browse Source
This commit is contained in:
bruce
2026-05-20 23:25:30 +08:00
parent 16f9a325d7
commit 7188dd49b5
20 changed files with 571 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/com/bruce/rag/controller/RagDocumentController.java
View File

@@ -2,7 +2,7 @@ package com.bruce.rag.controller;
import com.bruce.common.domain.model.RequestResult;
import com.bruce.rag.dto.request.RagDocumentQueryRequest;
import com.bruce.rag.entity.RagDocument;
import com.bruce.rag.dto.response.RagDocumentResponse;
import com.bruce.rag.service.IRagDocumentService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
@@ -25,13 +25,13 @@ public class RagDocumentController {
@Operation(summary = "查询全部知识库文档")
@GetMapping
public RequestResult<List<RagDocument>> list() {
return RequestResult.success(ragDocumentService.list());
public RequestResult<List<RagDocumentResponse>> list() {
return RequestResult.success(ragDocumentService.listResponses());
}
@Operation(summary = "按条件查询知识库文档")
@PostMapping("/query")
public RequestResult<List<RagDocument>> query(@RequestBody RagDocumentQueryRequest request) {
public RequestResult<List<RagDocumentResponse>> query(@RequestBody RagDocumentQueryRequest request) {
return RequestResult.success(ragDocumentService.query(request));
}
}

8
src/main/java/com/bruce/rag/controller/RagStoreController.java
View File

@@ -2,7 +2,7 @@ package com.bruce.rag.controller;
import com.bruce.common.domain.model.RequestResult;
import com.bruce.rag.dto.request.RagStoreQueryRequest;
import com.bruce.rag.entity.RagStore;
import com.bruce.rag.dto.response.RagStoreResponse;
import com.bruce.rag.service.IRagStoreService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
@@ -25,13 +25,13 @@ public class RagStoreController {
@Operation(summary = "查询全部知识库")
@GetMapping
public RequestResult<List<RagStore>> list() {
return RequestResult.success(ragStoreService.list());
public RequestResult<List<RagStoreResponse>> list() {
return RequestResult.success(ragStoreService.listResponses());
}
@Operation(summary = "按条件查询知识库")
@PostMapping("/query")
public RequestResult<List<RagStore>> query(@RequestBody RagStoreQueryRequest request) {
public RequestResult<List<RagStoreResponse>> query(@RequestBody RagStoreQueryRequest request) {
return RequestResult.success(ragStoreService.query(request));
}
}

50
src/main/java/com/bruce/rag/dto/response/RagDocumentResponse.java Normal file
View File

@@ -0,0 +1,50 @@
package com.bruce.rag.dto.response;
import com.bruce.rag.entity.RagDocument;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import org.springframework.beans.BeanUtils;
@Data
@Schema(description = "RAG知识库文档响应")
public class RagDocumentResponse {
@Schema(description = "主键ID")
private Long id;
@Schema(description = "知识库ID")
private Long storeId;
@Schema(description = "附件ID")
private Long attachmentId;
@Schema(description = "文档标题")
private String documentTitle;
@Schema(description = "文档摘要")
private String documentSummary;
@Schema(description = "解析状态")
private String parseStatus;
@Schema(description = "索引状态")
private String indexStatus;
@Schema(description = "是否启用")
private Boolean enabled;
@Schema(description = "失败原因")
private String errorMessage;
@Schema(description = "备注")
private String remark;
public static RagDocumentResponse fromEntity(RagDocument entity) {
if (entity == null) {
return null;
}
RagDocumentResponse response = new RagDocumentResponse();
BeanUtils.copyProperties(entity, response);
return response;
}
}

38
src/main/java/com/bruce/rag/dto/response/RagStoreResponse.java Normal file
View File

@@ -0,0 +1,38 @@
package com.bruce.rag.dto.response;
import com.bruce.rag.entity.RagStore;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import org.springframework.beans.BeanUtils;
@Data
@Schema(description = "RAG知识库响应")
public class RagStoreResponse {
@Schema(description = "主键ID")
private Long id;
@Schema(description = "知识库编码")
private String storeCode;
@Schema(description = "知识库名称")
private String storeName;
@Schema(description = "知识库描述")
private String description;
@Schema(description = "状态")
private String status;
@Schema(description = "备注")
private String remark;
public static RagStoreResponse fromEntity(RagStore entity) {
if (entity == null) {
return null;
}
RagStoreResponse response = new RagStoreResponse();
BeanUtils.copyProperties(entity, response);
return response;
}
}

5
src/main/java/com/bruce/rag/service/IRagDocumentService.java
View File

@@ -2,11 +2,14 @@ package com.bruce.rag.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.bruce.rag.dto.request.RagDocumentQueryRequest;
import com.bruce.rag.dto.response.RagDocumentResponse;
import com.bruce.rag.entity.RagDocument;
import java.util.List;
public interface IRagDocumentService extends IService<RagDocument> {
List<RagDocument> query(RagDocumentQueryRequest request);
List<RagDocumentResponse> listResponses();
List<RagDocumentResponse> query(RagDocumentQueryRequest request);
}

5
src/main/java/com/bruce/rag/service/IRagStoreService.java
View File

@@ -2,11 +2,14 @@ package com.bruce.rag.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.bruce.rag.dto.request.RagStoreQueryRequest;
import com.bruce.rag.dto.response.RagStoreResponse;
import com.bruce.rag.entity.RagStore;
import java.util.List;
public interface IRagStoreService extends IService<RagStore> {
List<RagStore> query(RagStoreQueryRequest request);
List<RagStoreResponse> listResponses();
List<RagStoreResponse> query(RagStoreQueryRequest request);
}

18
src/main/java/com/bruce/rag/service/impl/RagDocumentServiceImpl.java
View File

@@ -2,6 +2,7 @@ package com.bruce.rag.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.bruce.rag.dto.request.RagDocumentQueryRequest;
import com.bruce.rag.dto.response.RagDocumentResponse;
import com.bruce.rag.entity.RagDocument;
import com.bruce.rag.mapper.RagDocumentMapper;
import com.bruce.rag.service.IRagDocumentService;
@@ -13,17 +14,28 @@ import java.util.List;
public class RagDocumentServiceImpl extends ServiceImpl<RagDocumentMapper, RagDocument> implements IRagDocumentService {
@Override
public List<RagDocument> query(RagDocumentQueryRequest request) {
public List<RagDocumentResponse> listResponses() {
return toResponses(list());
}
@Override
public List<RagDocumentResponse> query(RagDocumentQueryRequest request) {
if (request == null) {
throw new IllegalArgumentException("查询请求不能为空");
}
return lambdaQuery()
return toResponses(lambdaQuery()
.eq(request.getStoreId() != null, RagDocument::getStoreId, request.getStoreId())
.eq(request.getAttachmentId() != null, RagDocument::getAttachmentId, request.getAttachmentId())
.eq(request.getParseStatus() != null, RagDocument::getParseStatus, request.getParseStatus())
.eq(request.getIndexStatus() != null, RagDocument::getIndexStatus, request.getIndexStatus())
.eq(request.getEnabled() != null, RagDocument::getEnabled, request.getEnabled())
.orderByDesc(RagDocument::getId)
.list();
.list());
}
private List<RagDocumentResponse> toResponses(List<RagDocument> documents) {
return documents.stream()
.map(RagDocumentResponse::fromEntity)
.toList();
}
}

18
src/main/java/com/bruce/rag/service/impl/RagStoreServiceImpl.java
View File

@@ -2,6 +2,7 @@ package com.bruce.rag.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.bruce.rag.dto.request.RagStoreQueryRequest;
import com.bruce.rag.dto.response.RagStoreResponse;
import com.bruce.rag.entity.RagStore;
import com.bruce.rag.mapper.RagStoreMapper;
import com.bruce.rag.service.IRagStoreService;
@@ -14,15 +15,26 @@ import java.util.List;
public class RagStoreServiceImpl extends ServiceImpl<RagStoreMapper, RagStore> implements IRagStoreService {
@Override
public List<RagStore> query(RagStoreQueryRequest request) {
public List<RagStoreResponse> listResponses() {
return toResponses(list());
}
@Override
public List<RagStoreResponse> query(RagStoreQueryRequest request) {
if (request == null) {
throw new IllegalArgumentException("查询请求不能为空");
}
return lambdaQuery()
return toResponses(lambdaQuery()
.eq(StringUtils.hasText(request.getStoreCode()), RagStore::getStoreCode, request.getStoreCode())
.like(StringUtils.hasText(request.getStoreName()), RagStore::getStoreName, request.getStoreName())
.eq(StringUtils.hasText(request.getStatus()), RagStore::getStatus, request.getStatus())
.orderByAsc(RagStore::getStoreCode)
.list();
.list());
}
private List<RagStoreResponse> toResponses(List<RagStore> stores) {
return stores.stream()
.map(RagStoreResponse::fromEntity)
.toList();
}
}