在 SpringBoot中的WebSocket使用介绍

在 Spring Boot 中，WebSocket 是一种非常有用的通信协议。WebSocket 可以使客户端和服务器之间实现双向通信，而不是传统的单向 HTTP 请求和响应。本文将介绍在 Spring Boot 中如何使用 WebSocket。

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

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

2. 创建 WebSocket 配置类
   创建一个 WebSocketConfig 类，在类上加上 @Configuration 和 @EnableWebSocket 注解。在类中添加一个方法，返回一个 WebSocketHandler 对象。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

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

    @Bean
    public WebSocketHandler myHandler() {
        return new MyHandler();
    }
}

其中，/myHandler 是 WebSocket 的端点，MyHandler 是具体的处理类，setAllowedOrigins(“*”) 表示允许跨域请求。

3. 创建 WebSocket 处理类
   创建一个 MyHandler 类实现 WebSocketHandler 接口，并重写其中的方法。具体实现可以根据业务需求来进行定制。

public class MyHandler implements WebSocketHandler {

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 连接建立后的操作
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        // 处理收到的消息
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        // 处理传输错误
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        // 连接关闭后的操作
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
}

4. 发送 WebSocket 消息
   在前端页面中，可以使用 JavaScript 的 WebSocket 对象实现与服务器的通信。例如：

var socket = new WebSocket("ws://localhost:8080/myHandler");

socket.onopen = function(event) {
    // 连接建立后的操作
};

socket.onmessage = function(event) {
    // 处理收到的消息
};

socket.onclose = function(event) {
    // 连接关闭后的操作
};

socket.send("hello world");

5. 测试 WebSocket
   运行 Spring Boot 应用程序，并通过浏览器访问前端页面。在控制台中可以看到 WebSocket 连接的日志信息。可以使用 JavaScript 发送消息，也可以在 MyHandler 中主动发送消息。

session.sendMessage(new TextMessage("hello world"));

总结：
Spring Boot 提供了非常方便的操作 WebSocket 的方式，使得我们可以快速地实现双向通信功能。在实际开发中，可以根据具体业务需求来定制 WebSocket 的处理类。