springboot集成 websocket

SpringBoot集成WebSocket可以使用Spring框架提供的WebSocket API来实现。在SpringBoot中，使用WebSocket需要进行以下几个步骤：

1. 添加依赖：在pom.xml文件中添加spring-boot-starter-websocket依赖。

2. 创建WebSocket配置类：创建一个WebSocket配置类，用于配置WebSocket相关的参数，如注册EndPoint、消息编解码器、拦截器等。

3. 创建EndPoint：创建一个WebSocket的EndPoint类，用于处理WebSocket连接、消息发送、关闭等操作。可以通过实现Spring提供的WebSocketHandler接口或者继承TextWebSocketHandler来实现。

4. 添加拦截器：可以添加自定义的拦截器，用于处理WebSocket连接建立、消息发送等事件。

5. 配置WebSocket消息代理：使用Spring提供的消息代理，可以实现WebSocket消息的广播和点对点传递。

以下是一个简单的SpringBoot集成WebSocket的示例代码：

1. 添加依赖

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

2. 创建WebSocket配置类

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
 
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myHandler(), "/myHandler")
                .addInterceptors(new WebSocketInterceptor());
    }
 
    @Bean
    public WebSocketHandler myHandler() {
        return new MyHandler();
    }
 
}

3. 创建EndPoint

public class MyHandler extends TextWebSocketHandler {
 
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 处理连接建立事件
    }
 
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 处理消息事件
    }
 
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        // 处理连接关闭事件
    }
 
}

4. 添加拦截器

public class WebSocketInterceptor extends HandshakeInterceptor {
 
    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
            Map<String, Object> attributes) throws Exception {
        // 处理连接建立前事件
        return true;
    }
 
    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
            Exception ex) {
        // 处理连接建立后事件
    }
 
}

5. 配置WebSocket消息代理

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketMessageBrokerConfig implements WebSocketMessageBrokerConfigurer {
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }
 
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/myEndpoint")
                .setAllowedOrigins("*")
                .withSockJS();
    }
 
}