在Spring Boot中实现WebSocket实时通信

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.*;
 
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myHandler(), "/my-websocket-endpoint")
            .setAllowedOrigins("*");
    }
 
    @Bean
    public WebSocketHandler myHandler() {
        // 实现自定义的WebSocketHandler
        return new MyCustomWebSocketHandler();
    }
}

在这个例子中，我们定义了一个配置类WebSocketConfig，实现了WebSocketConfigurer接口。我们注册了一个名为/my-websocket-endpoint的WebSocket处理器，并且设置了允许来自任何源的连接（通常在生产环境中应当限制允许的源）。我们还定义了一个Bean方法myHandler()来提供我们自定义的WebSocket处理器MyCustomWebSocketHandler。这个处理器需要你自己实现org.springframework.web.socket.WebSocketHandler接口。