Spring Boot整合WebSocket

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
 
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
 
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myHandler(), "/ws/echo")
                .setAllowedOrigins("*"); // 允许所有域进行WebSocket连接
    }
 
    public WebSocketHandler myHandler() {
        // 返回自定义的WebSocketHandler实例
        // 这里需要你自己实现WebSocketHandler接口
        return new MyCustomWebSocketHandler();
    }
}

在这个配置类中，我们使用@EnableWebSocket注解来开启Spring Boot对WebSocket的支持，并实现WebSocketConfigurer接口来注册一个自定义的WebSocket处理器。这个处理器需要你自己实现WebSocketHandler接口。这个例子展示了如何将WebSocket处理器映射到特定的URL上，并设置了允许跨域请求。