From df80448fbd1bccdba131d797c0c7e93274ed6ba1 Mon Sep 17 00:00:00 2001
From: ZhuDongming <773644075@qq.com>
Date: Sun, 28 Jun 2020 16:51:44 +0800
Subject: [PATCH] update离线去除风向图标

---
 src/main/java/com/moral/webSocketServer/WebSocketServer.java |   20 +++++++++++++++-----
 1 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/moral/webSocketServer/WebSocketServer.java b/src/main/java/com/moral/webSocketServer/WebSocketServer.java
index e019555..59a9783 100644
--- a/src/main/java/com/moral/webSocketServer/WebSocketServer.java
+++ b/src/main/java/com/moral/webSocketServer/WebSocketServer.java
@@ -2,6 +2,7 @@
 
 import java.io.IOException;
 import java.util.concurrent.CopyOnWriteArraySet;
+import java.util.concurrent.TimeoutException;
 
 import javax.websocket.OnClose;
 import javax.websocket.OnError;
@@ -38,6 +39,10 @@
 
     private String regionCode;
 
+    private Connection connection;
+
+    private Channel channel;
+
     // ������session���������������������������
     private static CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();
 
@@ -64,8 +69,8 @@
             factory.setUsername("guest");
             factory.setPassword("guest_pass");
             String routingKey = orgId+".*";
-            Connection connection = factory.newConnection();
-            Channel channel = connection.createChannel();
+            connection = factory.newConnection();
+            channel = connection.createChannel();
             //���������������������������������������������������������������������������������������������������������
             channel.queueDeclare(QUEUE_NAME, false, false, true, null);
             channel.queueBind(QUEUE_NAME,"screens_data",routingKey);
@@ -82,7 +87,6 @@
                 sendMessage(message);
             }
         }catch (Exception e){
-            e.printStackTrace();
             log.error(e.getMessage());
         }
     }
@@ -94,6 +98,12 @@
     public void onClose() {
         /**���������Set��� ������������������������*/
         webSocketSet.remove(this);
+        try {
+            channel.close();
+            connection.close();
+        }catch (IOException | TimeoutException e){
+            log.error(e.getMessage());
+        }
     }
 
     @OnMessage
@@ -127,7 +137,7 @@
                 this.session.getBasicRemote().sendText(message);
             }
         } catch (IOException e) {
-            e.printStackTrace();
+            log.error(e.getMessage());
         }
     }
 
@@ -139,7 +149,7 @@
      */
     @OnError
     public void onError(Session session, Throwable error) {
-        error.printStackTrace();
+        log.error(error.getMessage());
     }
 
 }

--
Gitblit v1.8.0