springboot 使用 websocket

一. 添加依赖

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

二. 添加websocket服务节点的配置

@Configuration
public class WebsocketConfig {

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

三. 定义websocket服务

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

    private static ConcurrentHashMap<String, Session> sessionMap = new ConcurrentHashMap<>();

    /**
     * 前端发送请求建立websocket链接，就会执行@OnOpen方法
     * @param orderId
     * @param session
     */
    @OnOpen
    public void open(@PathParam("oid") // Websocket的要使用 @PathParam
    				 String orderId, Session session) {
        sessionMap.put(orderId, session);
    }

    /**
     * 前端关闭网页获取主动关闭websocket，就会执行@OnClose
     * @param orderId
     */
    @OnClose
    public void close(@PathParam("oid") String orderId) {
        sessionMap.remove(orderId);
    }

    /**
     * 发送消息
     * @param orderId
     * @param msg
     */
    public static void sendMsg(String orderId, String msg) {
        try {
            Session session = sessionMap.get(orderId);
            session.getBasicRemote().sendText(msg);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}