48 lines
1.4 KiB
Java
48 lines
1.4 KiB
Java
package com.sky.websocket;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import javax.websocket.OnMessage;
|
|
import javax.websocket.OnOpen;
|
|
import javax.websocket.Session;
|
|
import javax.websocket.server.PathParam;
|
|
import javax.websocket.server.ServerEndpoint;
|
|
import java.util.Collection;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
@Component
|
|
@ServerEndpoint("/ws/{sid}")
|
|
@Slf4j
|
|
public class WebSocketServer {
|
|
// 存放会话对象
|
|
private static final Map<String, Session> sessionMap = new HashMap<>();
|
|
|
|
// 连接建立成功调用的方法
|
|
@OnOpen
|
|
public void onOpen(Session session, @PathParam("sid") String sid) {
|
|
log.info("客户端：{}--->建立连接", sid);
|
|
sessionMap.put(sid, session);
|
|
}
|
|
|
|
// 收到客户端消息后调用的方法
|
|
@OnMessage
|
|
public void onMessage(String message, @PathParam("sid") String sid) {
|
|
log.info("收到来自客户端：{}；信息为：{}", sid, message);
|
|
}
|
|
|
|
// 群发
|
|
public void sendToAllClient(String message) {
|
|
Collection<Session> sessions = sessionMap.values();
|
|
for (Session session : sessions) {
|
|
try {
|
|
// 服务器向客户端发送消息
|
|
session.getBasicRemote().sendText(message);
|
|
} catch (Exception exception) {
|
|
exception.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
}
|