引言
随着Web技术的发展，实时通信的需求日益增强。WebSocket协议作为HTML5的一项重要特性，提供了全双工的、持久化的网络连接，使得服务器和客户端能够进行低延迟的数据交换。Spring Framework自4.0版本开始全面支持WebSocket，并在Spring Boot中通过WebSocketMessageBrokerConfigurer接口进一步简化了配置和使用流程。
WebSocketMessageBrokerConfigurer简介
WebSocketMessageBrokerConfigurer是Spring WebSocket模块中的一个关键接口，它允许开发者对WebSocket消息代理进行高级定制。在Spring Boot 2环境中，我们可以通过实现这个接口来自定义WebSocket端点配置、消息代理（Message Broker）规则以及STOMP协议相关的细节。
import org.springframework.messaging.simp.config.WebSocketMessageBrokerConfigurer;
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
// 实现接口方法
}
核心配置方法
registerStompEndpoints()
此方法用于注册WebSocket STOMP端点，通常我们会在此处指定WebSocket服务的实际URL路径，并可以选择是否启用SockJS作为WebSocket的回退方案以兼容不支持WebSocket的浏览器。
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
registry.addEndpoint("/ws").withSockJS();
}
上述代码会在"/ws"路径下创建一个WebSocket端点，并启用了SockJS来确保跨浏览器兼容性。
configureMessageBroker()
该方法用于配置消息代理，包括设置代理的目标前缀（destination prefixes），并决定哪些目的地应该由简单代理（如内存中的队列或主题）处理。
@Override
public void configureMessageBroker(MessageBrokerRegistry config) {
config.enableSimpleBroker("/topic", "/queue");
config.setApplicationDestinationPrefixes("/app");
}
这里设置了两个简单消息代理的目的地前缀："/topic"用于发布/订阅模式的消息，"/queue"用于点对点的消息传递。同时，将所有发往"/app"前缀的目的地的消息路由到应用层处理。
应用场景示例
结合@Controller和@MessageMapping注解，我们可以轻松处理从WebSocket客户端发送过来的消息：
@Controller
public class WebSocketController {
@MessageMapping("/chat")
@SendTo("/topic/messages")
public ChatMessage handleChatMessage(ChatMessage message) {
// 处理聊天消息逻辑
return message;
}
}
在这个例子中，当客户端向"/app/chat"目的地发送消息时，handleChatMessage方法会被调用，处理后的结果会广播到"/topic/messages"目的地的所有订阅者。
扩展功能配置
除了基本的WebSocket端点和消息代理配置外，还可以通过以下方式扩展功能：
- 添加拦截器（Interceptors）：通过registry.addInterceptor()可以在STOMP消息的生命周期内插入自定义拦截器，用于认证、审计等操作。
- 自定义握手处理器（Handshake Handlers）：针对WebSocket握手过程进行个性化处理，例如添加自定义HTTP头部验证。
总结
总结来说，在Spring Boot 2中，WebSocketMessageBrokerConfigurer为开发者提供了一套完整的框架来管理和配置WebSocket服务，极大地简化了WebSocket应用程序的开发和维护工作。通过合理利用这一接口，我们可以构建出基于WebSocket协议的高性能、可扩展的实时通信系统。
原文地址：https://www.toutiao.com/article/7330472362636739107/
