[关闭]
@File 2019-10-29T01:05:58.000000Z 字数 4381 阅读 220

spring-websocket 长连接

java

服务器搭建

一、依赖

  1. <dependency>
  2.     <groupId>org.springframework.boot</groupId>
  3.     <artifactId>spring-boot-starter-websocket</artifactId>
  4. </dependency>

二、config 配置

  1. @Configuration
  2. public class WebSocketConfig {
  3.     @Bean
  4.     public ServerEndpointExporter serverEndpointExporter() {
  5.         return new ServerEndpointExporter();
  6.     }
  7. }

三、websocket 服务器配置

  1. @Component
  2. // 指定 websocker 映射的url路径
  3. @ServerEndpoint("/webSocket")
  4. public class MyWebSocket {
  5.     /**
  6.      * 记录当前在线连接数。
  7.      */
  8.     private static final AtomicInteger onlineCount = new AtomicInteger(0);
  10.     /**
  11.      * 存放每个客户端对应的MyWebSocket对象。
  12.      */
  13.     private static ConcurrentHashMap<String,MyWebSocket> webSocketMap = new ConcurrentHashMap<>();
  15.     /**
  16.      * 当前客户端的连接对象
  17.      */
  18.     private Session session;
  20.     /**
  21.      * 连接建立成功调用的方法
  22.      * @param session           连接的客户端信息
  23.      */
  24.     @OnOpen
  25.     public void onOpen(Session session) {
  26.         this.session = session;
  27.         // 加入set中
  28.         webSocketMap.put(session.getId(), this);
  29.         // 在线数加1
  30.         addOnlineCount();
  31.         // 广播通知
  32.         sendAll("有新用户连接，id:"+session.getId());
  33.     }
  35.     /**
  36.      * 连接关闭调用的方法
  37.      * @param session           断开连接的客户端信息
  38.      */
  39.     @OnClose
  40.     public void onClose(Session session) {
  41.         // 从map中删除
  42.         webSocketMap.remove(session.getId());
  43.         // 在线数减1
  44.         subOnlineCount();
  45.         // 广播退出信息
  46.         sendAll("id:"+session.getId()+" 用户退出了");
  47.     }
  49.     /**
  50.      * 收到客户端消息后调用的方法
  51.      * @param message           客户端发送过来的消息
  52.      * @param session           发信息的客户端信息
  53.      */
  54.     @OnMessage
  55.     public void onMessage(String message, Session session) {
  56.         // 客户端发信息逻辑
  57.         // 信息中是否带冒号
  58.         int index = message.indexOf(':');
  59.         if(index > 0){
  60.             String sendId = message.substring(0, index);
  61.             if(sendId != ""){
  62.                 try {
  63.                     // 带冒号时根据冒号前的id单独发送
  64.                     sendToId(sendId, message.substring(index+1));
  65.                     return;
  66.                 }catch (Exception e){
  68.                 }
  69.             }
  70.         }
  72.         // 没带冒号就广播
  73.         sendAll(message);
  74.     }
  76.     /**
  77.      * 发生错误时调用
  78.      * @param error             错误信息
  79.      * @param session           客户端信息
  80.      */
  81.     @OnError
  82.     public void onError(Throwable error, Session session) {
  83.         error.printStackTrace();
  84.     }
  86.     /**
  87.      * 当前客户端发布消息
  88.      * @param message           消息
  89.      * @throws IOException
  90.      */
  91.     public void sendMessage(String message) throws IOException {
  92.         session.getBasicRemote().sendText(message);
  93.     }
  95.     /**
  96.      * 群发自定义消息
  97.      * @param message           广播
  98.      */
  99.     public static void sendAll(String message) {
  100.         webSocketMap.forEach((key, myWebSocket) -> {
  101.             try {
  102.                 myWebSocket.sendMessage(message);
  103.             } catch (IOException e) {
  104.                 e.printStackTrace();
  105.             }
  106.         });
  107.     }
  109.     /**
  110.      * 发送给指定id的客户端
  111.      * @param id                客户端id
  112.      * @param message           消息
  113.      */
  114.     public static void sendToId(String id, String message) {
  115.         try {
  116.             webSocketMap.get(id).sendMessage(message);
  117.         } catch (IOException e) {
  118.             e.printStackTrace();
  119.         }
  120.     }
  122.     /**
  123.      * 获取总连接数
  124.      * @return      总连接数
  125.      */
  126.     public static synchronized int getOnlineCount() {
  127.         return onlineCount.get();
  128.     }
  130.     /**
  131.      * 连接数 +1
  132.      * @return      +1 后的连接数
  133.      */
  134.     public static synchronized int addOnlineCount() {
  135.         return onlineCount.addAndGet(1);
  136.     }
  138.     /**
  139.      * 连接数 -1
  140.      * @return      -1 后的连接数
  141.      */
  142.     public static synchronized int subOnlineCount() {
  143.         return onlineCount.decrementAndGet();
  144.     }
  145. }

四、前端测试页面（可用客户端代替）

  1. <!DOCTYPE html>
  2. <html>
  3. <head>
  4.     <meta charset="UTF-8">
  5.     <title>My WebSocket Test</title>
  6. </head>
  7. <body>
  9. Welcome<br/>
  10. <input id="text" type="text" />
  11. <button onclick="send()">Send</button>
  12. <button onclick="closeWebSocket()">Close</button>
  13. <div id="message">
  14. </div>
  16. </body>
  18. <script type="text/javascript">
  20.     var websocket = null;
  22.     //判断当前浏览器是否支持WebSocket
  23.     if('WebSocket' in window){
  24.         websocket = new WebSocket("ws://localhost:8080/webSocket");
  25.     }
  26.     else{
  27.         alert('Not support websocket')
  28.     }
  30.     //连接发生错误的回调方法
  31.     websocket.onerror = function(){
  32.         setMessageInnerHTML("error");
  33.     };
  35.     //连接成功建立的回调方法
  36.     websocket.onopen = function(event){
  37.         setMessageInnerHTML("open");
  38.     }
  40.     //接收到消息的回调方法
  41.     websocket.onmessage = function(event){
  42.         setMessageInnerHTML(event.data);
  43.     }
  45.     //连接关闭的回调方法
  46.     websocket.onclose = function(){
  47.         setMessageInnerHTML("close");
  48.     }
  50.     //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
  51.     window.onbeforeunload = function(){
  52.         websocket.close();
  53.     }
  55.     //将消息显示在网页上
  56.     function setMessageInnerHTML(innerHTML){
  57.         document.getElementById('message').innerHTML += innerHTML + '<br/>';
  58.     }
  60.     //关闭连接
  61.     function closeWebSocket(){
  62.         websocket.close();
  63.     }
  65.     //发送消息
  66.     function send(){
  67.         var message = document.getElementById('text').value;
  68.         websocket.send(message);
  69.     }
  70. </script>
  71. </html>

客户端搭建

一、依赖

  1. <dependency>
  2.     <groupId>org.java-websocket</groupId>
  3.     <artifactId>Java-WebSocket</artifactId>
  4.     <version>1.3.8</version>
  5. </dependency>

二、重写 WebSocketClient

  1. public class MyWebSocketClient extends WebSocketClient {
  3.     public MyWebSocketClient(String serverUri) throws URISyntaxException {
  4.         // 配置连接地址
  5.         super(new URI(serverUri));
  6.         // 实例化时立即创建连接
  7.         this.connect();
  8.     }
  10.     /**
  11.      * 连接成功时
  12.      * @param serverHandshake
  13.      */
  14.     @Override
  15.     public void onOpen(ServerHandshake serverHandshake) {
  16.         // 成功连接逻辑
  17.     }
  19.     /**
  20.      * 接收到信息时
  21.      * @param s     信息内容
  22.      */
  23.     @Override
  24.     public void onMessage(String s) {
  25.         // 收信息时逻辑
  26.     }
  28.     /**
  29.      * 关闭连接是
  30.      * @param i
  31.      * @param s
  32.      * @param b
  33.      */
  34.     @Override
  35.     public void onClose(int i, String s, boolean b) {
  36.         // 断开时逻辑
  37.     }
  39.     /**
  40.      * 出现错误时
  41.      * @param e     错误对象
  42.      */
  43.     @Override
  44.     public void onError(Exception e) {
  45.         // 异常处理逻辑
  46.     }
  47. }

三、发消息

  1. // 实例化刚重写的类，这里有个异常需要处理
  2. WebSocketClient webSocketClient = new MyWebSocketClient("ws://localhost:8080/webSocket");
  4. // 向服务器发送消息
  5. webSocketClient.send("发送信息");

参考连接：https://blog.csdn.net/ffj0721/article/details/82630134

添加新批注
在作者公开此批注前，只有你和作者可见。
回复批注