一、背景
我们都知道 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。代码如下：
@Configurationpublic class WebSocketConfig {@Beanpublic 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());//处理连接建立@OnOpenpublic void onOpen(Session session){this.session=session;webSocketMap.put(session.getId(),this);addCount();log.info("新的连接加入：{}",session.getId());}//接受消息@OnMessagepublic void onMessage(String message,Session session){log.info("收到客户端{}消息：{}",session.getId(),message);try{this.sendMessage("收到消息："+message);}catch (Exception e){e.printStackTrace();}}//处理错误@OnErrorpublic void onError(Throwable error,Session session){log.info("发生错误{},{}",session.getId(),error.getMessage());}//处理连接关闭@OnClosepublic 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 onclick="sendMessage()">测试</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 测试群发，代码如下：
@RestControllerpublic 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
点击测试按钮，可在服务端看到如下输出：
2018-08-01 15:00:34.644 INFO 12708 --- [nio-8080-exec-6] com.fxb.h5websocket.MyWebSocket : 收到客户端2消息：这是一个测试数据
再次打开 html 页面，这样就有两个 websocket 客户端，然后在浏览器访问localhost:8080/broadcast测试群发功能,每个客户端都会输出如下信息：
MessageEvent {isTrusted: true, data: "这是一条测试广播", origin: "ws://localhost:8080", lastEventId: "", source: null, …}
源码可在 github上下载
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。
本文标题: 通过实例讲解springboot整合WebSocket
以上就上有关通过实例讲解springboot整合WebSocket的全部内容，学步园全面介绍编程技术、操作系统、数据库、web前端技术等内容。