springboot WebSocket 代码示例

以下是一个简单的Spring Boot WebSocket示例：

1. 创建一个Spring Boot项目并添加以下依赖项：

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

2. 创建一个WebSocket配置类并实现WebSocketConfigurer接口：

@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();
    }
}

3. 创建一个WebSocket处理程序类并实现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端点：

@RestController
public class MyController {

    @GetMapping("/websocket")
    public String websocket() {
        return "websocket";
    }
}

5. 在前端页面添加WebSocket连接：

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

6. 运行应用程序并打开浏览器控制台，您将看到WebSocket连接已建立。

以上就是一个简单的Spring Boot WebSocket示例。您可以根据自己的需要修改它。