SprinBoot——SpringBoot项目WebSocket推送

webSocketSet.add(this); //加入set中

addOnlineCount(); //在线数加1

System.out.println(“有新连接加入！当前在线人数为” + getOnlineCount());

try {

sendMessage(“新人上限”);

} catch (IOException e) {

System.out.println(“IO异常”);

}

}

/**

• 连接关闭调用的方法

*/

@OnClose

public void onClose() {

webSocketSet.remove(this); //从set中删除

subOnlineCount(); //在线数减1

System.out.println(“有一连接关闭！当前在线人数为” + getOnlineCount());

}

/**

• 收到客户端消息后调用的方法

• @param message 客户端发送过来的消息

*/

@OnMessage

public void onMessage(String message, Session session) {

System.out.println(“来自客户端的消息:” + message);

//群发消息

for (MyWebSocket item : webSocketSet) {

try {

item.sendMessage(message);

} catch (IOException e) {

e.printStackTrace();

}

}

}

/**

• 发生错误时调用

*/

@OnError

public void onError(Session session, Throwable error) {

System.out.println(“发生错误”);

error.printStackTrace();

}

public void sendMessage(String message) throws IOException {

this.session.getBasicRemote().sendText(message);

}

/**

• 群发自定义消息

*/

public static void sendInfo(String message) throws IOException {

for (MyWebSocket item : webSocketSet) {

try {

item.sendMessage(message);

} catch (IOException e) {

continue;

}

}

}

public static synchronized int getOnlineCount() {

return onlineCount;

}

public static synchronized void addOnlineCount() {

MyWebSocket.onlineCount++;

}

public static synchronized void subOnlineCount() {

MyWebSocket.onlineCount–;

}

}

[](()3.创建前端页面

---

js中的端口号是本人自己定义的，可以随意定义。

Welcome

<button οnclick=“send()”>Send <button οnclick=“closeWebSocket()”>Close