【WebSocket】SpringBoot项目中使用WebSocket

1. 导入坐标

如果springboot父工程没有加入websocket的起步依赖，添加它的坐标的时候需要带上版本号。

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

2. 创建WebSocketServer服务类

【豆包】

@ServerEndpoint注解

1. 声明服务器端点：通过在类上添加 @ServerEndpoint 注解，该类就成为了一个 WebSocket 服务器端点，可以接收客户端的连接。
2. 指定访问路径：注解的 value 属性用于指定客户端连接的 URI 路径，例如 @ServerEndpoint("/ws") 表示客户端可以通过 ws://服务器地址/ws 连接到该端点。
3. 和@PathParam注解配合使用

@PathParam注解

• 动态路径匹配：通过在 @ServerEndpoint 的路径中定义参数占位符（如 /{username}），可以捕获客户端连接 URI 中的实际值。
• 参数注入：将捕获的路径参数值注入到 @OnOpen、@OnMessage 或 @OnClose 注解方法的参数中。
• 类型转换：自动将路径参数从字符串转换为基本数据类型（如 int、long、boolean 等）。

@OnOpen注解

• 连接初始化：在客户端与服务器成功建立 WebSocket 连接后，被 @OnOpen 注解标记的方法会立即执行。
• 参数注入：方法可以接收特定参数，如 Session 对象、连接参数等。

@OnMessage

• 自动解码：配合 Decoder 使用时，可以自动将消息转换为自定义对象。
• 异步响应：支持同步或异步发送响应。

@OnClose

• 资源释放：清理连接相关的资源（如会话数据、数据库连接）。
• 通知其他客户端：广播用户离开的消息。
• 获取关闭状态：通过 CloseReason 参数了解关闭原因。

@Component
@ServerEndpoint("/ws/{sid}")
public class WebSocketServer {

    //存放会话对象
    private static Map<String, Session> sessionMap = new HashMap();

    //连接建立成功调用的方法
    @OnOpen
    public void onOpen(Session session, @PathParam("sid") String sid) {
        System.out.println("客户端：" + sid + "建立连接");
        sessionMap.put(sid, session);
    }

    //收到客户端消息后调用的方法
    @OnMessage
    public void onMessage(String message, @PathParam("sid") String sid) {
        System.out.println("收到来自客户端：" + sid + "的信息:" + message);
    }

    //连接关闭调用的方法
    @OnClose
    public void onClose(@PathParam("sid") String sid) {
        System.out.println("连接断开:" + sid);
        sessionMap.remove(sid);
    }

    //主动发送消息 （群发消息）
    public void sendToAllClient(String message) {
        Collection<Session> sessions = sessionMap.values();
        for (Session session : sessions) {
            try {
                //服务器向客户端发送消息
                session.getBasicRemote().sendText(message);
                System.out.println("发送websocket");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

}

3. 将WebSocket配置成bean

@Configuration
public class WebSocketConfiguration {

    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }

}