springboot 整合websocket

以下是一个示例，演示了如何在Spring Boot中整合WebSocket：

1. 首先，确保你的项目中已经添加了Spring Boot和WebSocket的依赖。在`pom.xml`文件中添加以下依赖：

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

2. 创建一个WebSocket配置类，用于配置WebSocket相关的内容。在该类上添加`@Configuration`和`@EnableWebSocket`注解，并实现`WebSocketConfigurer`接口。在`registerWebSocketHandlers`方法中，注册一个`WebSocketHandler`实例，并指定WebSocket的访问路径。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

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

    @Bean
    public WebSocketHandler myWebSocketHandler() {
        return new MyWebSocketHandler();
    }
}

3. 创建一个WebSocketHandler类，用于处理WebSocket的连接、消息发送和接收等操作。在该类上添加`@Component`注解，并实现`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;
    }
}

4. 在需要使用WebSocket的地方，注入`WebSocketHandler`实例，并使用`WebSocketSession`对象进行消息的发送和接收等操作。

@Autowired
private WebSocketHandler myWebSocketHandler;

public void sendMessage(String message) {
    // 发送消息
    myWebSocketHandler.handleMessage(new TextMessage(message));
}

以上是一个简单的示例，演示了如何在Spring Boot中整合WebSocket。你可以根据自己的需求进行进一步的扩展和定制。