Spring Boot 集成 WebSocket

Spring Boot集成WebSocket主要是为了实现实时双向通信，例如在线聊天、推送通知等功能。WebSocket是一种在单个TCP连接上进行全双工通信的技术，允许服务器主动向客户端推送消息。

在Spring Boot中集成WebSocket，首先需要添加WebSocket相关的依赖，比如`spring-boot-starter-websocket`。然后，你可以创建一个WebSocket处理类，通常继承自`org.springframework.messaging.simp.stomp.StompSessionHandlerAdapter`或`org.springframework.web.socket.TextMessageHandler`，并在其中实现消息的发送和接收逻辑。

以下是一个简单的步骤：

1. 添加WebSocket配置：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig extends StompEndpointConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic");
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }
}

这一步设置了WebSocket的消息代理路径和应用前缀。

2. 创建WebSocket处理器：

@Component
public class ChatWebSocketHandler extends TextMessageHandler {
    // 实现发送和接收消息的方法...
}

3. 注册WebSocket到消息总线：

@Autowired
private MessageBrokerTemplate messageBroker;

// 发送消息
messageBroker.convertAndSendToUser("userId", "/topic/messages", message);

// 接收消息
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) {
    // 处理接收到的消息...
}