Handle remoing wssession from notebook map once the session is close …

…:: avoiding socket connection to be ide
vrlo · Nov 15, 2016 · 4affe25 · 4affe25
1 parent c32192a
commit 4affe25
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...a/org/apache/zeppelin/notebook/repo/zeppelinhub/websocket/listener/ZeppelinWebsocket.java b/...a/org/apache/zeppelin/notebook/repo/zeppelinhub/websocket/listener/ZeppelinWebsocket.java
@@ -43,7 +43,7 @@ public void onWebSocketBinary(byte[] arg0, int arg1, int arg2) {
   @Override
   public void onWebSocketClose(int code, String message) {
     LOG.info("Zeppelin connection closed with code: {}, message: {}", code, message);
-    // parentClient.removeConnMap(noteId);
+    ZeppelinClient.getInstance().removeNoteConnection(noteId);
   }
 
   @Override
@@ -54,7 +54,8 @@ public void onWebSocketConnect(Session session) {
 
   @Override
   public void onWebSocketError(Throwable e) {
-    LOG.warn("Zeppelin socket connection error: {}", e.toString());
+    LOG.warn("Zeppelin socket connection error ", e);
+    ZeppelinClient.getInstance().removeNoteConnection(noteId);
   }
 
   @Override