refacor: NotificationDTO 변경

Yanabada · Jan 24, 2024 · 81fe120 · 81fe120
1 parent 6721dbf
commit 81fe120
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/src/main/java/kr/co/fastcampus/yanabada/domain/notification/dto/ChatNotificationDto.java b/src/main/java/kr/co/fastcampus/yanabada/domain/notification/dto/ChatNotificationDto.java
@@ -13,10 +13,7 @@
 public record ChatNotificationDto(
     Member sender,
     Member receiver,
-    String accommodationName,
-    String chatRoomCode,
-    String senderNickname,
-    String image
+    String accommodationName
 ) {
 
     public static ChatNotificationDto from(
@@ -30,15 +27,13 @@ public static ChatNotificationDto from(
             .accommodationName(
                 chatRoom.getProduct().getOrder().getRoom().getAccommodation().getName()
             )
-            .image(sender.getImage())
             .build();
     }
 
     public String convertMapToJsonStr(ObjectMapper objectMapper) {
         Map<String, String> contentMap = new HashMap<>();
-        contentMap.put("senderNickname", senderNickname);
+        contentMap.put("senderNickname", sender().getNickName());
         contentMap.put("accommodationName", accommodationName);
-        contentMap.put("chatRoomCode", chatRoomCode);
         try {
             return objectMapper.writeValueAsString(contentMap);
         } catch (JsonProcessingException e) {

diff --git a/src/main/java/kr/co/fastcampus/yanabada/domain/notification/service/NotificationService.java b/src/main/java/kr/co/fastcampus/yanabada/domain/notification/service/NotificationService.java
@@ -85,7 +85,7 @@ public void sendChatCreated(ChatNotificationDto chatDto, String content) {
             .receiver(chatDto.receiver())
             .notificationType(CHAT)
             .content(chatDto.convertMapToJsonStr(objectMapper))
-            .image(chatDto.image())
+            .image(chatDto.sender().getImage())
             .build();
 
         notificationHistoryRepository.save(notificationHistory);