Spring Boot WebSocket 客户端集成指南

要在 Spring Boot 中集成 WebSocket 客户端，可以按照以下步骤进行操作：

在 Spring Boot 项目的'pom.xml'文件中添加以下依赖项：

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

创建一个类来处理 WebSocket 连接和消息的逻辑。这个类需要实现'WebSocketHandler'接口，并实现其中的方法。

@Component
public class MyWebSocketHandler implements WebSocketHandler {

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 在建立连接后执行的逻辑
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        // 处理接收到的消息
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        // 处理传输错误
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        // 在连接关闭后执行的逻辑
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
}

在 Spring Boot 的配置类中注册 WebSocket 处理器。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Autowired
    private MyWebSocketHandler myWebSocketHandler;

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myWebSocketHandler, "/websocket").setAllowedOrigins("*");
    }
}

在需要使用 WebSocket 客户端的地方，注入'SimpMessagingTemplate'类，并使用它来发送消息。

@Autowired
private SimpMessagingTemplate messagingTemplate;

public void sendMessage() {
    messagingTemplate.convertAndSend("/websocket", 'Hello, WebSocket!');
}

这样，你就可以在 Spring Boot 中集成 WebSocket 客户端了。