✨ feat: 커스텀 HandshakeInterceptor 추가 (#98)

* ♻️ refactor: 웹소켓 핸드쉐이크 인터셉터 변경

* ✨ feat: 웹소켓 로그에 사용자이름 추가

* chore: Java 스타일 수정

---------

Co-authored-by: github-actions <>

Author: LimKangHyun

backend/src/main/java/io/f1/backend/global/config/CustomHandshakeInterceptor.java

@@ -0,0 +1,46 @@
+package io.f1.backend.global.config;
+
+import lombok.extern.slf4j.Slf4j;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.http.server.ServerHttpRequest;
+import org.springframework.http.server.ServerHttpResponse;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.WebSocketHandler;
+import org.springframework.web.socket.server.HandshakeInterceptor;
+
+import java.util.Map;
+
+@Slf4j
+@Component
+public class CustomHandshakeInterceptor implements HandshakeInterceptor {
+
+    @Override
+    public boolean beforeHandshake(
+            ServerHttpRequest request,
+            ServerHttpResponse response,
+            WebSocketHandler wsHandler,
+            Map<String, Object> attributes)
+            throws Exception {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+        if (authentication == null || !authentication.isAuthenticated()) {
+            response.setStatusCode(HttpStatus.UNAUTHORIZED); // 서버 로그에만 적용되는 StatusCode
+            return false;
+        }
+
+        attributes.put("auth", authentication);
+        return true;
+    }
+
+    @Override
+    public void afterHandshake(
+            ServerHttpRequest request,
+            ServerHttpResponse response,
+            WebSocketHandler wsHandler,
+            Exception exception) {
+        // TODO : 연결 이후, 사용자 웹소켓 세션 로그 및 IP 등 추적 및 메트릭 수집 로직 추가
+
+    }
+}

backend/src/main/java/io/f1/backend/global/config/StompChannelInterceptor.java

@@ -27,16 +27,21 @@ public Message<?> preSend(Message<?> message, MessageChannel channel) {
             throw new IllegalArgumentException("Stomp command required");
         }
 
+        String username = "알수없는 사용자";
+        if (accessor.getUser() != null) {
+            username = accessor.getUser().getName();
+        }
+
         if (command.equals(StompCommand.CONNECT)) {
-            log.info("CONNECT : 세션 연결 - sessionId = {}", sessionId);
+            log.info("user : {} | CONNECT : 세션 연결 - sessionId = {}", username, sessionId);
         } else if (command.equals(StompCommand.SUBSCRIBE)) {
             if (destination != null && sessionId != null) {
-                log.info("SUBSCRIBE : 구독 시작 destination = {}", destination);
+                log.info("user : {} | SUBSCRIBE : 구독 시작 destination = {}", username, destination);
             }
         } else if (command.equals(StompCommand.SEND)) {
-            log.info("SEND : 요청 destination = {}", destination);
+            log.info("user : {} | SEND : 요청 destination = {}", username, destination);
         } else if (command.equals(StompCommand.DISCONNECT)) {
-            log.info("DISCONNECT : 연결 해제 sessionId = {}", sessionId);
+            log.info("user : {} | DISCONNECT : 연결 해제 sessionId = {}", username, sessionId);
         }
 
         return message;

backend/src/main/java/io/f1/backend/global/config/WebSocketConfig.java

@@ -8,19 +8,19 @@
 import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
-import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
 
 @Configuration
 @RequiredArgsConstructor
 @EnableWebSocketMessageBroker
 public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
 
     private final StompChannelInterceptor stompChannelInterceptor;
+    private final CustomHandshakeInterceptor customHandshakeInterceptor;
 
     @Override
     public void registerStompEndpoints(StompEndpointRegistry registry) {
         registry.addEndpoint("/ws/game-room")
-                .addInterceptors(new HttpSessionHandshakeInterceptor())
+                .addInterceptors(customHandshakeInterceptor)
                 .setAllowedOriginPatterns("*");
     }