50 lines
1.6 KiB
Java
50 lines
1.6 KiB
Java
package xin.merlin.myplayerbackend.utils.websocket;
|
|
|
|
|
|
import org.springframework.stereotype.Component;
|
|
import org.springframework.web.socket.CloseStatus;
|
|
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));
|
|
}
|
|
}
|
|
}
|
|
|
|
public void closeAll() {
|
|
websocketSessions.forEach((uid, session) -> {
|
|
try {
|
|
session.close(CloseStatus.GOING_AWAY);
|
|
} catch (Exception ignored) {}
|
|
});
|
|
}
|
|
|
|
}
|