1.生成会话

2.刷新会话

src/main/java/com/rf/AIquantum/dialogue/rest/ChatHistoryController.java

@@ -3,33 +3,32 @@ package com.rf.AIquantum.dialogue.rest;
 import cn.hutool.core.date.DateUtil;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
-import com.auth0.jwt.interfaces.DecodedJWT;
 import com.rf.AIquantum.base.rest.BaseController;
+import com.rf.AIquantum.dao.dto.SseResultDataDto;
 import com.rf.AIquantum.dialogue.dao.model.ChatHistoryEntity;
-import com.rf.AIquantum.dialogue.dao.model.DialogueEntity;
 import com.rf.AIquantum.dialogue.service.ChatHistoryService;
-import com.rf.AIquantum.user.dao.model.UserEntity;
-import com.rf.AIquantum.user.service.UserService;
+import com.rf.AIquantum.filter.JwtIgnore;
 import com.rf.AIquantum.utils.Constant;
-import com.rf.AIquantum.utils.JWTUtil;
 import com.rf.AIquantum.utils.Result;
+import com.rf.AIquantum.utils.SseEmitterService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.MediaType;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.Response;
+import okio.BufferedSource;
 import org.apache.commons.lang.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.domain.Page;
-import org.springframework.util.DigestUtils;
 import org.springframework.web.bind.annotation.*;
-
-import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
-import java.io.File;
-import java.io.UnsupportedEncodingException;
-import java.nio.charset.StandardCharsets;
+import java.io.BufferedReader;
+import java.io.IOException;
 import java.util.Date;
 import java.util.List;
-
-import static com.rf.AIquantum.dialogue.rest.DialogueController.HttpClientChat;
+import java.util.concurrent.TimeUnit;
 
 /**
  * @Description: 聊天记录相关接口
@@ -38,71 +37,149 @@ import static com.rf.AIquantum.dialogue.rest.DialogueController.HttpClientChat;
  */
 @RestController
 @RequestMapping("/chat")
+@Slf4j
 @Api(tags = "聊天记录相关接口")
 public class ChatHistoryController extends BaseController {
 
     @Autowired
     private ChatHistoryService chatHistoryService;
 
+    @Autowired
+    private SseEmitterService sseEmitterService;
+
     @GetMapping("/findChats")
-    @ApiOperation(value = "查询聊天记录",notes = "参数包括：pageNum:页码, pageSize:数量, dialogueId:对话id")
-    public Result findChatHistorys(@RequestParam int pageNum, @RequestParam int pageSize, @RequestParam String dialogueId){
-        Page<ChatHistoryEntity> chatHistoryEntities = chatHistoryService.findByDialogueIdAndStatus(pageNum,pageSize,dialogueId,1);
+    @ApiOperation(value = "查询聊天记录", notes = "参数包括：pageNum:页码, pageSize:数量, dialogueId:对话id")
+    public Result findChatHistorys(@RequestParam int pageNum, @RequestParam int pageSize, @RequestParam String dialogueId) {
+        Page<ChatHistoryEntity> chatHistoryEntities = chatHistoryService.findByDialogueIdAndStatus(pageNum, pageSize, dialogueId, 1);
         return success(chatHistoryEntities);
     }
 
     @PostMapping("/updateChatHistory")
-    @ApiOperation(value = "修改聊天记录信息（点赞、点踩）",notes = "ChatHistory对象,点赞时只需将endorse的值改为2；点赞时只需将endorse的值改为3,如果有反馈意见放到feedback字段；endorse：1：未评价（默认）；2：赞同；3：不赞同")
+    @ApiOperation(value = "修改聊天记录信息（点赞、点踩）", notes = "ChatHistory对象,点赞时只需将endorse的值改为2；点赞时只需将endorse的值改为3,如果有反馈意见放到feedback字段；endorse：1：未评价（默认）；2：赞同；3：不赞同")
     public Result updateChatHistory(@RequestBody String json) {
-        ChatHistoryEntity chatHistoryEntity = JSONObject.parseObject(json,ChatHistoryEntity.class);
+        ChatHistoryEntity chatHistoryEntity = JSONObject.parseObject(json, ChatHistoryEntity.class);
         chatHistoryEntity.setUpdateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss.SSS"));
         this.chatHistoryService.save(chatHistoryEntity);
         return success("修改成功");
     }
 
     @PostMapping("/refresh")
-    @ApiOperation(value = "刷新某条回答",notes = "ChatHistory对象")
-    public Result refresh(@RequestBody String json) throws UnsupportedEncodingException {
-        ChatHistoryEntity chatHistoryEntity = JSONObject.parseObject(json,ChatHistoryEntity.class);
+    @JwtIgnore
+    @ApiOperation(value = "刷新某条回答", notes = "ChatHistory对象")
+    public Result refresh(@RequestBody String json) {
+        ChatHistoryEntity chatHistoryEntity = JSONObject.parseObject(json, ChatHistoryEntity.class);
         String dialogueId = chatHistoryEntity.getDialogueId();
         String createTime = chatHistoryEntity.getCreateTime();
-        this.chatHistoryService.deleteByDialogueIdAndCreateTime(dialogueId,createTime);
+        this.chatHistoryService.deleteByDialogueIdAndCreateTime(dialogueId, createTime);
         //调用模型相关操作
         List<ChatHistoryEntity> chatHistoryEntities = this.chatHistoryService.findChatHistoryByDialogueIdAndStatus(dialogueId);
         JSONArray messages = new JSONArray();
         for (ChatHistoryEntity chatHistory : chatHistoryEntities) {
             JSONArray contents = new JSONArray();
             JSONObject jsonText = new JSONObject();
-            jsonText.put("type","text");
-            jsonText.put("text",chatHistory.getContent());
+            jsonText.put("type", "text");
+            jsonText.put("text", chatHistory.getContent());
             if (chatHistory.getImage() != null && !chatHistory.getImage().equals("")) {
                 JSONObject jsonImage = new JSONObject();
-                jsonImage.put("type","image_url");
+                jsonImage.put("type", "image_url");
                 JSONObject jsonUrl = new JSONObject();
-                jsonUrl.put("url",chatHistory.getImage());
-                jsonImage.put("image_url",jsonUrl);
+                jsonUrl.put("url", chatHistory.getImage());
+                jsonImage.put("image_url", jsonUrl);
                 contents.add(jsonImage);
             }
             contents.add(jsonText);
             JSONObject jsonRole = new JSONObject();
-            jsonRole.put("role",chatHistory.getRole());
-            jsonRole.put("content",contents);
+            jsonRole.put("role", chatHistory.getRole());
+            jsonRole.put("content", contents);
             messages.add(jsonRole);
         }
         JSONObject jsonChat = new JSONObject();
-        jsonChat.put("messages",messages);
+        jsonChat.put("messages", messages);
 
-        String url = Constant.INVOKE_IP_PROT + Constant.CHAT_PATH;
-        String data = HttpClientChat(jsonChat,url);
-        JSONObject jsonSystem = JSONObject.parseObject(data);
-        if (jsonSystem == null || !jsonSystem.containsKey("response")) {
-            return fail("", "模型服务内部错误");
+//        String url = Constant.INVOKE_IP_PROT + Constant.CHAT_PATH;
+//        String data = HttpClientChat(jsonChat,url);
+//        JSONObject jsonSystem = JSONObject.parseObject(data);
+//        if (jsonSystem == null || !jsonSystem.containsKey("response")) {
+//            return fail("", "模型服务内部错误");
+//        }
+//        String content = jsonSystem.getString("response");
+        SseResultDataDto sseResultDataDto = new SseResultDataDto();
+        sseResultDataDto.setDialogueId(dialogueId);
+        OkHttpClient client = new OkHttpClient.Builder().connectTimeout(60, TimeUnit.MINUTES).writeTimeout(60, TimeUnit.MINUTES).readTimeout(60, TimeUnit.MINUTES).build();
+        StringBuilder stringBuilder = new StringBuilder();
+        // 创建请求体
+        okhttp3.RequestBody requestBody = okhttp3.RequestBody.create(jsonChat.toJSONString(), MediaType.parse("application/json"));
+
+        // 创建请求
+        Request request = new Request.Builder().header("Content-Type", "application/json").header("Accept", "text/event-stream")
+                .url(Constant.INVOKE_IP_PROT + Constant.CHAT_PATH)
+                .post(requestBody)
+                .build();
+
+        // 发送请求并处理响应
+        try (Response response = client.newCall(request).execute()) {
+            if (!response.isSuccessful()) {
+                log.error("Unexpected code " + response);
+                return fail();
+            }
+            // 获取响应流
+            okhttp3.ResponseBody responseBody = response.body();
+            String flag = "think";
+            if (responseBody != null) {
+                try ( BufferedSource bufferedSource = responseBody.source()) {
+                    String line;
+                    byte[] buffer = new byte[1024];
+                    while (!bufferedSource.exhausted() ) {
+                        int bytesRead =  bufferedSource.read(buffer);
+                        line = new String(buffer, 0, bytesRead);
+                        if( sseEmitterService.getEmitter(dialogueId) == null){
+                            break;
+                        }
+                        sseResultDataDto.setType(flag);
+                        System.out.println("line=="+line);
+                        if(StringUtils.isEmpty(line) ||!line.contains("data:") || !line.contains("event")){
+                            continue;
+                        }
+                        String[] datas = line.split("data:");
+                        String event = datas[0].split("event:")[1];
+                        if(event.equals("error")){
+                            if(flag.equals("think")){
+                                stringBuilder.append("</").append(flag).append(">");
+                            }
+                            break;
+                        }
+                        if(event.equals("start\n") || event.equals("done\n")){
+                            continue;
+                        }
+                        for (int i=1;i<datas.length;i++){
+                            String data = datas[i];
+                            data =  data.replace("\n\n","");
+                            if(StringUtils.isNotEmpty(data)){
+                                if(i>1){
+                                    data = "\n\n"+data;
+                                }
+                                stringBuilder.append(data);
+                                sseResultDataDto.setContent(data);
+                            }
+                        }
+                        if(StringUtils.isNotEmpty(sseResultDataDto.getContent())){
+                            sseEmitterService.sendMessage(dialogueId, sseResultDataDto);
+                        }
+                        if (line.contains("</think>")){
+                            flag = "text";
+                        }
+                    }
+                }
+            }
+        } catch (IOException e) {
+            e.printStackTrace();
+            log.error(e.getMessage());
+            return fail();
         }
-        String content = jsonSystem.getString("response");
         chatHistoryEntity = new ChatHistoryEntity();
         chatHistoryEntity.setDialogueId(dialogueId);
         chatHistoryEntity.setRole("system");
-        chatHistoryEntity.setContent(content);
+        chatHistoryEntity.setContent(stringBuilder.toString());
         chatHistoryEntity.setStatus(1);
         chatHistoryEntity.setEndorse(1);
         chatHistoryEntity.setCreateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss.SSS"));
@@ -113,8 +190,8 @@ public class ChatHistoryController extends BaseController {
     }
 
     @GetMapping("/clearChats")
-    @ApiOperation(value = "清空聊天记录",notes = "参数包括：dialogueId:对话id")
-    public Result clearChats(@RequestParam String dialogueId){
+    @ApiOperation(value = "清空聊天记录", notes = "参数包括：dialogueId:对话id")
+    public Result clearChats(@RequestParam String dialogueId) {
         this.chatHistoryService.deleteByDialogueId(dialogueId);
         return success();
     }
@@ -122,7 +199,7 @@ public class ChatHistoryController extends BaseController {
     @PostMapping("/exportPdf")
     @ApiOperation(value = "导出PDF")
     public String exportPdf(@RequestBody String json, HttpServletResponse response) {
-        List<ChatHistoryEntity> chatHistoryEntityList = JSONArray.parseArray(json,ChatHistoryEntity.class);
+        List<ChatHistoryEntity> chatHistoryEntityList = JSONArray.parseArray(json, ChatHistoryEntity.class);
         if (chatHistoryEntityList.size() > 0) {
             return "导出内容为空";
             /*UserRecordEntity userRecordEntity = this.userRecordService.getUserRecordById(id);
@@ -157,7 +234,7 @@ public class ChatHistoryController extends BaseController {
                 e.printStackTrace();
                 return "下载失败";
             }*/
-        }else {
+        } else {
             return "导出内容为空";
         }
     }

src/main/java/com/rf/AIquantum/dialogue/rest/DialogueController.java

@@ -17,10 +17,12 @@ import com.rf.AIquantum.user.service.UserService;
 import com.rf.AIquantum.utils.*;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import lombok.extern.slf4j.Slf4j;
 import okhttp3.MediaType;
 import okhttp3.OkHttpClient;
 import okhttp3.Request;
 import okhttp3.Response;
+import okio.BufferedSource;
 import org.apache.commons.io.FileUtils;
 import org.apache.commons.lang.StringUtils;
 import org.apache.http.HttpEntity;
@@ -35,12 +37,16 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.domain.Page;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 
 import javax.servlet.http.HttpServletRequest;
 import java.io.*;
+import java.nio.charset.Charset;
+import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 /**
  * @Description: 对话相关接口
@@ -49,6 +55,7 @@ import java.util.List;
  */
 @RestController
 @RequestMapping("/dialogue")
+@Slf4j
 @Api(tags = "对话相关接口")
 public class DialogueController extends BaseController {
 
@@ -65,11 +72,12 @@ public class DialogueController extends BaseController {
     private SseEmitterService sseEmitterService;
 
     @PostMapping("/saveChat")
-    @ApiOperation(value = "保存对话",notes = "参数包括：phone:手机号, dialogueId:对话id（为空时是新建对话）, content:消息内容,image：图片（可以为空）")
-    public Result saveChat(MultipartFile image, String phone, String dialogueId, String content) throws UnsupportedEncodingException, InterruptedException {
+    @JwtIgnore
+    @ApiOperation(value = "保存对话", notes = "参数包括：phone:手机号, dialogueId:对话id（为空时是新建对话）, content:消息内容,image：图片（可以为空）")
+    public Result saveChat(MultipartFile image, String phone, String dialogueId, String content) throws InterruptedException {
         UserEntity user = this.userService.findUserByPhone(phone);
-        if (user == null){
-            return fail(null,"用户不存在");
+        if (user == null) {
+            return fail(null, "用户不存在");
         }
         String imageUrl = "";
         if (image != null) {
@@ -91,9 +99,9 @@ public class DialogueController extends BaseController {
             }
         }
         List<ChatHistoryEntity> chatHistoryEntities = chatHistoryService.findChatHistoryByDialogueIdAndStatus(dialogueId);
-        if (chatHistoryEntities.size() < 1 ){
-            if (content.length() > 50){
-                content = content.substring(0,50);
+        if (chatHistoryEntities.size() < 1) {
+            if (content.length() > 50) {
+                content = content.substring(0, 50);
             }
             //新建对话
             DialogueEntity dialogueEntity = new DialogueEntity();
@@ -121,25 +129,25 @@ public class DialogueController extends BaseController {
         for (ChatHistoryEntity chatHistory : chatHistoryEntities) {
             JSONArray contents = new JSONArray();
             JSONObject jsonText = new JSONObject();
-            jsonText.put("type","text");
-            jsonText.put("text",chatHistory.getContent());
+            jsonText.put("type", "text");
+            jsonText.put("text", chatHistory.getContent());
             if (chatHistory.getImage() != null && !chatHistory.getImage().equals("")) {
                 JSONObject jsonImage = new JSONObject();
-                jsonImage.put("type","image_url");
+                jsonImage.put("type", "image_url");
                 JSONObject jsonUrl = new JSONObject();
-                jsonUrl.put("url",chatHistory.getImage());
-                jsonImage.put("image_url",jsonUrl);
+                jsonUrl.put("url", chatHistory.getImage());
+                jsonImage.put("image_url", jsonUrl);
                 contents.add(jsonImage);
             }
             contents.add(jsonText);
             JSONObject jsonRole = new JSONObject();
-            jsonRole.put("role",chatHistory.getRole());
-            jsonRole.put("content",contents);
+            jsonRole.put("role", chatHistory.getRole());
+            jsonRole.put("content", contents);
             messages.add(jsonRole);
         }
         JSONObject jsonChat = new JSONObject();
-        jsonChat.put("messages",messages);
-        jsonChat.put("stream",true);
+        jsonChat.put("messages", messages);
+        jsonChat.put("stream", true);
 
        /* String url = Constant.INVOKE_IP_PROT + Constant.CHAT_PATH;
         String data = HttpClientChat(jsonChat,url);
@@ -148,10 +156,9 @@ public class DialogueController extends BaseController {
             return fail("", "模型服务内部错误");
         }
         content = jsonSystem.getString("response");*/
-
         SseResultDataDto sseResultDataDto = new SseResultDataDto();
         sseResultDataDto.setDialogueId(dialogueId);
-        OkHttpClient client = new OkHttpClient();
+        OkHttpClient client = new OkHttpClient.Builder().connectTimeout(60, TimeUnit.MINUTES).writeTimeout(60, TimeUnit.MINUTES).readTimeout(60, TimeUnit.MINUTES).build();
         StringBuilder stringBuilder = new StringBuilder();
         // 创建请求体
         okhttp3.RequestBody requestBody = okhttp3.RequestBody.create(jsonChat.toJSONString(), MediaType.parse("application/json"));
@@ -165,41 +172,63 @@ public class DialogueController extends BaseController {
         // 发送请求并处理响应
         try (Response response = client.newCall(request).execute()) {
             if (!response.isSuccessful()) {
-                throw new IOException("Unexpected code " + response);
+                log.error("Unexpected code " + response);
+                return fail();
             }
-
             // 获取响应流
             okhttp3.ResponseBody responseBody = response.body();
             String flag = "think";
             if (responseBody != null) {
-                try (BufferedReader reader = new BufferedReader(responseBody.charStream())) {
+                try ( BufferedSource bufferedSource = responseBody.source()) {
                     String line;
-                    while ((line = reader.readLine()) != null) {
-//                        System.out.println("line===="+line);
+                    byte[] buffer = new byte[1024];
+                    while (!bufferedSource.exhausted() ) {
+                        int bytesRead =  bufferedSource.read(buffer);
+                        line = new String(buffer, 0, bytesRead);
+                        if( sseEmitterService.getEmitter(dialogueId) == null){
+                            break;
+                        }
                         sseResultDataDto.setType(flag);
-                        int index = line.indexOf("data:");
-                        if(index != -1){
-                            String responseContent = line.substring(index+5);
-                            if(responseContent.trim().length() != 0){
-                                JSONObject jsonObject = JSONObject.parseObject(responseContent.trim());
-                                responseContent = jsonObject.getString("content");
-                                sseResultDataDto.setContent(responseContent);
-//                                System.out.println("flag==="+flag+"responseContent=="+responseContent); // 处理每一行响应
-                                stringBuilder.append(responseContent);
-//                                System.out.println("dto==="+ sseResultDataDto);
-                                sseEmitterService.sendMessage(dialogueId, sseResultDataDto);
-                                if (responseContent.contains("</think>")) {
-                                    flag = "text";
+                        System.out.println("line=="+line);
+                        if(StringUtils.isEmpty(line) ||!line.contains("data:") || !line.contains("event")){
+                            continue;
+                        }
+                        String[] datas = line.split("data:");
+                        String event = datas[0].split("event:")[1];
+                        if(event.equals("error")){
+                            if(flag.equals("think")){
+                                stringBuilder.append("</").append(flag).append(">");
+                            }
+                            break;
+                        }
+                        if(event.equals("start\n") || event.equals("done\n")){
+                            continue;
+                        }
+                        for (int i=1;i<datas.length;i++){
+                            String data = datas[i];
+                           data =  data.replace("\n\n","");
+                            if(StringUtils.isNotEmpty(data)){
+                                if(i>1){
+                                    data = "\n\n"+data;
                                 }
+                                stringBuilder.append(data);
+                                sseResultDataDto.setContent(data);
                             }
                         }
+                        if(StringUtils.isNotEmpty(sseResultDataDto.getContent())){
+                            sseEmitterService.sendMessage(dialogueId, sseResultDataDto);
+                        }
+                        if (line.contains("</think>")){
+                            flag = "text";
+                        }
                     }
                 }
             }
         } catch (IOException e) {
             e.printStackTrace();
+            log.error(e.getMessage());
+            return fail();
         }
-
         chatHistoryEntity = new ChatHistoryEntity();
         chatHistoryEntity.setDialogueId(dialogueId);
         chatHistoryEntity.setRole("system");
@@ -213,27 +242,35 @@ public class DialogueController extends BaseController {
         return success(chatHistoryEntity);
     }
 
+    @GetMapping("/stop/{dialogueId}")
+    @ApiOperation(value = "停止对话", notes = "参数包括：dialogueId:对话id")
+    public Result stop(@PathVariable String dialogueId) {
+
+        sseEmitterService.stopEmitter(dialogueId);
+        return success();
+    }
+
     @GetMapping("/findDialogues")
-    @ApiOperation(value = "查询对话列表",notes = "参数包括：pageNum:页码, pageSize:数量, search:搜索内容")
-    public Result findDialogues(@RequestParam int pageNum, @RequestParam int pageSize, @RequestParam String search, HttpServletRequest request){
+    @ApiOperation(value = "查询对话列表", notes = "参数包括：pageNum:页码, pageSize:数量, search:搜索内容")
+    public Result findDialogues(@RequestParam int pageNum, @RequestParam int pageSize, @RequestParam String search, HttpServletRequest request) {
         String token = request.getHeader("Authorization");
         String[] tokens = token.split(" ");
         if (tokens.length != 2) {
             token = tokens[0];
-        }else {
+        } else {
             token = tokens[1];
         }
         JWTUtil.verify(token);
         DecodedJWT verify = JWTUtil.verify(token);
         String phone = verify.getClaim("phone").asString();
-        Page<DialogueEntity> dialogueEntities = dialogueService.findByPhoneAndStatusAndSearch(pageNum,pageSize,phone,1,search);
+        Page<DialogueEntity> dialogueEntities = dialogueService.findByPhoneAndStatusAndSearch(pageNum, pageSize, phone, 1, search);
         return success(dialogueEntities);
     }
 
     @PostMapping("/updateDialogues")
-    @ApiOperation(value = "修改对话信息（重命名和删除）",notes = "Dialogues对象,修改对话名称时只需将修改后的名称放入dialogueName中；删除对话名称时只需将status的值改为0")
+    @ApiOperation(value = "修改对话信息（重命名和删除）", notes = "Dialogues对象,修改对话名称时只需将修改后的名称放入dialogueName中；删除对话名称时只需将status的值改为0")
     public Result updateDialogues(@RequestBody String json) {
-        DialogueEntity dialogueEntity = JSONObject.parseObject(json,DialogueEntity.class);
+        DialogueEntity dialogueEntity = JSONObject.parseObject(json, DialogueEntity.class);
         dialogueEntity.setUpdateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss.SSS"));
         this.dialogueService.save(dialogueEntity);
         return success("修改成功");
@@ -278,10 +315,10 @@ public class DialogueController extends BaseController {
     }*/
 
     public static String HttpClientChat(JSONObject jsonChat, String url) throws UnsupportedEncodingException {
-        System.out.println("11:"+jsonChat);
+        System.out.println("11:" + jsonChat);
         CloseableHttpClient httpclient = HttpClients.createDefault();
         HttpPost httpPost = new HttpPost(url);
-        StringEntity entity = new StringEntity(jsonChat.toString(),"UTF-8");
+        StringEntity entity = new StringEntity(jsonChat.toString(), "UTF-8");
         entity.setContentEncoding("UTF-8");
         entity.setContentType("application/json");
         httpPost.setEntity(entity);
@@ -289,9 +326,9 @@ public class DialogueController extends BaseController {
         String responseData = null;
         try {
             response = httpclient.execute(httpPost);
-            if (response.getStatusLine().getStatusCode() == 200){
-                responseData =  EntityUtils.toString(response.getEntity());
-                System.out.println("22:"+responseData);
+            if (response.getStatusLine().getStatusCode() == 200) {
+                responseData = EntityUtils.toString(response.getEntity());
+                System.out.println("22:" + responseData);
             }
         } catch (IOException e) {
             e.printStackTrace();
@@ -306,4 +343,8 @@ public class DialogueController extends BaseController {
         return responseData;
     }
 
+
+
+
+
 }

src/main/java/com/rf/AIquantum/dialogue/rest/SseController.java

@@ -6,11 +6,13 @@ package com.rf.AIquantum.dialogue.rest;
  * @Description:
  */
 import com.rf.AIquantum.filter.JwtIgnore;
+import com.rf.AIquantum.utils.Result;
 import com.rf.AIquantum.utils.SseEmitterService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.http.MediaType;
 import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
@@ -40,4 +42,7 @@ public class SseController {
 
         return emitter;
     }
+
+
+
 }

src/main/java/com/rf/AIquantum/utils/SseEmitterService.java

@@ -26,6 +26,10 @@ public class SseEmitterService {
         emitters.remove(clientId);
     }
 
+    public SseEmitter getEmitter(String clientId) {
+        return emitters.get(clientId);
+    }
+
     public void sendMessage(String clientId, SseResultDataDto message) {
         SseEmitter emitter = emitters.get(clientId);
         if (emitter != null) {
@@ -37,4 +41,12 @@ public class SseEmitterService {
             }
         }
     }
+
+    public void stopEmitter(String clientId) {
+        SseEmitter emitter = emitters.get(clientId);
+        if(emitter != null) {
+            emitter.complete();
+        }
+
+    }
 }