Merge pull request #14 from lihaoyi-databricks/master

Provide a simple builtin websocket client in `cask.WsClient`

1 files changed, 39 insertions, 0 deletions

diff --git a/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala b/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala
new file mode 100644
index 0000000..5570356
--- /dev/null
+++ b/cask/util/src-jvm/cask/util/WebsocketClientImpl.scala
@@ -0,0 +1,39 @@
+package cask.util
+import org.java_websocket.client.WebSocketClient
+import org.java_websocket.handshake.ServerHandshake
+
+abstract class WebsocketClientImpl(url: String) extends WebsocketBase{
+  var websocket: Client = null
+
+  def connect(): Unit = {
+    websocket = new Client()
+    websocket.connect()
+  }
+  def onOpen(): Unit
+  def onMessage(message: String): Unit
+  def send(message: String) = try{
+    websocket.send(message)
+    true
+  }catch{
+    case e: org.java_websocket.exceptions.WebsocketNotConnectedException => false
+  }
+  def send(message: Array[Byte]) = try{
+    websocket.send(message)
+    true
+  }catch{
+    case e: org.java_websocket.exceptions.WebsocketNotConnectedException => false
+  }
+  def onClose(code: Int, reason: String): Unit
+  def onError(ex: Exception): Unit
+  def close(): Unit = websocket.close()
+  def isClosed() = websocket.isClosed()
+  class Client() extends WebSocketClient(new java.net.URI(url)){
+    def onOpen(handshakedata: ServerHandshake) = {
+      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 onError(ex: Exception) = WebsocketClientImpl.this.onError(ex)
+
+  }
+}
\ No newline at end of file