通过实例讲解springboot整合WebSocket

一、背景

我们都知道 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 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 测试群发,代码如下:

@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

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

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 的全部内容, 来源链接: utcz.com/z/318559.html

回到顶部