如何在Spring Boot中集成WebSocket

在Spring Boot中集成WebSocket主要涉及以下几个步骤：

1. 添加依赖：

在你的Spring Boot项目的pom.xml文件中，添加spring-boot-starter-websocket依赖。这个依赖会自动配置WebSocket的支持。

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2. 配置WebSocket：

创建一个配置类来启用WebSocket支持，并配置WebSocket的端点和消息处理器。例如：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS(); // 使用SockJS库简化WebSocket操作
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic"); // 启用简单的消息代理，用于广播消息
        config.setApplicationDestinationPrefixes("/app"); // 设置应用消息的前缀
    }
}

在这个配置类中，我们启用了WebSocket支持，并注册了一个端点/ws，这个端点使用SockJS库来简化WebSocket操作。我们还配置了一个简单的消息代理，用于广播消息。

3. 创建WebSocket消息处理器：

创建一个类来处理WebSocket消息。这个类需要实现MessageHandler接口，并重写handleMessage方法。例如：

@Component
public class MyWebSocketHandler implements MessageHandler {

    @Override
    public void handleMessage(String payload, MessageHeaders headers) {
        // 处理接收到的消息
        System.out.println("Received message: " + payload);
    }
}

4. 发送WebSocket消息：

在你的服务类或控制器类中，你可以使用SimpMessagingTemplate来发送WebSocket消息。例如：

@Autowired
private SimpMessagingTemplate messagingTemplate;

public void sendMessage(String message) {
    messagingTemplate.convertAndSend("/topic/messages", message); // 发送消息到指定的主题
}

在这个例子中，我们使用convertAndSend方法发送一条消息到/topic/messages主题。这个主题将与我们在WebSocketConfig类中配置的简单消息代理相关联。

5. 客户端集成：

你可以使用任何支持WebSocket的客户端库来集成WebSocket功能。例如，你可以使用SockJS库在HTML页面中创建一个WebSocket客户端，并通过JavaScript发送和接收消息。

以上就是在Spring Boot中集成WebSocket的基本步骤。你可以根据自己的需求进行进一步的定制和扩展。