SpringBoot使用WebSocket的方法实例详解

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

/**
 * token为前端连接时的标识，后端根据此token维护用户与webSocket的绑定
 */
@Component
@ServerEndpoint("/webSocket/{token}")
public class WebSocket {
  private Session session;
  private static ConcurrentHashMap<String, WebSocket> webSocketMap = new ConcurrentHashMap<>();
  @OnOpen
  public void onOpen(@PathParam(value = "token") String token, Session session) {
    this.session = session;
    webSocketMap.put(token, this);
    System.out.println("new connect：" + token + ". total :" + webSocketMap.size());
  }
  @OnClose
  public void onClose(@PathParam(value = "token") String token) {
    webSocketMap.remove(token);
    System.out.println("disconnect:" + token + ", total:" + webSocketMap.size());
  }
  @OnMessage
  public void onMessage(String message) {
    System.out.println("get message: " + message);
  }
  public void sendMessage(String message) {
    // 根据业务逻辑，找到token指向的webSocket
    WebSocket webSocket = webSocketMap.get(String.valueOf(message.charAt(0)));
    // 找不到返回
    if (webSocket == null) {
      return;
    }
    try {
      webSocket.session.getBasicRemote().sendText(message);
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
}

/**
   * WebSocket客户端
   *
   * 使用说明：
   * 1、WebSocket客户端通过回调函数来接收服务端消息。例如：webSocket.onmessage
   * 2、WebSocket客户端通过send方法来发送消息给服务端。例如：webSocket.send();
   */
  function getWebSocket() {
    /**
     * WebSocket客户端 PS：URL开头表示WebSocket协议 中间是域名端口 结尾是服务端映射地址
     */
    var webSocket = new WebSocket(/*[[${webSocketUrl}]]*/ 'ws://localhost:8080/webSocket/2');
    /**
     * 当服务端打开连接
     */
    webSocket.onopen = function (event) {
      console.log('WebSocket打开连接');
    };
    /**
     * 当服务端发来消息：1.广播消息 2.更新在线人数
     */
    webSocket.onmessage = function (event) {
      console.log('WebSocket收到消息：%c' + event.data, 'color:green');
    };
    /**
     * 关闭连接
     */
    webSocket.onclose = function (event) {
      console.log('WebSocket关闭连接');
    };
    /**
     * 通信失败
     */
    webSocket.onerror = function (event) {
      console.log('WebSocket发生异常');
    };
    return webSocket;
  }