feat: websocket basic frame built

src/main/java/xin/merlin/myplayerbackend/utils/websocket/CustomWebSocketHandler.java

@@ -0,0 +1,56 @@
+package xin.merlin.myplayerbackend.utils.websocket;
+
+import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONObject;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.CloseStatus;
+import org.springframework.web.socket.TextMessage;
+import org.springframework.web.socket.WebSocketSession;
+import org.springframework.web.socket.handler.TextWebSocketHandler;
+
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class CustomWebSocketHandler extends TextWebSocketHandler {
+
+    private final WebSocketSessionManager webSocketSessionManager;
+
+    @Override
+    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
+        Integer userId = (Integer) session.getAttributes().get("userId");
+        webSocketSessionManager.addSession(userId, session);
+        log.info("用户 {} 已连接", userId);
+    }
+
+    @Override
+    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
+        String payload = message.getPayload();
+        JSONObject msg = JSON.parseObject(payload);
+        String type = msg.getString("type");
+        switch (type) {
+            case "chat":
+                Integer toUser = msg.getInteger("toUser");
+                webSocketSessionManager.sendToUser(toUser, payload);
+                break;
+
+            case "broadcast":
+                webSocketSessionManager.broadcast(payload);
+                break;
+
+            case "signal":
+                webSocketSessionManager.sendToUser(msg.getInteger("toUser"), payload);
+                break;
+
+        }
+    }
+
+    @Override
+    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
+        Integer userId = (Integer) session.getAttributes().get("userId");
+        webSocketSessionManager.removeSession(userId);
+        log.info("用户 {} 已断开", userId);
+    }
+}

src/main/java/xin/merlin/myplayerbackend/utils/websocket/WebSocketSessionManager.java

@@ -0,0 +1,41 @@
+package xin.merlin.myplayerbackend.utils.websocket;
+
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.TextMessage;
+import org.springframework.web.socket.WebSocketSession;
+
+import java.io.IOException;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+@Component
+public class WebSocketSessionManager {
+
+    private static final Map<Integer, WebSocketSession> websocketSessions = new ConcurrentHashMap<>();
+
+    public void addSession(Integer id, WebSocketSession session) {websocketSessions.put(id, session);}
+
+    public WebSocketSession getSession(Integer id) {return websocketSessions.get(id);}
+
+    public void removeSession(Integer id) {websocketSessions.remove(id);}
+
+    public Map<Integer, WebSocketSession> getSessions() {return websocketSessions;}
+
+    public void sendToUser(Integer userId, String message) throws IOException {
+        WebSocketSession session = websocketSessions.get(userId);
+        if (session != null && session.isOpen()) {
+            session.sendMessage(new TextMessage(message));
+        }
+    }
+
+    public void broadcast(String message) throws IOException {
+        for (WebSocketSession session : websocketSessions.values()) {
+            if (session.isOpen()) {
+                session.sendMessage(new TextMessage(message));
+            }
+        }
+    }
+
+
+}