1 files changed, 7 insertions, 3 deletions
diff --git a/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala b/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala
index 5570356..0ac8c1d 100644
--- a/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala
+++ b/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala
@@ -4,8 +4,9 @@ import org.java_websocket.handshake.ServerHandshake
 
 abstract class WebsocketClientImpl(url: String) extends WebsocketBase{
   var websocket: Client = null
-
+  var closed = false
   def connect(): Unit = {
+    closed = false
     websocket = new Client()
     websocket.connect()
   }
@@ -32,8 +33,11 @@ abstract class WebsocketClientImpl(url: String) extends WebsocketBase{
       WebsocketClientImpl.this.onOpen()
     }
     def onMessage(message: String) = WebsocketClientImpl.this.onMessage(message)
-    def onClose(code: Int, reason: String, remote: Boolean) = WebsocketClientImpl.this.onClose(code, reason)
+    def onClose(code: Int, reason: String, remote: Boolean) = {
+      closed = true
+      WebsocketClientImpl.this.onClose(code, reason)
+    }
     def onError(ex: Exception) = WebsocketClientImpl.this.onError(ex)
 
   }
-}
-\ No newline at end of file
+}