Feat: WebSocketConfig에 세션 관리 연동 로직 추가

Author: jueunk617

src/main/java/com/back/global/config/WebSocketConfig.java

@@ -2,6 +2,7 @@
 
 import com.back.global.security.CustomUserDetails;
 import com.back.global.security.JwtTokenProvider;
+import com.back.global.websocket.service.WebSocketSessionManager;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.context.annotation.Configuration;
@@ -28,6 +29,7 @@
 public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
 
     private final JwtTokenProvider jwtTokenProvider;
+    private final WebSocketSessionManager sessionManager;
 
     /**
      * 메시지 브로커 설정
@@ -55,11 +57,11 @@ public void registerStompEndpoints(StompEndpointRegistry registry) {
 
     /**
      * WebSocket 메시지 채널 설정
-     * JWT 인증 인터셉터 등록
+     * JWT 인증 인터셉터 및 세션 관리 로직 등록
      */
     @Override
     public void configureClientInboundChannel(ChannelRegistration registration) {
-        // JWT 인증 인터셉터 등록
+        // JWT 인증 + 세션 관리 인터셉터 등록
         registration.interceptors(new ChannelInterceptor() {
             @Override
             public Message<?> preSend(Message<?> message, MessageChannel channel) {
@@ -74,9 +76,19 @@ public Message<?> preSend(Message<?> message, MessageChannel channel) {
                         authenticateUser(accessor);
                     }
 
-                    // SEND 시점에서도 인증 확인 (추가 보안)
+                    // SEND 시점에서 인증 확인 및 활동 시간 업데이트
                     else if (StompCommand.SEND.equals(accessor.getCommand())) {
-                        validateAuthentication(accessor);
+                        validateAuthenticationAndUpdateActivity(accessor);
+                    }
+
+                    // SUBSCRIBE 시점에서 방 입장 처리
+                    else if (StompCommand.SUBSCRIBE.equals(accessor.getCommand())) {
+                        handleRoomSubscription(accessor);
+                    }
+
+                    // UNSUBSCRIBE 시점에서 방 퇴장 처리
+                    else if (StompCommand.UNSUBSCRIBE.equals(accessor.getCommand())) {
+                        handleRoomUnsubscription(accessor);
                     }
                 }
 
@@ -111,24 +123,109 @@ private void authenticateUser(StompHeaderAccessor accessor) {
             // 세션에 사용자 정보 저장
             accessor.setUser(authentication);
 
+            log.info("WebSocket 인증 성공 - 사용자: {} (ID: {}), 세션: {}",
+                    userDetails.getUsername(), userDetails.getUserId(), accessor.getSessionId());
+
         } catch (Exception e) {
+            log.error("WebSocket 인증 실패: {}", e.getMessage());
             throw new RuntimeException("WebSocket 인증에 실패했습니다: " + e.getMessage());
         }
     }
 
     /**
-     * 메시지 전송 시 인증 상태 확인
+     * 메시지 전송 시 인증 상태 확인 및 활동 시간 업데이트
      */
-    private void validateAuthentication(StompHeaderAccessor accessor) {
+    private void validateAuthenticationAndUpdateActivity(StompHeaderAccessor accessor) {
         if (accessor.getUser() == null) {
             throw new RuntimeException("인증이 필요합니다");
         }
 
-        // 인증된 사용자 정보 로깅
+        // 인증된 사용자 정보 추출 및 활동 시간 업데이트
         Authentication auth = (Authentication) accessor.getUser();
         if (auth.getPrincipal() instanceof CustomUserDetails userDetails) {
+            Long userId = userDetails.getUserId();
+
+            // 사용자 활동 시간 업데이트
+            sessionManager.updateLastActivity(userId);
+
             log.debug("인증된 사용자 메시지 전송 - 사용자: {} (ID: {}), 목적지: {}",
-                    userDetails.getUsername(), userDetails.getUserId(), accessor.getDestination());
+                    userDetails.getUsername(), userId, accessor.getDestination());
+        }
+    }
+
+    /**
+     * 방 채팅 구독 시 방 입장 처리
+     */
+    private void handleRoomSubscription(StompHeaderAccessor accessor) {
+        try {
+            String destination = accessor.getDestination();
+
+            // 방 채팅 구독인지 확인: /topic/rooms/{roomId}/chat
+            if (destination != null && destination.matches("/topic/rooms/\\d+/chat")) {
+                Long roomId = extractRoomIdFromDestination(destination);
+
+                if (roomId != null && accessor.getUser() != null) {
+                    Authentication auth = (Authentication) accessor.getUser();
+                    if (auth.getPrincipal() instanceof CustomUserDetails userDetails) {
+                        Long userId = userDetails.getUserId();
+
+                        // 방 입장 처리
+                        sessionManager.joinRoom(userId, roomId);
+
+                        log.info("방 입장 처리 완료 - 사용자: {} (ID: {}), 방: {}",
+                                userDetails.getUsername(), userId, roomId);
+                    }
+                }
+            }
+        } catch (Exception e) {
+            log.error("방 구독 처리 중 오류 발생: {}", e.getMessage(), e);
+        }
+    }
+
+    /**
+     * 방 채팅 구독 해제 시 방 퇴장 처리
+     */
+    private void handleRoomUnsubscription(StompHeaderAccessor accessor) {
+        try {
+            String destination = accessor.getDestination();
+
+            // 방 채팅 구독 해제인지 확인: /topic/rooms/{roomId}/chat
+            if (destination != null && destination.matches("/topic/rooms/\\d+/chat")) {
+                Long roomId = extractRoomIdFromDestination(destination);
+
+                if (roomId != null && accessor.getUser() != null) {
+                    Authentication auth = (Authentication) accessor.getUser();
+                    if (auth.getPrincipal() instanceof CustomUserDetails userDetails) {
+                        Long userId = userDetails.getUserId();
+
+                        // 방 퇴장 처리
+                        sessionManager.leaveRoom(userId, roomId);
+
+                        log.info("방 퇴장 처리 완료 - 사용자: {} (ID: {}), 방: {}",
+                                userDetails.getUsername(), userId, roomId);
+                    }
+                }
+            }
+        } catch (Exception e) {
+            log.error("방 구독 해제 처리 중 오류 발생: {}", e.getMessage(), e);
+        }
+    }
+
+    /**
+     * destination에서 방 ID 추출
+     * ex) "/topic/rooms/123/chat" -> 123
+     */
+    private Long extractRoomIdFromDestination(String destination) {
+        try {
+            if (destination == null) return null;
+
+            String[] parts = destination.split("/");
+            if (parts.length >= 4 && "rooms".equals(parts[2])) {
+                return Long.parseLong(parts[3]);
+            }
+        } catch (NumberFormatException e) {
+            log.warn("방 ID 추출 실패 - destination: {}, 에러: {}", destination, e.getMessage());
         }
+        return null;
     }
-}
+}