feat: websocket basic frame built

This commit is contained in:
merlin
2025-12-03 17:01:51 +08:00
parent b7f6d3477d
commit a1a23bec7a
5 changed files with 176 additions and 5 deletions

View File

@@ -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));
}
}
}
}