[MM-175] AI 채팅 메시지를 스트리밍 이후 분할하여 저장

src/main/java/makeus/cmc/malmo/adaptor/out/persistence/adapter/ChatRoomPersistenceAdapter.java

@@ -82,6 +82,20 @@ public ChatMessage saveChatMessage(ChatMessage chatMessage) {
         return chatMessageMapper.toDomain(savedEntity);
     }
 
+    @Override
+    public List<ChatMessage> saveChatMessages(List<ChatMessage> chatMessages) {
+        if (chatMessages == null || chatMessages.isEmpty()) {
+            return List.of();
+        }
+        List<ChatMessageEntity> entities = chatMessages.stream()
+                .map(chatMessageMapper::toEntity)
+                .toList();
+        List<ChatMessageEntity> savedEntities = chatMessageRepository.saveAll(entities);
+        return savedEntities.stream()
+                .map(chatMessageMapper::toDomain)
+                .toList();
+    }
+
     @Override
     public Optional<ChatRoom> loadChatRoomById(ChatRoomId chatRoomId) {
         return chatRoomRepository.findById(chatRoomId.getValue())

src/main/java/makeus/cmc/malmo/application/helper/chat_room/ChatRoomCommandHelper.java

@@ -31,4 +31,8 @@ public void deleteChatRooms(List<ChatRoomId> chatRoomIds) {
     public ChatMessage saveChatMessage(ChatMessage chatMessage) {
         return saveChatMessagePort.saveChatMessage(chatMessage);
     }
+
+    public List<ChatMessage> saveChatMessages(List<ChatMessage> chatMessages) {
+        return saveChatMessagePort.saveChatMessages(chatMessages);
+    }
 }

src/main/java/makeus/cmc/malmo/application/port/out/chat/SaveChatMessagePort.java

@@ -2,6 +2,10 @@
 
 import makeus.cmc.malmo.domain.model.chat.ChatMessage;
 
+import java.util.List;
+
 public interface SaveChatMessagePort {
     ChatMessage saveChatMessage(ChatMessage chatMessage);
+
+    List<ChatMessage> saveChatMessages(List<ChatMessage> chatMessages);
 }

src/main/java/makeus/cmc/malmo/application/service/chat/ChatMessageService.java

@@ -30,11 +30,13 @@
 import makeus.cmc.malmo.domain.value.id.CoupleId;
 import makeus.cmc.malmo.domain.value.id.CoupleQuestionId;
 import makeus.cmc.malmo.domain.value.id.MemberId;
+import makeus.cmc.malmo.util.ChatMessageSplitter;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
+import java.util.stream.Collectors;
 
 @Slf4j
 @Service
@@ -287,8 +289,23 @@ private CompletableFuture<Void> requestNextStageOpening(Member member, ChatRoom
     }
 
     private void saveAiMessage(MemberId memberId, ChatRoomId chatRoomId, int level, int detailedLevel, String fullAnswer) {
-        ChatMessage aiTextMessage = chatRoomDomainService.createAiMessage(chatRoomId, level, detailedLevel, fullAnswer);
-        ChatMessage savedMessage = chatRoomCommandHelper.saveChatMessage(aiTextMessage);
-        chatSseSender.sendAiResponseId(memberId, savedMessage.getId());
+        // fullAnswer를 문장 단위로 분할하고 세 문장씩 그룹화
+        List<String> groupedTexts = ChatMessageSplitter.splitIntoGroups(fullAnswer);
+
+        // 각 그룹을 ChatMessage로 생성
+        List<ChatMessage> chatMessages = groupedTexts.stream()
+                .map(groupText -> chatRoomDomainService.createAiMessage(chatRoomId, level, detailedLevel, groupText))
+                .collect(Collectors.toList());
+
+        // bulk 저장
+        List<ChatMessage> savedMessages = chatRoomCommandHelper.saveChatMessages(chatMessages);
+
+        // 저장된 메시지들의 ID 리스트 추출
+        List<Long> messageIds = savedMessages.stream()
+                .map(ChatMessage::getId)
+                .collect(Collectors.toList());
+
+        // SSE로 ID 리스트 전송
+        chatSseSender.sendAiResponseIds(memberId, messageIds);
     }
 }

src/main/java/makeus/cmc/malmo/application/service/chat/ChatService.java

@@ -19,10 +19,13 @@
 import makeus.cmc.malmo.domain.value.id.ChatRoomId;
 import makeus.cmc.malmo.domain.value.id.MemberId;
 import makeus.cmc.malmo.domain.value.state.ChatRoomState;
+import makeus.cmc.malmo.util.ChatMessageSplitter;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.util.List;
 import java.util.concurrent.CompletableFuture;
+import java.util.stream.Collectors;
 
 //import static makeus.cmc.malmo.util.GlobalConstants.FINAL_MESSAGE;
 
@@ -112,8 +115,23 @@ private ChatMessage saveUserMessage(ChatRoom chatRoom, String message) {
     }
 
     private void saveAiMessage(MemberId memberId, ChatRoomId chatRoomId, int level, int detailedLevel, String fullAnswer) {
-        ChatMessage aiTextMessage = chatRoomDomainService.createAiMessage(chatRoomId, level, detailedLevel, fullAnswer);
-        ChatMessage savedMessage = chatRoomCommandHelper.saveChatMessage(aiTextMessage);
-        chatSseSender.sendAiResponseId(memberId, savedMessage.getId());
+        // fullAnswer를 문장 단위로 분할하고 세 문장씩 그룹화
+        List<String> groupedTexts = ChatMessageSplitter.splitIntoGroups(fullAnswer);
+
+        // 각 그룹을 ChatMessage로 생성
+        List<ChatMessage> chatMessages = groupedTexts.stream()
+                .map(groupText -> chatRoomDomainService.createAiMessage(chatRoomId, level, detailedLevel, groupText))
+                .collect(Collectors.toList());
+
+        // bulk 저장
+        List<ChatMessage> savedMessages = chatRoomCommandHelper.saveChatMessages(chatMessages);
+
+        // 저장된 메시지들의 ID 리스트 추출
+        List<Long> messageIds = savedMessages.stream()
+                .map(ChatMessage::getId)
+                .collect(Collectors.toList());
+
+        // SSE로 ID 리스트 전송
+        chatSseSender.sendAiResponseIds(memberId, messageIds);
     }
 }

src/main/java/makeus/cmc/malmo/application/service/chat/ChatSseSender.java

@@ -5,6 +5,8 @@
 import makeus.cmc.malmo.domain.value.id.MemberId;
 import org.springframework.stereotype.Service;
 
+import java.util.List;
+
 @Service
 @RequiredArgsConstructor
 public class ChatSseSender {
@@ -32,6 +34,15 @@ public void sendAiResponseId(MemberId memberId, Long messageId) {
                 ));
     }
 
+    public void sendAiResponseIds(MemberId memberId, List<Long> messageIds) {
+        sendSseEventPort.sendToMember(
+                memberId,
+                new SendSseEventPort.NotificationEvent(
+                        SendSseEventPort.SseEventType.AI_RESPONSE_ID,
+                        messageIds
+                ));
+    }
+
     public void sendError(MemberId memberId, String errorMessage) {
         sendSseEventPort.sendToMember(
                 memberId,

src/main/java/makeus/cmc/malmo/util/ChatMessageSplitter.java

@@ -0,0 +1,106 @@
+package makeus.cmc.malmo.util;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+/**
+ * 채팅 메시지를 문장 단위로 분할하고 그룹화하는 유틸리티 클래스
+ */
+public class ChatMessageSplitter {
+
+    /**
+     * 문장 종결 부호 패턴 (규칙 변경 시 이 값만 수정하면 됨)
+     */
+    private static final Pattern SENTENCE_END_PATTERN = Pattern.compile("[.!?]");
+
+    /**
+     * 그룹당 문장 수 (규칙 변경 시 이 값만 수정하면 됨)
+     */
+    private static final int SENTENCES_PER_GROUP = 3;
+
+    /**
+     * 텍스트를 문장부호 기준으로 문장 단위로 분할합니다.
+     * 문장부호(. ! ?) 하나만 나와도 문장 종결로 간주합니다.
+     *
+     * @param text 분할할 텍스트
+     * @return 문장 단위로 분할된 리스트 (문장부호 포함)
+     */
+    public static List<String> splitIntoSentences(String text) {
+        if (text == null || text.trim().isEmpty()) {
+            return new ArrayList<>();
+        }
+
+        List<String> sentences = new ArrayList<>();
+        Matcher matcher = SENTENCE_END_PATTERN.matcher(text);
+
+        int lastIndex = 0;
+        while (matcher.find()) {
+            int endIndex = matcher.end();
+            String sentence = text.substring(lastIndex, endIndex);
+            // 첫 번째 문장만 앞뒤 공백 제거, 나머지는 앞 공백 유지
+            if (sentences.isEmpty()) {
+                sentence = sentence.trim();
+            } else {
+                // 앞 공백은 유지하되, 전체가 공백만 있는 경우는 제외
+                sentence = sentence.trim().isEmpty() ? sentence : sentence;
+            }
+            if (!sentence.trim().isEmpty()) {
+                sentences.add(sentence);
+            }
+            lastIndex = endIndex;
+        }
+
+        // 마지막 문장부호 이후의 텍스트 처리
+        if (lastIndex < text.length()) {
+            String remaining = text.substring(lastIndex);
+            if (!remaining.trim().isEmpty()) {
+                // 첫 번째 문장이 아니면 앞 공백 유지
+                if (!sentences.isEmpty()) {
+                    sentences.add(remaining);
+                } else {
+                    sentences.add(remaining.trim());
+                }
+            }
+        }
+
+        return sentences;
+    }
+
+    /**
+     * 텍스트를 문장 단위로 분할한 후, 세 문장씩 그룹화합니다.
+     * 예: 10문장인 경우 4개의 그룹으로 생성됩니다.
+     *
+     * @param text 그룹화할 텍스트
+     * @return 세 문장씩 그룹화된 텍스트 리스트
+     */
+    public static List<String> splitIntoGroups(String text) {
+        List<String> sentences = splitIntoSentences(text);
+
+        if (sentences.isEmpty()) {
+            return new ArrayList<>();
+        }
+
+        List<String> groups = new ArrayList<>();
+        StringBuilder currentGroup = new StringBuilder();
+
+        for (int i = 0; i < sentences.size(); i++) {
+            if (i > 0 && i % SENTENCES_PER_GROUP == 0) {
+                // 세 문장이 모였으면 그룹 완성
+                groups.add(currentGroup.toString());
+                currentGroup = new StringBuilder();
+            }
+
+            // sentences는 이미 적절한 공백을 포함하고 있으므로 그대로 추가
+            currentGroup.append(sentences.get(i));
+        }
+
+        // 마지막 남은 문장들 처리
+        if (currentGroup.length() > 0) {
+            groups.add(currentGroup.toString());
+        }
+
+        return groups;
+    }
+}