temporarilyBecome.scala

import akka.actor._

package object akkautil {

  /** Switches the implicit `ActorContext`'s receive behavior to `behavior`
    * until one message is handled, then returns to the previous behavior.
    */
  def temporarilyBecome(behavior: Actor.Receive)(implicit context: ActorContext) {

    import context.unbecome
    val become = context.become(_: Actor.Receive, discardOld = false)

    become {
      case m if behavior isDefinedAt m =>
        unbecome()
        behavior(m)
    }
  }

}