chore(debug): 디버깅 및 테스트를 위한 핸들러 수정, 로그 추가

Author: yyubin

src/main/java/app/benchmark/WebSocketBenchmarkHandler.java

@@ -53,24 +53,19 @@ public void onError(WebSocketSession session, Throwable error) {
         error.printStackTrace();
     }
 
-    /**
-     * Echo: 받은 메시지를 그대로 반환
-     */
     @MessageMapping("/echo")
     public void handleEcho(WebSocketSession session, @Payload String message) throws IOException {
         totalMessagesReceived++;
         session.sendText(createResponse("/echo", message));
+        System.out.println("[WebSocket Benchmark] Echo: " + message);
         totalMessagesSent++;
     }
 
-    /**
-     * Broadcast: 모든 연결된 클라이언트에 메시지 전송
-     */
     @MessageMapping("/broadcast")
     public void handleBroadcast(WebSocketSession session, @Payload String message) throws IOException {
         totalMessagesReceived++;
         String response = createResponse("/broadcast", "From " + session.getId() + ": " + message);
-
+        System.out.println("[WebSocket Benchmark] Broadcast: " + message);
         for (WebSocketSession s : sessions.values()) {
             if (s.isOpen()) {
                 try {
@@ -83,9 +78,6 @@ public void handleBroadcast(WebSocketSession session, @Payload String message) t
         }
     }
 
-    /**
-     * Chat: 채팅 메시지 처리
-     */
     @MessageMapping("/chat")
     public void handleChat(WebSocketSession session, @Payload String message) throws IOException {
         totalMessagesReceived++;
@@ -96,6 +88,7 @@ public void handleChat(WebSocketSession session, @Payload String message) throws
         }
 
         String chatMessage = username + ": " + message;
+        System.out.println("[WebSocket Benchmark] Chat: " + chatMessage);
         String response = createResponse("/chat", chatMessage);
 
         // 채팅방의 모든 사용자에게 전송
@@ -111,21 +104,18 @@ public void handleChat(WebSocketSession session, @Payload String message) throws
         }
     }
 
-    /**
-     * Ping-Pong: 간단한 응답
-     */
     @MessageMapping("/ping")
-    public void handlePing(WebSocketSession session, String message) throws IOException {
+    public void handlePing(WebSocketSession session, @Payload String message) throws IOException {
         totalMessagesReceived++;
-        session.sendText(createResponse("/pong", "pong"));
+        // 실제 WebSocket Ping 프레임 전송 (브라우저가 자동으로 Pong 응답)
+        byte[] pingData = "ping".getBytes();
+        session.sendPing(pingData);
+        System.out.println("[WebSocket Benchmark] Sent Ping frame to client: " + session.getId());
         totalMessagesSent++;
     }
 
-    /**
-     * Stats: 통계 정보 반환
-     */
     @MessageMapping("/stats")
-    public void handleStats(WebSocketSession session, String message) throws IOException {
+    public void handleStats(WebSocketSession session, @Payload String message) throws IOException {
         totalMessagesReceived++;
         String stats = String.format(
             "연결: %d, 수신: %d, 송신: %d, 누적 연결: %d",
@@ -135,18 +125,12 @@ public void handleStats(WebSocketSession session, String message) throws IOExcep
         totalMessagesSent++;
     }
 
-    /**
-     * JSON 응답 메시지 생성
-     */
     private String createResponse(String destination, String payload) {
         // JSON 형식: {"destination": "...", "payload": "..."}
         return String.format("{\"destination\":\"%s\",\"payload\":\"%s\"}",
                            destination, escapeJson(payload));
     }
 
-    /**
-     * JSON 문자열 이스케이프
-     */
     private String escapeJson(String str) {
         if (str == null) return "";
         return str.replace("\\", "\\\\")
@@ -156,14 +140,12 @@ private String escapeJson(String str) {
                   .replace("\t", "\\t");
     }
 
-    // 통계 초기화 (테스트용)
     public static void resetStats() {
         totalMessagesReceived = 0;
         totalMessagesSent = 0;
         totalConnections = 0;
     }
 
-    // 통계 조회 (테스트용)
     public static String getStats() {
         return String.format(
             "Sessions: %d, Received: %d, Sent: %d, Total Connections: %d",

src/main/java/sprout/server/argument/WebSocketArgumentResolver.java

@@ -6,6 +6,5 @@
 
 public interface WebSocketArgumentResolver {
     boolean supports(Parameter parameter, InvocationContext context); // <- context 추가
-
     Object resolve(Parameter parameter, InvocationContext context) throws Exception; // <- context로 통합
 }

src/main/java/sprout/server/websocket/DefaultWebSocketSession.java

@@ -181,17 +181,29 @@ private void scheduleWrite(ByteBuffer buf) {
 
     private void processFrame(WebSocketFrame frame) throws Exception {
         if (WebSocketFrameDecoder.isCloseFrame(frame)) {
-            callOnCloseMethod(WebSocketFrameDecoder.getCloseCode(frame.getPayloadBytes()));
+            byte[] payload = frame.getPayloadBytes();
+            System.out.printf("[DEBUG] Close frame - Payload length: %d%n", payload.length);
+            if (payload.length >= 2) {
+                int code = ((payload[0] & 0xFF) << 8) | (payload[1] & 0xFF);
+                System.out.printf("[DEBUG] Close code: %d (0x%X)%n", code, code);
+            }
+            callOnCloseMethod(WebSocketFrameDecoder.getCloseCode(payload));
             return;
         } else if (WebSocketFrameDecoder.isPingFrame(frame)) {
-            System.out.println("Received Ping frame from client " + id);
-            sendPong(frame.getPayloadBytes());
+            byte[] payload = frame.getPayloadBytes();
+            System.out.printf("[DEBUG] Received Ping frame from client %s (payload: %d bytes)%n", id, payload.length);
+            sendPong(payload);
+            return;
         } else if (WebSocketFrameDecoder.isPongFrame(frame)) {
-            System.out.println("Received Pong frame from client " + id);
+            byte[] payload = frame.getPayloadBytes(); // 반드시 스트림을 소비해야 함!
+            System.out.printf("[DEBUG] Received Pong frame from client %s (payload: %d bytes)%n", id, payload.length);
+            return;
         } else if (WebSocketFrameDecoder.isDataFrame(frame)) {
             dispatchMessage(frame);
         } else {
-            System.err.println("Unknown or unsupported WebSocket opcode: " + frame.getOpcode());
+            System.err.printf("[ERROR] Unknown WebSocket opcode: 0x%X%n", frame.getOpcode());
+            // 알 수 없는 opcode의 경우에도 payload를 소비해야 함
+            frame.getPayloadBytes();
             callOnErrorMethod(new WebSocketException("Unknown WebSocket opcode: " + frame.getOpcode()));
         }
     }