✨Feat: 쪽지 알람 기능 구현

imjuyongp · imjuyongp · commit 38d42335a1e3 · 2026-02-09T16:52:17.000+09:00
diff --git a/src/main/java/com/be/sportizebe/domain/chat/websocket/handler/ChatStompController.java b/src/main/java/com/be/sportizebe/domain/chat/websocket/handler/ChatStompController.java
@@ -5,8 +5,13 @@
 import com.be.sportizebe.domain.chat.dto.request.ChatSendRequest;
 import com.be.sportizebe.domain.chat.entity.ChatMessage;
 import com.be.sportizebe.domain.chat.entity.ChatRoom;
+import com.be.sportizebe.domain.chat.entity.ChatRoomMember;
+import com.be.sportizebe.domain.chat.repository.ChatRoomMemberRepository;
 import com.be.sportizebe.domain.chat.service.ChatMessageService;
 import com.be.sportizebe.domain.chat.service.ChatRoomService;
+import com.be.sportizebe.domain.notification.service.NotificationService;
+import com.be.sportizebe.domain.user.entity.User;
+import com.be.sportizebe.domain.user.repository.UserRepository;
 import lombok.RequiredArgsConstructor;
 import org.springframework.messaging.handler.annotation.MessageMapping;
 import org.springframework.messaging.simp.SimpMessageHeaderAccessor;
@@ -17,9 +22,12 @@
 @RequiredArgsConstructor
 public class ChatStompController {
     private final ChatMessageService chatMessageService;
-    private final ChatRoomService chatRoomService;   // ✅ 추가
+    private final ChatRoomService chatRoomService;
     private final SimpMessagingTemplate messagingTemplate;
     private final ChatSessionRegistry registry;
+    private final NotificationService notificationService;
+    private final ChatRoomMemberRepository chatRoomMemberRepository;
+    private final UserRepository userRepository;
 
     /**
      * 클라이언트가 SEND로 보낸 메세지를 받는 주소가 여기
@@ -45,6 +53,24 @@ public void send(ChatSendRequest req){
             "/sub/chat/rooms/" + req.getRoomId(), // 채팅방 단위 브로드캐스트
             ChatMessageResponse.from(saved)
         );
+
+        // 1:1 채팅방(쪽지)인 경우 상대방에게 알림 전송
+        if (room.getChatRoomType() == ChatRoom.ChatRoomType.NOTE) {
+            sendNoteNotification(saved, req.getSenderUserId());
+        }
+    }
+
+    /**
+     * 쪽지 알림 전송 (발신자 제외한 상대방에게)
+     */
+    private void sendNoteNotification(ChatMessage chatMessage, Long senderUserId) {
+        chatRoomMemberRepository.findAllByRoom_IdAndLeftAtIsNull(chatMessage.getRoom().getId())
+            .stream()
+            .map(ChatRoomMember::getUserId)
+            .filter(userId -> !userId.equals(senderUserId))
+            .findFirst() // 첫번째 사람만 선택
+            .flatMap(userRepository::findById)
+            .ifPresent(receiver -> notificationService.createNoteNotification(chatMessage, receiver));
     }
 
     @MessageMapping("/chat.join")
diff --git a/src/main/java/com/be/sportizebe/domain/notification/dto/response/NotificationResponse.java b/src/main/java/com/be/sportizebe/domain/notification/dto/response/NotificationResponse.java
@@ -18,12 +18,15 @@ public record NotificationResponse(
     @Schema(description = "게시글 ID") Long postId,
     @Schema(description = "게시글 제목") String postTitle,
     @Schema(description = "댓글 작성자 닉네임") String commenterNickname,
-    @Schema(description = "관련 대상 ID (채팅 등)") Long targetId,
+    @Schema(description = "채팅방 ID") Long chatRoomId,
+    @Schema(description = "쪽지 발신자 닉네임") String senderNickname,
+    @Schema(description = "관련 대상 ID") Long targetId,
     @Schema(description = "알림 생성 일시") LocalDateTime createdAt
 ) {
   public static NotificationResponse from(Notification notification) {
     var joinRequest = notification.getJoinClubRequest();
     var comment = notification.getComment();
+    var chatMessage = notification.getChatMessage();
 
     return NotificationResponse.builder()
         .id(notification.getId())
@@ -35,6 +38,8 @@ public static NotificationResponse from(Notification notification) {
         .postId(comment != null ? comment.getPost().getId() : null)
         .postTitle(comment != null ? comment.getPost().getTitle() : null)
         .commenterNickname(comment != null ? comment.getUser().getNickname() : null)
+        .chatRoomId(chatMessage != null ? chatMessage.getRoom().getId() : null)
+        .senderNickname(chatMessage != null ? chatMessage.getSenderNickname() : null)
         .targetId(notification.getTargetId())
         .createdAt(notification.getCreatedAt())
         .build();
diff --git a/src/main/java/com/be/sportizebe/domain/notification/entity/Notification.java b/src/main/java/com/be/sportizebe/domain/notification/entity/Notification.java
@@ -1,5 +1,6 @@
 package com.be.sportizebe.domain.notification.entity;
 
+import com.be.sportizebe.domain.chat.entity.ChatMessage;
 import com.be.sportizebe.domain.comment.entity.Comment;
 import com.be.sportizebe.domain.user.entity.User;
 import com.be.sportizebe.global.common.BaseTimeEntity;
@@ -26,7 +27,8 @@ public enum NotificationType {
     JOIN_REJECTED,  // 가입 거절 (신청자에게)
     CHAT,           // 새 채팅 메시지
     COMMENT,        // 새 댓글 (게시글 작성자에게)
-    REPLY           // 대댓글 (부모 댓글 작성자에게)
+    REPLY,           // 대댓글 (부모 댓글 작성자에게)
+    NOTE            // 쪽지 (수신자에게)
   }
 
   @Id
@@ -55,8 +57,12 @@ public enum NotificationType {
   @JoinColumn(name = "comment_id")
   private Comment comment;
 
+  // 쪽지 알림용
+  @ManyToOne(fetch = FetchType.LAZY)
+  @JoinColumn(name = "chat_message_id")
+  private ChatMessage chatMessage;
+
   // 기타 알림용 - targetId로 다형성 처리
-  // CHAT: chatRoomId 등
   private Long targetId;
 
   public void markAsRead() {
diff --git a/src/main/java/com/be/sportizebe/domain/notification/service/NotificationService.java b/src/main/java/com/be/sportizebe/domain/notification/service/NotificationService.java
@@ -1,5 +1,6 @@
 package com.be.sportizebe.domain.notification.service;
 
+import com.be.sportizebe.domain.chat.entity.ChatMessage;
 import com.be.sportizebe.domain.comment.entity.Comment;
 import com.be.sportizebe.domain.notification.dto.response.NotificationResponse;
 import com.be.sportizebe.domain.notification.entity.JoinClubRequest;
@@ -25,6 +26,9 @@ public interface NotificationService {
   // 대댓글 알림 생성 및 웹소켓 전송 (부모 댓글 작성자에게)
   void createReplyNotification(Comment reply);
 
+  // 쪽지 알림 생성 및 웹소켓 전송 (수신자에게)
+  void createNoteNotification(ChatMessage chatMessage, User receiver);
+
   // 사용자의 모든 알림 조회
   List<NotificationResponse> getNotifications(User user);
 
diff --git a/src/main/java/com/be/sportizebe/domain/notification/service/NotificationServiceImpl.java b/src/main/java/com/be/sportizebe/domain/notification/service/NotificationServiceImpl.java
@@ -1,5 +1,6 @@
 package com.be.sportizebe.domain.notification.service;
 
+import com.be.sportizebe.domain.chat.entity.ChatMessage;
 import com.be.sportizebe.domain.comment.entity.Comment;
 import com.be.sportizebe.domain.notification.dto.response.NotificationResponse;
 import com.be.sportizebe.domain.notification.entity.JoinClubRequest;
@@ -145,6 +146,24 @@ public void createReplyNotification(Comment reply) {
     sendNotificationToUser(parentAuthor.getId(), notification);
   }
 
+  @Override
+  @Transactional
+  public void createNoteNotification(ChatMessage chatMessage, User receiver) {
+    // 자신에게 보낸 경우 알림 생성하지 않음
+    if (receiver.getId() == chatMessage.getSenderUserId()) {
+      return;
+    }
+
+    Notification notification = Notification.builder()
+        .receiver(receiver)
+        .type(Notification.NotificationType.NOTE)
+        .chatMessage(chatMessage)
+        .build();
+
+    notificationRepository.save(notification);
+    sendNotificationToUser(receiver.getId(), notification);
+  }
+
   /**
    * 웹소켓으로 알림 전송
    */
