Fix StateMachine to properly inherit from BatchActor

Remove WsClient.close

2 files changed, 3 insertions, 5 deletions

diff --git a/cask/util/src/cask/util/BatchActor.scala b/cask/util/src/cask/util/BatchActor.scala
index fd4861e..4985fc3 100644
--- a/cask/util/src/cask/util/BatchActor.scala
+++ b/cask/util/src/cask/util/BatchActor.scala
@@ -36,9 +36,9 @@ abstract class BatchActor[T]()(implicit ec: ExecutionContext,
   }
 }
 
-abstract class StateMachine[T]()
-                              (implicit ec: ExecutionContext,
-                               log: Logger) {
+abstract class StateMachineActor[T]()
+                                   (implicit ec: ExecutionContext,
+                                    log: Logger) extends BatchActor[T](){
   class State(val run: T => State)
   protected[this] def initialState: State
   protected[this] var state: State = initialState
diff --git a/cask/util/src/cask/util/WsClient.scala b/cask/util/src/cask/util/WsClient.scala
index 6ba039e..fbde444 100644
--- a/cask/util/src/cask/util/WsClient.scala
+++ b/cask/util/src/cask/util/WsClient.scala
@@ -13,8 +13,6 @@ class WsClient(impl: WebsocketBase)
     case Ws.Close(_, _) => impl.close()
     case Ws.ChannelClosed() => impl.close()
   }
-    
-  def close(code: Int = 1005, reason: String = "") = this.send(Ws.Close(code, reason))
 }
 
 object WsClient{