From e44b9de83a91f0e3e56a9234b6880312f585138b Mon Sep 17 00:00:00 2001 From: Li Haoyi Date: Thu, 19 Sep 2019 14:16:10 +0800 Subject: Update WebsocketClientImpl.scala --- cask/util/src-jvm/cask/util/WebsocketClientImpl.scala | 10 +++++++--- 1 file 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 +} -- cgit v1.2.3