一、概述
在上一节 WebSocket的那些事（1-概念篇）中我们简单的介绍了关于WebSocket协议的相关概念、与HTTP的联系区别等等。
这一节将会带来WebSocket的实操环节，主要包括Spring本身提供的WebSocket API以及对STOMP协议的支持。
二、Websocket API
1、引入相关依赖
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency>
2、配置WebSocket处理器
这里我们创建了简单的EchoWebsocketHandler ，会原封不动返回来自客户端的消息。
@Slf4j public class EchoWebsocketHandler extends TextWebSocketHandler { @Override protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { String content = new String(message.asBytes(), StandardCharsets.UTF_8); log.info("WebSocket message received: {}", content); session.sendMessage(message); } @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { log.info("Connection established, request uri:{}", session.getUri()); } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { log.info("Connection closed, request uri:{}", session.getUri()); } }
3、WebSocket配置
@Configuration @EnableWebSocket public class WebsocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { // HttpSessionHandshakeInterceptor可以将HttpSession中的属性转换为WebsocketSession的属性 registry.addHandler(new EchoWebsocketHandler(), "/echo") .addInterceptors(new HttpSessionHandshakeInterceptor()); } @Bean public ServletServerContainerFactoryBean createWebSocketContainer() { ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean(); // 指定单次最大可传输的文本消息大小为4 * 1024字符 container.setMaxTextMessageBufferSize(4 * 1024); // 指定单次最大可传输的二进制消息大小为8 * 1024字节 container.setMaxBinaryMessageBufferSize(8 * 1024); return container; } }
4、测试
直接启动项目，打开Postman，输入 ws://localhost:8080/echo ，如下：
点击连接后，显示连接成功，直接发送消息即可。可以看到，客户端发什么，服务端就响应什么
三、总结
使用原生Websocket会有一些缺点，我们总结一下：
- 兼容性问题：Websocket是HTML5标准中新增的技术，并不是所有浏览器都支持该技术。
- 连接维持问题：WebSocket连接是一直保持开放状态的，因此会占用一定的带宽和资源。如果连接数量过多或者连接持续时间过长，可能会对服务器造成一定的压力。
- 安全问题：WebSocket连接的开放性可能会引起一些安全问题，例如未经授权的用户访问WebSocket服务、数据被劫持等。
- 服务端实现问题：由于消息内容并没有传输标准，无法对消息进行路由和处理。
下篇文章我们将会结合SockJS和STOMP来实现WebSocket消息传输。
