SpringBoot之WebSocket服务搭建

以下是一个简单的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(), "/ws/echo")
                .setAllowedOrigins("*");
    }
 
    @Bean
    public WebSocketHandler myHandler() {
        // 实现WebSocketHandler接口或使用默认实现，例如TextWebSocketHandler
        return new TextWebSocketHandler() {
            @Override
            public void handleTextMessage(WebSocketSession session, TextMessage message) {
                // 接收消息并处理，发送响应
                session.sendMessage(new TextMessage("Echo: " + message.getPayload()));
            }
        };
    }
}

在这个例子中，我们定义了一个WebSocket的配置类WebSocketConfig，实现了WebSocketConfigurer接口。在registerWebSocketHandlers方法中，我们添加了一个处理器myHandler，它的URL映射是/ws/echo。这个处理器是一个TextWebSocketHandler，它会在收到文本消息时回显这些消息。最后，我们通过setAllowedOrigins("*")允许所有源的WebSocket连接。

这个简单的例子展示了如何在Spring Boot应用程序中设置和配置WebSocket服务。在实际应用中，你可能需要扩展WebSocketHandler以处理更复杂的逻辑，并添加额外的配置来满足特定的安全需求或其他性能要求。