From c3a79d24dc10551b7a7830f1d11f66e3adb1430e Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Sat, 11 Oct 2025 09:54:11 +0800
Subject: [PATCH] feat(ai): 实现报告列表流式传输功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java | 35 ++++++++++++++++++++++++++++++-----
src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java | 2 +-
src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java | 2 +-
3 files changed, 32 insertions(+), 7 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java b/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java
index dcb7346..6557e59 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java
@@ -65,7 +65,7 @@
@ApiResponse(code = 200, message = "success", response = ApiTalkReportListVo.class)
})
@PostMapping(value = "/talkReportList")
- public FebsResponse talkReportList(@RequestBody @Validated ApiTalkReportListDto dto) {
+ public Flux<FebsResponse> talkReportList(@RequestBody @Validated ApiTalkReportListDto dto) {
return aiMemberTalkStreamService.talkReportList(dto);
}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java b/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java
index df9144a..4befa80 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java
@@ -24,7 +24,7 @@
FebsResponse talkMemberList(ApiTalkMemberListDto dto);
- FebsResponse talkReportList(ApiTalkReportListDto dto);
+ Flux<FebsResponse> talkReportList(ApiTalkReportListDto dto);
FebsResponse reload(ApiMemberTalkReloadStreamDto dto);
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
index a3a9a44..ac91786 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
@@ -122,15 +122,40 @@
}
@Override
- public FebsResponse talkReportList(ApiTalkReportListDto dto) {
- ApiTalkReportListVo apiTalkReportListDto = new ApiTalkReportListVo();
-
+ public Flux<FebsResponse> talkReportList(ApiTalkReportListDto dto) {
String memberTalkId = dto.getMemberTalkId();
AiMemberTalk byId = aiMemberTalkService.getById(memberTalkId);
- apiTalkReportListDto.setAnalysis(byId.getAnalysis());
- return new FebsResponse().success().data(apiTalkReportListDto);
+ String analysis = byId.getAnalysis();
+
+ if (StrUtil.isEmpty(analysis)) {
+ return Flux.just(new FebsResponse().success().data(""));
+ }
+
+ // 将分析报告分块，每块100个字符
+ return Flux.generate(
+ () -> 0, // 初始状态
+ (index, sink) -> {
+ int endIndex = Math.min(index + 100, analysis.length());
+ String chunk = analysis.substring(index, endIndex);
+
+ Map<String, Object> responseData = new HashMap<>();
+ responseData.put("chunk", chunk);
+ responseData.put("isLast", endIndex >= analysis.length());
+ responseData.put("progress", Math.min(100, (endIndex * 100) / analysis.length()));
+
+ sink.next(new FebsResponse().success().data(responseData));
+
+ if (endIndex >= analysis.length()) {
+ sink.complete();
+ return index; // 结束状态
+ }
+
+ return endIndex; // 下一次的起始索引
+ }
+ );
}
+
@Override
public FebsResponse reload(ApiMemberTalkReloadStreamDto dto) {
--
Gitblit v1.9.1