[COZY-225] fix : 반환 값 persona 추가 (#88)

cozy-mate · Aug 29, 2024 · a6135cc · a6135cc
1 parent dfe6ce5
commit a6135cc
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/main/java/com/cozymate/cozymate_server/domain/chatroom/converter/ChatRoomConverter.java b/src/main/java/com/cozymate/cozymate_server/domain/chatroom/converter/ChatRoomConverter.java
@@ -13,11 +13,13 @@ public static ChatRoom toEntity(Member sender, Member recipient) {
             .build();
     }
 
-    public static ChatRoomResponseDto toResponseDto(String nickName, String content, Long chatRoomId) {
+    public static ChatRoomResponseDto toResponseDto(String nickName, String content,
+        Long chatRoomId, Integer persona) {
         return ChatRoomResponseDto.builder()
             .nickName(nickName)
             .lastContent(content)
             .chatRoomId(chatRoomId)
+            .persona(persona)
             .build();
     }
 }
diff --git a/src/main/java/com/cozymate/cozymate_server/domain/chatroom/dto/ChatRoomResponseDto.java b/src/main/java/com/cozymate/cozymate_server/domain/chatroom/dto/ChatRoomResponseDto.java
@@ -6,6 +6,7 @@
 @Getter
 @Builder
 public class ChatRoomResponseDto {
+    private Integer persona;
     private String nickName;
     private String lastContent;
     private Long chatRoomId;

diff --git a/src/main/java/com/cozymate/cozymate_server/domain/chatroom/service/ChatRoomQueryService.java b/src/main/java/com/cozymate/cozymate_server/domain/chatroom/service/ChatRoomQueryService.java
@@ -64,6 +64,9 @@ private ChatRoomResponseDto toChatRoomResponseDto(ChatRoom chatRoom, Member memb
         return ChatRoomConverter.toResponseDto(
             member.getNickname().equals(chatRoom.getMemberA().getNickname()) ?
                 chatRoom.getMemberB().getNickname() : chatRoom.getMemberA().getNickname(),
-            chat.getContent(), chatRoom.getId());
+            chat.getContent(),
+            chatRoom.getId(),
+            member.getNickname().equals(chatRoom.getMemberA().getNickname()) ?
+                chatRoom.getMemberB().getPersona() : chatRoom.getMemberA().getPersona());
     }
 }