springboot整合webSocket

好的，下面是Spring Boot集成WebSocket的步骤：

1. 添加依赖

在pom.xml文件中添加以下依赖：

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

2. 创建WebSocket配置类

创建一个WebSocket配置类，用于注册WebSocket服务器和处理器。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MyWebSocketHandler(), "/ws").setAllowedOrigins("*");
    }
}

在这个配置类中，我们注册了一个WebSocket处理器和它的访问路径。

3. 创建WebSocket处理器

创建一个WebSocket处理器类，用于处理WebSocket连接和消息。

public class MyWebSocketHandler extends TextWebSocketHandler {

    @Override
    public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 处理收到的消息
        String payload = message.getPayload();
        session.sendMessage(new TextMessage("Hello, " + payload));
    }
}

在这个处理器类中，我们重写了handleTextMessage方法来处理WebSocket消息。

4. 创建WebSocket客户端

创建一个WebSocket客户端类，用于连接和发送消息。

public class MyWebSocketClient extends WebSocketClient {

    @Override
    public void onOpen(ServerHandshake handshake) {
        // 连接成功后发送消息
        send("World");
    }

    @Override
    public void onMessage(String message) {
        // 处理收到的消息
        System.out.println("Received message: " + message);
    }

    @Override
    public void onClose(int code, String reason, boolean remote) {
        // 连接关闭时的处理
    }

    @Override
    public void onError(Exception ex) {
        // 异常处理
    }
}

在这个客户端类中，我们重写了WebSocketClient的几个方法，包括连接成功时的onOpen方法、接收消息时的onMessage方法等。

5. 启动WebSocket服务器

在Spring Boot应用程序的main方法中添加以下代码，启动WebSocket服务器：

public static void main(String[] args) {
    SpringApplication.run(MyApplication.class, args);
    
    // 启动WebSocket服务器
    WebSocketServerFactory factory = new DefaultWebSocketServerFactory();
    WebSocketServer webSocketServer = new DefaultWebSocketServer(factory);
    webSocketServer.start();
}

现在，您已经完成了Spring Boot与WebSocket的集成。可以使用WebSocket客户端连接到服务器，发送消息并接收回复。