实时通信:使用 Spring Boot 实现 WebSocket

https://keyholesoftware.com/2017/04/10/websockets-with-spring-boot/
作者:THOMAS KENDALL
译者:oopsguy.com

在开发 Web 应用时,我们有时需要将服务端事件推送到连接的客户端。但 HTTP 并不能做到:客户端打开与服务端的连接并请求数据,反之服务端无法实现。

为了突破这个限制,我们可以建立一个轮询模式,网页间隔轮询服务器以获取新事件。但这种模式不太理想,因为它增加了 HTTP 开销,速度也受轮询速率影响,并且给服务器增加了不必要的负载。

幸运的是,HTML5 WebSocket 出现了。WebSocket 协议允许浏览器与 Web 服务器之间进行低开销的交互。在文中,我们将介绍 Websockets API,并讲解如何使用 Spring Boot 实现 Websocket。

HTML5 来救场!

WebSocket 通过浏览器和服务器之间的单连接提供全双工通信。它不存在 HTTP 开销,并且允许服务器将消息实时推送到客户端。

WebSocket API 实际上很简单。你只需要创建一个 WebSocket 对象,附加事件监听器和发送消息即可。

以下是一个例子:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
var socket = new WebSocket('ws://' + window.location.host + '/my-websocket-endpoint');

// 链接打开的监听事件
socket.onopen = function() {
console.log('WebSocket connection opened. Ready to send messages.');

// 发送一条消息到服务端
socket.send('Hello, from WebSocket client!');
};

// 接受到服务端消息的监听器
socket.onmessage = function(message) {
console.log('Message received from server: ' + message);
};

Spring Boot

Spring 为 WebSocket 接口提供了很好的支持。

首先,我们需要创建一个类,继承 Spring 的 TextWebSocketHandler 类。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
public class MyMessageHandler extends TextWebSocketHandler {

@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
// WebSocket 被关闭后触发
}

@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
// 新 WebSocket 建立连接触发
// 可以保存 session 对象来发送消息

// 发送第一条消息
session.sendMessage(new TextMessage("You are now connected to the server. This is the first message."));
}

@Override
protected void handleTextMessage(WebSocketSession session, TextMessage textMessage) throws Exception {
// 收到消息触发
System.out.println("Message received: " + textMessage.getPayload());
}
}

接下来,我们需要配置 WebSocket 端点。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
@Configuration
@EnableWebSocket
public class WebsocketConfig implements WebSocketConfigurer {

@Bean
public WebSocketHandler myMessageHandler() {
return new MyMessageHandler();
}

@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(myMessageHandler(), "/my-websocket-endpoint");
}

}

由于 WebSockets API 是纯 JavaScript,你可以在大多数前端框架中使用它。包括 Angular,你可以在 TypeScript 中包含 JavaScript 代码。

总结

是不是相当简单?它解决了服务端和客户端之间传输数据的一大困扰。Spring Boot 使它变得更加简单。