https://keyholesoftware.com/2017/04/10/websockets-with-spring-boot/

作者：THOMAS KENDALL

转载自公众号：stackgc

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

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

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

HTML5 来救场！

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

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

以下是一个例子：

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 类。

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 端点。

@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 使它变得更加简单。