[BE] FEAT: 채팅 서버 - WebSocket 붙이기 #174

src/backend/chat-server/src/main/java/kickzo/stomp_chat/config/JacksonConfig.java

@@ -0,0 +1,14 @@
+package kickzo.stomp_chat.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+@Configuration
+public class JacksonConfig {
+	@Bean
+	public ObjectMapper objectMapper() {
+		return new ObjectMapper();
+	}
+}

src/backend/chat-server/src/main/java/kickzo/stomp_chat/config/KafkaConsumerConfig.java

@@ -2,19 +2,13 @@
 
 import jakarta.annotation.PostConstruct;
 import org.apache.kafka.clients.consumer.ConsumerConfig;
-import org.apache.kafka.clients.consumer.ConsumerRecord;
 import org.apache.kafka.common.serialization.StringDeserializer;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.kafka.annotation.EnableKafka;
 import org.springframework.kafka.core.ConsumerFactory;
 import org.springframework.kafka.core.DefaultKafkaConsumerFactory;
-import org.springframework.kafka.listener.ConcurrentMessageListenerContainer;
-import org.springframework.kafka.listener.ContainerProperties;
-import org.springframework.kafka.listener.MessageListener;
-import org.springframework.messaging.simp.SimpMessagingTemplate;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -38,9 +32,6 @@ public void init() {
         groupId = "my-group-" + serverPort; // 서버 포트에 따라 groupId 설정
     }
 
-    @Autowired
-    private SimpMessagingTemplate messagingTemplate; // Injected by Spring
-
     @Bean
     public ConsumerFactory<String, String> consumerFactory() {
         // Kafka Consumer 설정
@@ -52,33 +43,33 @@ public ConsumerFactory<String, String> consumerFactory() {
         return new DefaultKafkaConsumerFactory<>(configProps);
     }
 
-    @Bean
-    public ConcurrentMessageListenerContainer<String, String> messageListenerContainer() {
-        MessageListener<String, String> messageListener = new MessageListener<String, String>() {
-            @Override
-            public void onMessage(ConsumerRecord<String, String> record) {
-                String message = record.value();
-                System.out.println("Received Kafka message: " + message);
-
-
-                // 메시지를 처리하는 로직을 여기에 추가
-                String roomId = "1";
-
-                if (roomId != null && !roomId.isEmpty()) {
-                    // WebSocket으로 메시지 전송
-                    messagingTemplate.convertAndSend("/topic/" + roomId, message);
-                    System.out.println("Message sent to WebSocket topic: /topic/" + roomId);
-                    System.out.println("Broadcasting message: " + message);
-                } else {
-                    System.err.println("Failed to send message: roomId is null or empty.");
-                }
-            }
-        };
-
-        ContainerProperties containerProps = new ContainerProperties("chatting");
-        containerProps.setMessageListener(messageListener);
-        containerProps.setGroupId(groupId); // groupId 설정
-
-        return new ConcurrentMessageListenerContainer<>(consumerFactory(), containerProps);
-    }
+    // @Bean
+    // public ConcurrentMessageListenerContainer<String, String> messageListenerContainer() {
+    //     MessageListener<String, String> messageListener = new MessageListener<String, String>() {
+    //         @Override
+    //         public void onMessage(ConsumerRecord<String, String> record) {
+    //             String message = record.value();
+    //             System.out.println("Received Kafka message: " + message);
+    //
+    //
+    //             // 메시지를 처리하는 로직을 여기에 추가
+    //             String roomId = "1";
+    //
+    //             if (roomId != null && !roomId.isEmpty()) {
+    //                 // WebSocket으로 메시지 전송
+    //                 messagingTemplate.convertAndSend("/topic/" + roomId, message);
+    //                 System.out.println("Message sent to WebSocket topic: /topic/" + roomId);
+    //                 System.out.println("Broadcasting message: " + message);
+    //             } else {
+    //                 System.err.println("Failed to send message: roomId is null or empty.");
+    //             }
+    //         }
+    //     };
+    //
+    //     ContainerProperties containerProps = new ContainerProperties("chatting");
+    //     containerProps.setMessageListener(messageListener);
+    //     containerProps.setGroupId(groupId); // groupId 설정
+    //
+    //     return new ConcurrentMessageListenerContainer<>(consumerFactory(), containerProps);
+    // }
 }

src/backend/chat-server/src/main/java/kickzo/stomp_chat/controller/WebSocketController.java

@@ -0,0 +1,79 @@
+package kickzo.stomp_chat.controller;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import kickzo.stomp_chat.enums.EventType;
+import kickzo.stomp_chat.service.WebSocketRoomService;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.simp.SimpMessageHeaderAccessor;
+import org.springframework.stereotype.Controller;
+
+@Slf4j
+@Controller
+@RequiredArgsConstructor
+public class WebSocketController {
+
+    private final ObjectMapper objectMapper;
+    private final WebSocketRoomService webSocketRoomService;
+
+    @MessageMapping("/connect")
+    public void connect(String payload, SimpMessageHeaderAccessor headerAccessor) {
+        try {
+            // payload를 UserConnectRequest 객체로 변환
+            UserConnectRequest request = objectMapper.readValue(payload, UserConnectRequest.class);
+            Long userId = request.userId();
+            // 세션에 사용자 정보 저장
+            headerAccessor.getSessionAttributes().put("userId", userId);
+
+            webSocketRoomService.sendConnection(userId, EventType.JOIN);
+            log.info("User {} connected and is now online.", userId);
+        } catch (Exception e) {
+            log.error("Error processing connect message", e);
+        }
+    }
+
+    // @MessageMapping("/joinRoom")
+    // public void joinRoom(String payload, SimpMessageHeaderAccessor headerAccessor) throws Exception {
+    //     try {
+    //         JoinRoomRequest request = objectMapper.readValue(payload, JoinRoomRequest.class);
+    //         Long userId = request.userId();
+    //         long roomId = request.roomId();
+    //         headerAccessor.getSessionAttributes().put("userId", userId);
+    //         headerAccessor.getSessionAttributes().put("roomId", roomId);
+    //
+    //         webSocketRoomService.handleRoomUserJoin(request.roomId(), userId);
+    //         messagingTemplate.convertAndSend("/topic/room/" + roomId, userId + " has joined the room.");
+    //     } catch (Exception e) {
+    //         log.error("Error processing joinRoom message", e);
+    //     }
+    // }
+
+    @MessageMapping("/send-message")
+    public void sendMessage(String payload, SimpMessageHeaderAccessor headerAccessor) {
+        try {
+            SendMessageRequest request = objectMapper.readValue(payload, SendMessageRequest.class);
+            long userId = (Long) headerAccessor.getSessionAttributes().get("userId");
+
+            webSocketRoomService.handleMessageSend(request.roomId(), userId, request.nickname(), request.role(),
+                request.profileImageUrl(), request.content(), request.message());
+        } catch (Exception e) {
+            log.error("Error processing sendMessage message", e);
+        }
+    }
+
+    @MessageMapping("/play-time")
+    public void playTime(String payload) throws Exception {
+        PlayTimeRequest request = objectMapper.readValue(payload, PlayTimeRequest.class);
+        log.info("Received roomId : {}, play time: {}", request.roomId(), request.playTime());
+
+        webSocketRoomService.sendPlayTime(request.roomId(), request.playTime(), request.playerState());
+    }
+
+    public record UserConnectRequest(long userId) {}
+    public record SendMessageRequest(long roomId, long userId, String nickname, int role, String profileImageUrl, String content, String message) {}
+
+    public record PlayTimeRequest(long roomId, long playTime, String playerState) {}
+}

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/RoomEvent.java

@@ -2,11 +2,13 @@
 
 import lombok.AllArgsConstructor;
 import lombok.Getter;
+import lombok.NoArgsConstructor;
 import lombok.Setter;
 
 @Getter
 @Setter
 @AllArgsConstructor
+@NoArgsConstructor
 public class RoomEvent {
 	private String eventType;  // 이벤트 구분
 	private Object data;       // 이벤트 데이터

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/playlist/PlayTime.java

@@ -0,0 +1,14 @@
+package kickzo.stomp_chat.dto.playlist;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Getter
+@AllArgsConstructor
+@NoArgsConstructor
+public class PlayTime {
+	private long roomId;
+	private long playTime;
+	private String playerState;
+}

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/PlaylistItem.java → src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/playlist/PlaylistItem.java

@@ -1,4 +1,4 @@
-package kickzo.stomp_chat.dto;
+package kickzo.stomp_chat.dto.playlist;
 
 import lombok.AllArgsConstructor;
 import lombok.Getter;

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/PlaylistUpdateEvent.java → src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/playlist/PlaylistUpdateEvent.java

@@ -1,4 +1,4 @@
-package kickzo.stomp_chat.dto;
+package kickzo.stomp_chat.dto.playlist;
 
 import java.util.List;
 
@@ -11,5 +11,5 @@
 @NoArgsConstructor
 public class PlaylistUpdateEvent {
 	private Long roomId;
-	private List<PlaylistItem> playlistJson;
+	private List<PlaylistItem> playlist;
 }

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/room/ChatMessage.java

@@ -0,0 +1,3 @@
+package kickzo.stomp_chat.dto.room;
+
+public record ChatMessage(long roomId, long userId, String nickname, int role, String profileImageUrl, String content, String message) {}

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/NewUserJoinEvent.java → src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/room/NewUserJoinEvent.java

@@ -1,4 +1,4 @@
-package kickzo.stomp_chat.dto;
+package kickzo.stomp_chat.dto.room;
 
 import lombok.AllArgsConstructor;
 import lombok.Getter;

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/RoleChangeEvent.java → src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/room/RoleChangeEvent.java

@@ -1,10 +1,12 @@
-package kickzo.stomp_chat.dto;
+package kickzo.stomp_chat.dto.room;
 
 import lombok.AllArgsConstructor;
 import lombok.Getter;
+import lombok.NoArgsConstructor;
 
 @Getter
 @AllArgsConstructor
+@NoArgsConstructor
 public class RoleChangeEvent {
 	private Long roomId;
 	private Long targetUserId;

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/RoomData.java → src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/room/RoomData.java

@@ -1,4 +1,4 @@
-package kickzo.stomp_chat.dto;
+package kickzo.stomp_chat.dto.room;
 
 import lombok.Getter;
 import lombok.NoArgsConstructor;

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/UserInfoDto.java → src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/room/UserInfoDto.java

@@ -1,10 +1,12 @@
-package kickzo.stomp_chat.dto;
+package kickzo.stomp_chat.dto.room;
 
 import lombok.AllArgsConstructor;
 import lombok.Data;
+import lombok.NoArgsConstructor;
 
 @Data
 @AllArgsConstructor
+@NoArgsConstructor
 public class UserInfoDto {
 	private Long userId;
 	private int role;

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/user/ConnectionEvent.java

@@ -0,0 +1,11 @@
+package kickzo.stomp_chat.dto.user;
+
+import java.time.Instant;
+
+import kickzo.stomp_chat.enums.EventType;
+
+public record ConnectionEvent(long userId, EventType eventType, String serverPort, long timestamp) {
+	public ConnectionEvent(long userId, EventType eventType, String serverPort) {
+		this(userId, eventType, serverPort, Instant.now().toEpochMilli());
+	}
+}

src/backend/chat-server/src/main/java/kickzo/stomp_chat/dto/user/FriendNotification.java

@@ -0,0 +1,13 @@
+package kickzo.stomp_chat.dto.user;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Getter
+@NoArgsConstructor
+@AllArgsConstructor
+public class FriendNotification {
+	private long userId;
+	private String status;
+}