通过实例讲解springboot整合WebSocket

2023-08-21 14:50:05 163

一、背景

我们都知道http协议只能浏览器单方面向服务器发起请求获得响应，服务器不能主动向浏览器推送消息。想要实现浏览器的主动推送有两种主流实现方式：

轮询：缺点很多，但是实现简单
websocket：在浏览器和服务器之间建立tcp连接，实现全双工通信

springboot使用websocket有两种方式，一种是实现简单的websocket，另外一种是实现STOMP协议。这一篇实现简单的websocket，STOMP下一篇在讲。

注意：如下都是针对使用springboot内置容器

二、实现

1、依赖引入

要使用websocket关键是@ServerEndpoint这个注解，该注解是javaee标准中的注解,tomcat7及以上已经实现了,如果使用传统方法将war包部署到tomcat中，只需要引入如下javaee标准依赖即可：


javax
javaee-api
7.0
provided

如使用springboot内置容器,无需引入，springboot已经做了包含。我们只需引入如下依赖即可：


org.springframework.boot
spring-boot-starter-websocket
1.5.3.RELEASE
pom

2、注入Bean

首先注入一个ServerEndpointExporterBean,该Bean会自动注册使用@ServerEndpoint注解申明的websocketendpoint。代码如下：

@Configuration
publicclassWebSocketConfig{
@Bean
publicServerEndpointExporterserverEndpointExporter(){
returnnewServerEndpointExporter();
}
}

3、申明endpoint

建立MyWebSocket.java类，在该类中处理websocket逻辑

@ServerEndpoint(value="/websocket")//接受websocket请求路径
@Component//注册到spring容器中
publicclassMyWebSocket{
//保存所有在线socket连接
privatestaticMapwebSocketMap=newLinkedHashMap<>();
//记录当前在线数目
privatestaticintcount=0;
//当前连接（每个websocket连入都会创建一个MyWebSocket实例
privateSessionsession;
privateLoggerlog=LoggerFactory.getLogger(this.getClass());
//处理连接建立
@OnOpen
publicvoidonOpen(Sessionsession){
this.session=session;
webSocketMap.put(session.getId(),this);
addCount();
log.info("新的连接加入：{}",session.getId());
}
//接受消息
@OnMessage
publicvoidonMessage(Stringmessage,Sessionsession){
log.info("收到客户端{}消息：{}",session.getId(),message);
try{
this.sendMessage("收到消息："+message);
}catch(Exceptione){
e.printStackTrace();
}
}
//处理错误
@OnError
publicvoidonError(Throwableerror,Sessionsession){
log.info("发生错误{},{}",session.getId(),error.getMessage());
}
//处理连接关闭
@OnClose
publicvoidonClose(){
webSocketMap.remove(this.session.getId());
reduceCount();
log.info("连接关闭:{}",this.session.getId());
}
//群发消息
//发送消息
publicvoidsendMessage(Stringmessage)throwsIOException{
this.session.getBasicRemote().sendText(message);
}
//广播消息
publicstaticvoidbroadcast(){
MyWebSocket.webSocketMap.forEach((k,v)->{
try{
v.sendMessage("这是一条测试广播");
}catch(Exceptione){
}
});
}
//获取在线连接数目
publicstaticintgetCount(){
returncount;
}
//操作count，使用synchronized确保线程安全
publicstaticsynchronizedvoidaddCount(){
MyWebSocket.count++;
}
publicstaticsynchronizedvoidreduceCount(){
MyWebSocket.count--;
}
}

4、客户的实现

客户端使用h5原生websocket，部分浏览器可能不支持。代码如下：



websocket测试



测试

三、测试

建立一个controller测试群发，代码如下：

@RestController
publicclassHomeController{
@GetMapping("/broadcast")
publicvoidbroadcast(){
MyWebSocket.broadcast();
}
}

然后打开上面的html，可以看到浏览器和服务器都输出连接成功的信息：

浏览器：
Event{isTrusted:true,type:"open",target:WebSocket,currentTarget:WebSocket,eventPhase:2,…}
服务端：
2018-08-0114:05:34.727INFO12708---[nio-8080-exec-1]com.fxb.h5websocket.MyWebSocket:新的连接加入：0

点击测试按钮，可在服务端看到如下输出：

2018-08-0115:00:34.644INFO12708---[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

热门推荐

随机推荐