如何在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的基本步驟。你可以根據自己的需求進行進一步的定制和擴展。