我们都知道 http 协议只能浏览器单方面向服务器发起请求获得响应，服务器不能主动向浏览器推送消息。想要实现浏览器的主动推送有两种主流实现方式：
轮询：缺点很多，但是实现简单
websocket：在浏览器和服务器之间建立 tcp 连接，实现全双工通信
springboot 使用 websocket 有两种方式，一种是实现简单的 websocket，另外一种是实现STOMP协议。这一篇实现简单的 websocket，STOMP 下一篇在讲。
注意：如下都是针对使用 springboot 内置容器
二、实现 1、依赖引入要使用 websocket 关键是@ServerEndpoint这个注解，该注解是 javaee 标准中的注解,tomcat7 及以上已经实现了,如果使用传统方法将 war 包部署到 tomcat 中，只需要引入如下 javaee 标准依赖即可：
<dependency> <groupId>javax</groupId> <artifactId>javaee-api</artifactId> <version>7.0</version> <scope>provided</scope> </dependency>如使用 springboot 内置容器,无需引入，springboot 已经做了包含。我们只需引入如下依赖即可：
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> <version>1.5.3.RELEASE</version> <type>pom</type> </dependency> 2、注入 Bean首先注入一个ServerEndpointExporterBean,该 Bean 会自动注册使用@ServerEndpoint 注解申明的 websocket endpoint。代码如下：
@Configuration public class WebSocketConfig { @Bean public ServerEndpointExporter serverEndpointExporter(){ return new ServerEndpointExporter(); } } 3、申明 endpoint建立MyWebSocket.java类，在该类中处理 websocket 逻辑
@ServerEndpoint(value = "/websocket") //接受websocket请求路径 @Component //注册到spring容器中 public class MyWebSocket { //保存所有在线socket连接 private static Map<String,MyWebSocket> webSocketMap = new LinkedHashMap<>(); //记录当前在线数目 private static int count=0; //当前连接（每个websocket连入都会创建一个MyWebSocket实例 private Session session; private Logger log = LoggerFactory.getLogger(this.getClass()); //处理连接建立 @OnOpen public void onOpen(Session session){ this.session=session; webSocketMap.put(session.getId(),this); addCount(); log.info("新的连接加入：{}",session.getId()); } //接受消息 @OnMessage public void onMessage(String message,Session session){ log.info("收到客户端{}消息：{}",session.getId(),message); try{ this.sendMessage("收到消息："+message); }catch (Exception e){ e.printStackTrace(); } } //处理错误 @OnError public void onError(Throwable error,Session session){ log.info("发生错误{},{}",session.getId(),error.getMessage()); } //处理连接关闭 @OnClose public void onClose(){ webSocketMap.remove(this.session.getId()); reduceCount(); log.info("连接关闭:{}",this.session.getId()); } //群发消息 //发送消息 public void sendMessage(String message) throws IOException { this.session.getBasicRemote().sendText(message); } //广播消息 public static void broadcast(){ MyWebSocket.webSocketMap.forEach((k,v)->{ try{ v.sendMessage("这是一条测试广播"); }catch (Exception e){ } }); } //获取在线连接数目 public static int getCount(){ return count; } //操作count，使用synchronized确保线程安全 public static synchronized void addCount(){ MyWebSocket.count++; } public static synchronized void reduceCount(){ MyWebSocket.count--; } } 4、客户的实现客户端使用 h5 原生 websocket，部分浏览器可能不支持。代码如下：
<html> <head> <title>websocket测试</title> <meta charset="utf-8" /> </head> <body> <button>测试</button> <script> let socket = new WebSocket("ws://localhost:8080/websocket"); socket.onerror = err => { console.log(err); }; socket.onopen = event => { console.log(event); }; socket.onmessage = mess => { console.log(mess); }; socket.onclose = () => { console.log("连接关闭"); }; function sendMessage() { if (socket.readyState === 1) socket.send("这是一个测试数据"); else alert("尚未建立websocket连接"); } </script> </body> </html> 三、测试建立一个 controller 测试群发，代码如下：
@RestController public class HomeController { @GetMapping("/broadcast") public void broadcast(){ MyWebSocket.broadcast(); } }然后打开上面的 html，可以看到浏览器和服务器都输出连接成功的信息：
浏览器： Event {isTrusted: true, type: "open", target: WebSocket, currentTarget: WebSocket, eventPhase: 2, …} 服务端： 2018-08-01 14:05:34.727 INFO 12708 --- [nio-8080-exec-1] com.fxb.h5websocket.MyWebSocket : 新的连接加入：0