diff options
Diffstat (limited to 'cask/util/src-jvm')
-rw-r--r-- | cask/util/src-jvm/cask/util/Scheduler.scala | 16 | ||||
-rw-r--r-- | cask/util/src-jvm/cask/util/WebsocketClientImpl.scala | 39 |
2 files changed, 55 insertions, 0 deletions
diff --git a/cask/util/src-jvm/cask/util/Scheduler.scala b/cask/util/src-jvm/cask/util/Scheduler.scala new file mode 100644 index 0000000..74db37e --- /dev/null +++ b/cask/util/src-jvm/cask/util/Scheduler.scala @@ -0,0 +1,16 @@ +package cask.util + +import java.util.concurrent.{Executors, TimeUnit} + +object Scheduler{ + val scheduler = Executors.newSingleThreadScheduledExecutor() + def schedule(millis: Long)(body: => Unit) = { + scheduler.schedule( + new Runnable { + def run(): Unit = body + }, + millis, + TimeUnit.MILLISECONDS + ) + } +}
\ No newline at end of file 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 |