Fix horim

src/main/java/com/cagong/receiptpowerserver/domain/chat/ChatMessage.java

@@ -0,0 +1,45 @@
+package com.cagong.receiptpowerserver.domain.chat;
+
+import com.cagong.receiptpowerserver.domain.member.Member;
+import jakarta.persistence.*;
+import lombok.AccessLevel;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import org.springframework.data.annotation.CreatedDate;
+import org.springframework.data.jpa.domain.support.AuditingEntityListener;
+
+import java.time.LocalDateTime;
+
+@Entity
+@Getter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@EntityListeners(AuditingEntityListener.class) // [!!] createdAt 자동 생성을 위해 필요
+public class ChatMessage {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "chat_room_id", nullable = false)
+    private ChatRoom chatRoom;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "sender_id", nullable = false)
+    private Member sender; // 보낸 사람
+
+    @Column(nullable = false, length = 1000) // 메시지 길이 제한 (필요시 조절)
+    private String message; // 메시지 내용
+
+    @CreatedDate
+    @Column(nullable = false, updatable = false)
+    private LocalDateTime createdAt; // 보낸 시간
+
+    @Builder
+    public ChatMessage(ChatRoom chatRoom, Member sender, String message) {
+        this.chatRoom = chatRoom;
+        this.sender = sender;
+        this.message = message;
+    }
+}

src/main/java/com/cagong/receiptpowerserver/domain/chat/ChatMessageController.java

@@ -1,38 +1,90 @@
-// chat/ChatMessageController.java
-
 package com.cagong.receiptpowerserver.domain.chat;
 
 import com.cagong.receiptpowerserver.domain.chat.dto.ChatMessageRequest;
+import com.cagong.receiptpowerserver.domain.chat.dto.ChatMessageResponse; // [!!] 응답 DTO 임포트
+// [!!] 리포지토리 의존성 모두 제거됨
 import lombok.RequiredArgsConstructor;
+import org.springframework.context.event.EventListener;
 import org.springframework.messaging.handler.annotation.MessageMapping;
 import org.springframework.messaging.handler.annotation.Payload;
 import org.springframework.messaging.simp.SimpMessageSendingOperations;
 import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
 import org.springframework.stereotype.Controller;
+import org.springframework.web.socket.messaging.SessionDisconnectEvent;
 
+/**
+ * [리팩토링] WebSocket 메시지 중계(Routing) 역할만 담당하는 컨트롤러.
+ * 모든 비즈니스 로직은 ChatMessageService에 위임합니다.
+ */
 @Controller
 @RequiredArgsConstructor
 public class ChatMessageController {
 
     private final SimpMessageSendingOperations messagingTemplate;
+    private final ChatMessageService chatMessageService; // [!!] 신규 서비스 주입
+
+    // [!!] 모든 Repository 의존성 제거됨
 
+    /**
+     * /pub/chat/message 경로로 오는 메시지 처리
+     */
     @MessageMapping("/chat/message")
     public void message(@Payload ChatMessageRequest message, StompHeaderAccessor headerAccessor) {
 
-        // [핵심] STOMP 세션 속성에서 StompHandler가 저장해 둔 사용자 이름을 직접 가져옵니다.
+        // 1. 세션에서 사용자 정보 가져오기 (인증 정보)
         String senderName = (String) headerAccessor.getSessionAttributes().get("username");
+        Long senderId = (Long) headerAccessor.getSessionAttributes().get("userId");
+        Long roomId = message.getRoomId();
 
-        // 만약 비정상적인 접근으로 senderName이 없다면, 메시지 처리를 중단합니다.
-        if (senderName == null) {
-            return;
+        if (senderName == null || senderId == null) {
+            return; // 비정상 접근 차단
         }
+        message.setSender(senderName); // (ENTER, QUIT 메시지용)
+
+        // 2. TALK 메시지인 경우, 서비스에 저장을 위임
+        if (ChatMessageRequest.MessageType.TALK.equals(message.getType())) {
 
-        message.setSender(senderName);
+            // 서비스 호출: DB에 메시지 저장 후, 브로드캐스팅용 DTO 반환
+            ChatMessageResponse responseDto = chatMessageService.saveMessage(message, senderId, senderName);
 
-        if (ChatMessageRequest.MessageType.ENTER.equals(message.getType())) {
-            message.setMessage(senderName + "님이 입장하셨습니다.");
+            // 클라이언트가 일관된 응답(ChatMessageResponse)을 받도록 DTO로 브로드캐스팅
+            messagingTemplate.convertAndSend("/sub/chat/room/" + roomId, responseDto);
+
+        } else {
+            // 3. ENTER, QUIT 메시지는 DB 저장 없이 브로드캐스팅
+            // (기존 ChatMessageRequest 포맷 그대로 전송)
+            if (ChatMessageRequest.MessageType.ENTER.equals(message.getType())) {
+                message.setMessage(senderName + "님이 입장하셨습니다.");
+                // 비정상 종료 처리를 위해 세션에 roomId 저장
+                headerAccessor.getSessionAttributes().put("roomId", roomId);
+            } else if (ChatMessageRequest.MessageType.QUIT.equals(message.getType())) {
+                message.setMessage(senderName + "님이 퇴장하셨습니다.");
+            }
+
+            messagingTemplate.convertAndSend("/sub/chat/room/" + roomId, message);
         }
 
-        messagingTemplate.convertAndSend("/sub/chat/room/" + message.getRoomId(), message);
+        // 4. 참여자 수 업데이트 이벤트 전송 (ENTER, QUIT 일 때만)
+        if (ChatMessageRequest.MessageType.ENTER.equals(message.getType()) ||
+                ChatMessageRequest.MessageType.QUIT.equals(message.getType())) {
+
+            // 서비스의 헬퍼 메서드 호출
+            chatMessageService.broadcastParticipantUpdate(roomId);
+        }
+    }
+
+    /**
+     * 비정상 종료(Disconnect) 이벤트 처리
+     * (로직을 ChatMessageService에 위임)
+     */
+    @EventListener
+    public void handleWebSocketDisconnectListener(SessionDisconnectEvent event) {
+        StompHeaderAccessor headerAccessor = StompHeaderAccessor.wrap(event.getMessage());
+
+        String username = (String) headerAccessor.getSessionAttributes().get("username");
+        Long roomId = (Long) headerAccessor.getSessionAttributes().get("roomId");
+
+        // 서비스 메서드 호출
+        chatMessageService.handleDisconnect(username, roomId);
     }
 }

src/main/java/com/cagong/receiptpowerserver/domain/chat/ChatMessageRepository.java

@@ -0,0 +1,11 @@
+package com.cagong.receiptpowerserver.domain.chat;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+import java.util.List;
+
+public interface ChatMessageRepository extends JpaRepository<ChatMessage, Long> {
+
+    // [!!] ✅ 2번 (로그 조회) 기능을 위한 핵심 쿼리 메서드
+    // 특정 채팅방의 메시지를 생성 시간(createdAt) 오름차순으로 모두 조회
+    List<ChatMessage> findByChatRoomIdOrderByCreatedAtAsc(Long roomId);
+}

src/main/java/com/cagong/receiptpowerserver/domain/chat/ChatMessageService.java

@@ -0,0 +1,147 @@
+package com.cagong.receiptpowerserver.domain.chat;
+
+import com.cagong.receiptpowerserver.domain.chat.dto.ChatMessageRequest;
+import com.cagong.receiptpowerserver.domain.chat.dto.ChatMessageResponse;
+import com.cagong.receiptpowerserver.domain.member.Member;
+import com.cagong.receiptpowerserver.domain.member.MemberRepository;
+import com.cagong.receiptpowerserver.exception.NotFoundException;
+import lombok.RequiredArgsConstructor;
+import org.springframework.messaging.simp.SimpMessageSendingOperations;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * [신규] 채팅 메시지 관련 모든 비즈니스 로직을 전담하는 서비스
+ * (WebSocket, HTTP API, 이벤트 처리 로직 포함)
+ */
+@Service
+@RequiredArgsConstructor
+@Transactional(readOnly = true)
+public class ChatMessageService {
+
+    // --- 의존성 주입 ---
+    private final ChatMessageRepository chatMessageRepository;
+    private final ChatRoomRepository chatRoomRepository;
+    private final MemberRepository memberRepository;
+    private final ChatParticipantRepository chatParticipantRepository;
+    private final SimpMessageSendingOperations messagingTemplate;
+
+    /**
+     * [WebSocket] 실시간 채팅 메시지 저장 및 브로드캐스팅
+     * (ChatMessageController의 TALK 타입 로직 이관)
+     *
+     * @param request    클라이언트 요청 DTO
+     * @param senderId   JWT 토큰에서 추출한 발신자 ID
+     * @param senderName JWT 토큰에서 추출한 발신자 이름
+     * @return 브로드캐스팅에 사용될 ChatMessageResponse DTO
+     */
+    @Transactional
+    public ChatMessageResponse saveMessage(ChatMessageRequest request, Long senderId, String senderName) {
+        // 1. 발신자(Member) 엔티티 조회
+        Member sender = memberRepository.findById(senderId)
+                .orElseThrow(() -> new NotFoundException("Member not found: " + senderId));
+
+        // 2. 채팅방(ChatRoom) 엔티티 조회
+        ChatRoom chatRoom = chatRoomRepository.findById(request.getRoomId())
+                .orElseThrow(() -> new NotFoundException("Chat room not found: " + request.getRoomId()));
+
+        // 3. 메시지 엔티티 생성
+        ChatMessage chatMessage = ChatMessage.builder()
+                .chatRoom(chatRoom)
+                .sender(sender)
+                .message(request.getMessage())
+                .build();
+
+        // 4. DB에 저장 (INSERT)
+        ChatMessage savedMessage = chatMessageRepository.save(chatMessage);
+
+        // 5. 엔티티를 DTO로 변환하여 반환
+        return toChatMessageResponse(savedMessage);
+    }
+
+    /**
+     * [HTTP API] 특정 채팅방의 과거 메시지 목록 조회
+     * (ChatRoomService에서 이관)
+     *
+     * @param roomId 채팅방 ID
+     * @return 메시지 DTO 리스트
+     */
+    public List<ChatMessageResponse> getMessages(Long roomId) {
+        if (!chatRoomRepository.existsById(roomId)) {
+            throw new NotFoundException("Chat room not found: " + roomId);
+        }
+
+        List<ChatMessage> messages = chatMessageRepository.findByChatRoomIdOrderByCreatedAtAsc(roomId);
+
+        return messages.stream()
+                .map(this::toChatMessageResponse)
+                .toList();
+    }
+
+    /**
+     * [WebSocket Event] 비정상 종료 처리
+     * (ChatMessageController의 handleWebSocketDisconnectListener 로직 이관)
+     *
+     * @param username 세션에 저장된 사용자 이름
+     * @param roomId   세션에 저장된 방 ID
+     */
+    public void handleDisconnect(String username, Long roomId) {
+        if (username != null && roomId != null) {
+            // 1. QUIT 메시지 생성
+            ChatMessageRequest quitMessage = new ChatMessageRequest();
+            quitMessage.setType(ChatMessageRequest.MessageType.QUIT);
+            quitMessage.setSender(username);
+            quitMessage.setRoomId(roomId);
+            quitMessage.setMessage(username + "님이 퇴장하셨습니다.");
+
+            // 2. 해당 채팅방에 퇴장 메시지 브로드캐스팅
+            messagingTemplate.convertAndSend("/sub/chat/room/" + roomId, quitMessage);
+
+            // 3. 비정상 종료 시에도 인원수 업데이트 이벤트 전송
+            broadcastParticipantUpdate(roomId);
+        }
+    }
+
+    /**
+     * [Helper] 참여자 수 업데이트 이벤트 브로드캐스팅
+     * (ChatMessageController의 sendParticipantUpdate 로직 이관)
+     *
+     * @param roomId 방 ID
+     */
+    public void broadcastParticipantUpdate(Long roomId) {
+        if (roomId == null) return;
+
+        // DB에서 현재 인원수를 조회
+        long currentParticipants = chatParticipantRepository.countByChatRoom_Id(roomId);
+
+        Map<String, Object> updateEvent = Map.of(
+                "type", "PARTICIPANT_UPDATE",
+                "roomId", roomId,
+                "currentParticipants", currentParticipants
+        );
+
+        messagingTemplate.convertAndSend("/sub/chat/room/" + roomId, updateEvent);
+    }
+
+    /**
+     * [Helper] ChatMessage 엔티티를 ChatMessageResponse DTO로 변환
+     * (ChatRoomService의 헬퍼 메서드 이관)
+     */
+    private ChatMessageResponse toChatMessageResponse(ChatMessage entity) {
+        Member sender = entity.getSender();
+        Long senderId = (sender != null) ? sender.getId() : null;
+        String senderName = (sender != null) ? sender.getUsername() : "알 수 없는 사용자"; // getUsername() 또는 getNickname()
+
+        return ChatMessageResponse.builder()
+                .id(entity.getId())
+                .roomId(entity.getChatRoom().getId())
+                .senderId(senderId)
+                .senderName(senderName)
+                .message(entity.getMessage())
+                .timestamp(entity.getCreatedAt())
+                .build();
+    }
+}