@ServerEndpoint("/ws/test/{username}")
@Component
public class TestEndpoint {
static final Logger loggger = LoggerFactory.getLogger(TestEndpoint.class);
private static int onlineCount = 0;
// 用来存放每个客户端对应的MyWebSocket对象。
private static ConcurrentHashMap<String, TestEndpoint> webSocketMap = new ConcurrentHashMap<>();
// 与某个客户端的连接会话,需要通过它来给客户端发送数据
private Session session;
// 接收username
private String username = "";
// 发送自定义消息
public static void sendInfo(String message, String username) {
loggger.info("发送消息到:" + username + ",报文:" + message);
if (StringUtils.isNotBlank(username) && webSocketMap.containsKey(username)) {
try {
webSocketMap.get(username).sendMessage(message);
} catch (Exception e) {
loggger.error(e.getMessage(), e);
}
} else {
loggger.error("用户" + username + ",不在线!");
}
}
public static synchronized int getOnlineCount() {
return onlineCount;
}
public static synchronized void addOnlineCount() {
TestEndpoint.onlineCount++;
}
public static synchronized void subOnlineCount() {
TestEndpoint.onlineCount--;
}
// 打开连接
@OnOpen
public void onOpen(Session session, @PathParam("username") String username) {
this.session = session;
this.username = username;
if (webSocketMap.containsKey(username)) {
webSocketMap.remove(username);
webSocketMap.put(username, this);
} else {
webSocketMap.put(username, this);
addOnlineCount();
}
loggger.info("用户连接:" + username + ",当前在线人数为:" + getOnlineCount());
try {
sendMessage("连接成功");
} catch (IOException e) {
loggger.error("用户:" + username + ",网络异常!!!!!!");
}
}
// 关闭连接
@OnClose
public void onClose() {
if (webSocketMap.containsKey(username)) {
webSocketMap.remove(username);
subOnlineCount();
}
loggger.info("用户退出:" + username + ",当前在线人数为:" + getOnlineCount());
}
// 接收消息
@OnMessage
public void onMessage(String message, Session session) {
loggger.info("用户消息:" + username + ",报文:" + message);
}
@OnError
public void onError(Session session, Throwable error) {
loggger.error("用户错误:" + this.username + ",原因:" + error.getMessage());
error.printStackTrace();
}
// 推送消息
public void sendMessage(String message) throws IOException {
this.session.getBasicRemote().sendText(message);
}
}