WebSocket的那些事（2-实操篇）-Toy模板网

这篇具有很好参考价值的文章主要介绍了WebSocket的那些事（2-实操篇）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

一、概述

在上一节 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;
	}
}