deploy 1.8.2

배포 1.8.2

Author: rvbear

src/main/java/com/day_walk/backend/domain/chat/service/SseEmitters.java

@@ -43,7 +43,7 @@ public SseEmitter add(UUID userId) {
         SseEmitter old = emitters.remove(userId);
 
         if (old != null) {
-            old.complete();
+            old.completeWithError(new CustomException(ErrorCode.SSE_CONNECTION_ERROR));
         }
 
         SseEmitter emitter = new SseEmitter(30_000L);
@@ -107,4 +107,20 @@ public void sendToClient(UUID userId, String question, GetChatByMlDto getChatByM
             }
         }
     }
+
+    public void sendToClientDone(UUID userId) {
+        SseEmitter emitter = emitters.get(userId);
+
+        if (emitter != null) {
+            try {
+                emitter.send(SseEmitter.event()
+                        .name("chatbot")
+                        .data("[DONE]"));
+
+                emitter.complete();
+            } catch (IOException e) {
+                throw new CustomException(ErrorCode.SSE_CONNECTION_ERROR);
+            }
+        }
+    }
 }

src/main/java/com/day_walk/backend/global/config/SseClient.java

@@ -50,6 +50,7 @@ public void onOpen(EventSource eventSource, Response response) {
             @Override
             public void onEvent(EventSource eventSource, String id, String type, String data) {
                 if ("[DONE]".equals(data)) {
+                    sseEmitters.sendToClientDone(userId);
                     System.out.println("ML 응답 스트림 종료 신호 수신");
                     return;
                 }