feat: 채팅개수에 따라 방 종료 및 Book 생성 (#4)

hajithon · Jul 27, 2024 · 20fb224 · 20fb224
1 parent 509a8e2
commit 20fb224
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 2 deletions.
diff --git a/src/main/java/mychat/controller/ChatController.java b/src/main/java/mychat/controller/ChatController.java
@@ -51,11 +51,17 @@ public void enterUser(@Payload Chat chat, SimpMessageHeaderAccessor headerAccess
     }
 
     @MessageMapping("/chat/sendMessage")
-    public void sendMessage(@Payload Chat chat) {
+    public String sendMessage(@Payload Chat chat) {
         log.info("CHAT {}", chat);
         chat.setId(UUID.randomUUID().toString());
-        template.convertAndSend("/sub/chat/room/" + chat.getRoomId(), chat);
+        String roomId = chat.getRoomId();
+        template.convertAndSend("/sub/chat/room/" + roomId, chat);
         chatService.saveMessage(chat);
+        if(chatRoomService.addChatCntAndReturn(roomId)){
+            chatService.makeBook(roomId);
+            return "fin";
+        }
+        return "";
     }
 
     @EventListener

diff --git a/src/main/java/mychat/domain/ChatRoom.java b/src/main/java/mychat/domain/ChatRoom.java
@@ -27,6 +27,8 @@ public class ChatRoom {
     @Builder.Default
     private boolean done=false;
     private String firstSentence;
+    @Builder.Default
+    private int chatCnt=0;
 
     // 채팅방 삭제시 방장이 비밀번호 입력해야함
 
@@ -61,4 +63,9 @@ public void setAsDone(){
     public void setFirstSentence(String firstSentence){
         this.firstSentence = firstSentence;
     }
+
+    public int addChatCnt(){
+        chatCnt++;
+        return chatCnt;
+    }
 }
diff --git a/src/main/java/mychat/service/ChatRoomService.java b/src/main/java/mychat/service/ChatRoomService.java
@@ -72,4 +72,16 @@ public void setFirstSentence(String roomId, String firstSentence){
                 .orElseThrow(() -> new CustomException(ErrorCode.NOT_FOUND_CHATROOM));
         room.setFirstSentence(firstSentence);
     }
+
+    public boolean addChatCntAndReturn(String roomId){
+        ChatRoom room = chatRoomRepository.findById(roomId)
+                .orElseThrow(() -> new CustomException(ErrorCode.NOT_FOUND_CHATROOM));
+        int cnt = room.addChatCnt();
+        chatRoomRepository.save(room);
+        if(cnt >= room.getUserCount()){
+            room.setAsDone();
+            return true;
+        }
+        return false;
+    }
 }
diff --git a/src/main/java/mychat/service/ChatService.java b/src/main/java/mychat/service/ChatService.java
@@ -60,6 +60,7 @@ public void makeBook(String roomId){
                 .createdAt(LocalDateTime.ofInstant(instance, ZoneId.of("Asia/Seoul")))
                 .build();
         bookRepository.save(book);
+        log.info("[Book 생성] {}", book.getId());
     }
 
     public void mongoInsert(){