SprinBoot——SpringBoot项目WebSocket推送(1)，mybatis原理和实现机制

• 
  date 2018-09-20

*/

@Bean

public ServerEndpointExporter serverEndpointExporter() {

return new ServerEndpointExporter();

}

}

2.创建一个WebSocket推送类

---

package com.a

《一线大厂Java面试题解析+后端开发学习笔记+最新架构讲解视频+实战项目源码讲义》

【docs.qq.com/doc/DSmxTbFJ1cmN1R2dB】 完整内容开源分享

dc.da.publish.websocket;

import org.springframework.stereotype.Component;

import javax.websocket.OnClose;

import javax.websocket.OnError;

import javax.websocket.OnMessage;

import javax.websocket.OnOpen;

import javax.websocket.Session;

import javax.websocket.server.ServerEndpoint;

import java.io.IOException;

import java.util.concurrent.CopyOnWriteArraySet;

/**

• 单车监控websocket

• @author 刘朋

• 
  date 2018-09-20

*/

@ServerEndpoint("/testWebsocket")

@Component

public class MyWebSocket {

//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。

private static int onlineCount = 0;

//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。

private static CopyOnWriteArraySet webSocketSet = new CopyOnWriteArraySet<>();

//与某个客户端的连接会话，需要通过它来给客户端发送数据

private Session session;

/**

• 连接建立成功调用的方法

*/

@OnOpen

public void onOpen(Session session) {

this.session = session;

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