feat: add stashing helpers

actorapp · Nov 11, 2015 · b9ba4c7 · b9ba4c7
1 parent ce42600
commit b9ba4c7
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
+.idea/
 target
diff --git a/actor-concurrent/src/main/scala/im/actor/concurrent/ActorStashing.scala b/actor-concurrent/src/main/scala/im/actor/concurrent/ActorStashing.scala
@@ -0,0 +1,17 @@
+package im.actor.concurrent
+
+import akka.actor.{ Stash, ActorRef, Actor }
+
+trait ActorStashing {
+  self: Actor with Stash ⇒
+
+  protected def becomeStashing(f: ActorRef ⇒ Receive): Unit =
+    context.become(f(sender()) orElse stashing, discardOld = false)
+
+  protected def receiveStashing(f: ActorRef ⇒ Receive): Receive =
+    f(sender()) orElse stashing
+
+  private def stashing: Receive = {
+    case msg ⇒ stash()
+  }
+}