Scala Stash类代码示例

//设置package包名称以及导入依赖的类
package com.dazito.scala.akkademy.client

import java.util.concurrent.TimeUnit

import akka.actor.{Actor, Stash}
import com.dazito.scala.dakkabase.messages.{Disconnected, Connected, GetRequest}


class RemoteActorProxyWithBecome extends Actor with Stash {

    override def receive: Receive = {
        case x: GetRequest =>
            stash()
        case _: Connected =>
            context.become(online)
            unstashAll()
    }

    def online: Receive = {
        case x: GetRequest =>
            processMessage(x)
        case _: Disconnected =>
            context.unbecome()
    }

    def processMessage(message: AnyRef) = {
        // Do something
    }
} 

//设置package包名称以及导入依赖的类
package com.dazito.scala.akkademy.client

import akka.actor.Actor.Receive
import akka.actor.{Actor, Stash}
import com.dazito.scala.dakkabase.messages.{Disconnected, Connected, GetRequest}


class RemoteActorProxy extends Actor with Stash {

    var isOnline = false;

    override def receive: Receive = {
        case message: GetRequest =>
            if(isOnline) {
                processMessage(message)
            }
            else {
                stash()
            }
        case _: Connected =>
            isOnline = true
            unstashAll()
        case _: Disconnected =>
            isOnline = false;
    }

    def processMessage(message: AnyRef) = {
        // Do something
    }
} 

//设置package包名称以及导入依赖的类
package com.udemy.akka.become

import akka.actor.{Actor, Stash}
import akka.actor.Actor.Receive
import com.udemy.akka.become.UserStorage.{Connect, Disconnect, Operation}


case class User(username:String,email:String)

object UserStorage{
  trait DBOperation
  object DBOperation{
    case object Create extends DBOperation
    case object Read extends DBOperation
    case object Update extends DBOperation
    case object Delete extends DBOperation
  }

  case object Connect
  case object Disconnect

  case class Operation(operation:DBOperation,user:Option[User])
}

class UserStorage extends Actor with Stash{
  override def receive: Receive = disconnected

  def connected:Actor.Receive= {
    case Disconnect=>
      println("User storage disconnected from DB")
      unstashAll()
      context.unbecome()
    case Operation(op,user)=>
      println(s"User storage recieved $op to do in user : $user")
  }

  def disconnected:Actor.Receive = {
    case Connect =>
      println("User storage connected to DB")
      unstashAll()
      context.become(connected)
    case _ =>
      stash()
  }


} 

//设置package包名称以及导入依赖的类
package mesosphere.mesos.simulation

import akka.actor.{ Actor, Stash }
import akka.event.LoggingReceive
import mesosphere.marathon.stream._
import org.apache.mesos.Protos.{ FrameworkID, MasterInfo, Offer, TaskStatus }
import org.apache.mesos.{ Scheduler, SchedulerDriver }
import org.slf4j.LoggerFactory

import scala.collection.immutable.Seq

object SchedulerActor {
  private val log = LoggerFactory.getLogger(getClass)

  case class Registered(
    frameworkId: FrameworkID,
    master: MasterInfo)

  case class ResourceOffers(offers: Seq[Offer])
}

class SchedulerActor(scheduler: Scheduler) extends Actor with Stash {
  import SchedulerActor._

  var driverOpt: Option[SchedulerDriver] = None

  def receive: Receive = waitForDriver

  def waitForDriver: Receive = LoggingReceive.withLabel("waitForDriver") {
    case driver: SchedulerDriver =>
      log.info("received driver")
      driverOpt = Some(driver)
      context.become(handleCmds(driver))
      unstashAll()

    case _ => stash()
  }

  def handleCmds(driver: SchedulerDriver): Receive = LoggingReceive.withLabel("handleCmds") {
    case Registered(frameworkId, masterInfo) =>
      scheduler.registered(driver, frameworkId, masterInfo)

    case ResourceOffers(offers) =>
      scheduler.resourceOffers(driver, offers)

    case status: TaskStatus =>
      scheduler.statusUpdate(driver, status)
  }

  override def postStop(): Unit = {
    driverOpt.foreach { driver => scheduler.disconnected(driver) }
  }
} 

//设置package包名称以及导入依赖的类
package com.github.wildprairie.common.actors.shared

import akka.actor.{Actor, ActorRef, Stash}

import scala.concurrent.{Future, Promise}


trait ActorFolder { this: Actor with Stash =>

  def fold[B](actors: Traversable[ActorRef], input: Any)(accumulator: B)(
    accumulate: PartialFunction[(Any, B), B]): Future[B] = {
    val promise = Promise[B]()
    if (actors.isEmpty) {
      promise.success(accumulator)
    } else {
      actors.foreach(_ ! input)
      context.become(collectAndDispatch(actors.size, promise)(accumulator)(accumulate))
    }
    promise.future
  }

  private[this] def collectAndDispatch[B](remaining: Int, promise: Promise[B])(accumulator: B)(
    accumulate: PartialFunction[(Any, B), B]): Receive = {
    case msg =>
      val matched =
        accumulate.runWith { res =>
          if (remaining > 1) {
            context.become(
              collectAndDispatch(remaining - 1, promise)(res)(accumulate),
              discardOld = true
            )
          } else {
            promise.success(res)
            unstashAll()
            context.unbecome()
          }
        }.apply((msg, accumulator))

      if (!matched) stash()
  }
} 

//设置package包名称以及导入依赖的类
package database

import akka.actor.{Stash, Props, ActorSystem, Actor}
import database.UserStorage.{DBOperation, Connect, DisConnect, Operation}


case class User(username: String, email: String)

object UserStorage {
  trait DBOperation
  object DBOperation {
    case object Create extends DBOperation
    case object Update extends DBOperation
    case object Read extends DBOperation
    case object Delete extends DBOperation
  }

  case object Connect
  case object DisConnect
  case class Operation(dbOperation: DBOperation, user: Option[User])
}

class UserStorage extends Actor with Stash {

  def connected: Actor.Receive = {
    case DisConnect =>
      println("User Storage Disconnect from DB")
      context.unbecome()
    case Operation(op, user) =>
      println(s"User Storage receive ${op} to do in user: ${user} ")
  }

  def disconnected: Actor.Receive = {
    case Connect =>
      println(s"User Storage connected to DB")
      unstashAll() // inqueue's message
      context.become(connected)

    case _ =>
      stash()
  }

  override def receive: Receive = disconnected
}

object BecomeHotswap extends App {

  val system = ActorSystem("Hotswap-Become")
  val userStorage = system.actorOf(Props[UserStorage], "userStorage")

  userStorage ! Connect

  userStorage ! Operation(DBOperation.Create, Some(User("Admin", "[email protected]")) )

  userStorage ! DisConnect

  Thread.sleep(100)
  system.shutdown()

} 

//设置package包名称以及导入依赖的类
package hu.bme.mit.ire.nodes.unary

import akka.actor.{Actor, Stash}
import hu.bme.mit.ire._
import hu.bme.mit.ire.messages._

abstract class UnaryNode(val expectedTerminatorCount: Int = 1) extends Actor with Forwarder with Stash with TerminatorHandler {
  val log = context.system.log
  val name = self.path.name

  def onChangeSet(changeSet: ChangeSet)

  override def receive: Actor.Receive = {
    case pause: Pause => context.become({
      case resume: Resume => {
        if (resume.messageID == pause.messageID) {
          context.unbecome()
          unstashAll()
        } else stash()
      }
      case terminator: TerminatorMessage => handleTerminator(terminator)
      case _ => stash()
    })
    case changeSet: ChangeSet => onChangeSet(changeSet)
    case terminator: TerminatorMessage => handleTerminator(terminator)
    case Primary | Secondary =>
      throw new UnsupportedOperationException(s"$name received Beta-wrapped message")
    case _:SizeRequest => sender() ! onSizeRequest()
    case _ => throw new UnsupportedOperationException(s"$name received unknown message")
  }

  def onSizeRequest(): Long
} 

//设置package包名称以及导入依赖的类
package com.packt.chapter10

import akka.actor.{Actor, ActorLogging, ActorRef, Stash}
import com.packt.chapter10.PausableActor.{Ready, Work}

object PausableActor {
  case class Work(id: Int)
  case object Ready
}

class PausableActor(target: ActorRef) extends Actor with ActorLogging with Stash {
  def receive = {
    case work: Work =>
      target ! work
      log.info(s"Received Work [${work.id}]. Sending and pausing.")
      context.become(pausedBehavior, discardOld = false)
  }

  def pausedBehavior : Receive = {
    case work: Work =>
      stash()
    case Ready if sender == target =>
      log.info(s"[${target.path.name}] is ready again.")
      context.unbecome()
      unstashAll()
    case Ready =>
      log.info(s"Discarding [Ready] from other actor different from ${target.path.name}")
  }
} 

//设置package包名称以及导入依赖的类
package com.mikemunhall.simpletwitterstats.server

import akka.actor.{Actor, Props, Stash}
import com.mikemunhall.simpletwitterstats.model.metrics.Tweet
import com.mikemunhall.simpletwitterstats.server.util.ParseUtil
import com.typesafe.scalalogging.StrictLogging

object PersistingActor {
  def props = Props[PersistingActor]
  case class Persist(tweet: Tweet)
}

class PersistingActor extends Actor with Stash with StrictLogging with ParseUtil {
  import com.mikemunhall.simpletwitterstats.server.PersistingActor.Persist
  import com.mikemunhall.simpletwitterstats.timeSeriesData

  self ! "initialize"

  def receive = uninitialized

  def uninitialized: Receive = {
    case "initialize" =>
      logger.debug("Initializing PersistingActor and unstashing messages.")
      unstashAll()
      context.become(initialized)
    case _ =>
      logger.debug("PersistingActor invoked before initialization. Stashing message.")
      stash()
  }

  def initialized: Receive = {
    case Persist(tweet) =>
      timeSeriesData.add(tweet)
  }
} 

//设置package包名称以及导入依赖的类
package com.mikemunhall.simpletwitterstats.server

import akka.actor.{Actor, Props, Stash}
import com.mikemunhall.simpletwitterstats.model.metrics.Tweet
import com.mikemunhall.simpletwitterstats.server.util.ParseUtil
import com.typesafe.scalalogging.StrictLogging
import twitter4j.Status

object ParsingActor {
  case class Parse(status: Status)
}

class ParsingActor extends Actor with Stash with StrictLogging with ParseUtil {
  import com.mikemunhall.simpletwitterstats.server.ParsingActor.Parse
  import com.mikemunhall.simpletwitterstats.server.PersistingActor.Persist

  val persistingActor = context.actorOf(PersistingActor.props, "persistingActor")

  self ! "initialize"

  def receive = uninitialized

  def uninitialized: Receive = {
    case "initialize" =>
      logger.debug("Initializing ParsingActor and unstashing messages.")
      unstashAll()
      context.become(initialized)
    case _ =>
      logger.debug("ParsingActor invoked before initialization. Stashing message.")
      stash()
  }

  def initialized: Receive = {
    case Parse(status) =>
      val urlEntities = status.getURLEntities
      val tweet = Tweet(
        status.getId,
        dateToLocalDateTime(status.getCreatedAt),
        emojisFromText(status.getText),
        status.getHashtagEntities.map(_.getText).toList,
        urlEntities.map(_.getExpandedURL).filterNot(_ == "").map(domainFromUrl(_)).toList,
        // TODO: Externalize the domain filter
        urlEntities.map(_.getExpandedURL).filter(d => d.contains("pic.twitter") || d.contains("instagram")).map(domainFromUrl(_)).toList
      )
      persistingActor ! Persist(tweet)
  }
} 

//设置package包名称以及导入依赖的类
package com.pagerduty.scheduler.akka

import akka.actor.{ActorRef, ActorRefFactory, Props, Stash}
import com.pagerduty.scheduler.model.{Task, TaskKey}

import scala.collection.immutable.SortedMap

object OrderingExecutor {

  
class OrderingExecutor(
    orderingId: Task.OrderingId,
    partitionExecutor: ActorRef,
    taskExecutorFactory: (ActorRefFactory, Task, ActorRef) => ActorRef)
    extends ExtendedLoggingFSM[OrderingExecutor.State, OrderingExecutor.Data]
    with Stash {

  import OrderingExecutor._
  import PartitionExecutor._

  startWith(Idle, Data(SortedMap.empty))
  override val supervisorStrategy = Supervision.AlwaysEscalateStrategy

  when(Idle) {
    case Event(ExecuteOrderingTask(task), data) if task.orderingId == orderingId => {
      assert(data.taskQueue.isEmpty)
      execute(task)
      goto(Executing)
    }
  }

  when(Executing) {
    case Event(ExecuteOrderingTask(task), data) if task.orderingId == orderingId => {
      stay() using Data(data.taskQueue + (task.taskKey -> task))
    }
    case Event(TaskExecuted(taskKey), data) if taskKey.orderingId == orderingId => {
      partitionExecutor ! OrderingTaskExecuted(taskKey)
      if (data.taskQueue.isEmpty) {
        goto(Idle)
      } else {
        val (taskKey, task) = data.taskQueue.head
        val remaining = data.taskQueue.tail
        execute(task)
        stay() using Data(remaining)
      }
    }
  }

  def execute(task: Task): Unit = {
    taskExecutorFactory(context, task, context.self)
  }
} 

//设置package包名称以及导入依赖的类
package PokemonsDataStore

import java.io.FileReader

import Common.{FetchPokemonByIdRequest, FetchPokemonByIdResponse, GetStatsRequest, RequestStatus}
import PokemonsDataStore.PokemonModel.{PokemonStat, TypeRouter}
import akka.actor.{Actor, ActorRef, Props, Stash}
import java.nio.file.{Files, Paths}

import scala.io.Source

class Pokedex(pokeApiClient : ActorRef) extends Actor with Stash{

  var pokemonsLeftToInit = 721

  def OnPokemonReceived(response : FetchPokemonByIdResponse): Unit = {
    response match {
      case FetchPokemonByIdResponse(_, _, RequestStatus.Success) =>
        // Spawning pokemon actor containing info retrieved from pokeApi
        pokemonsLeftToInit -= 1
        val currentPokemonId = 722 - pokemonsLeftToInit
        context.actorOf(Props(new PokemonBase(response.result, typeRouter)), response.result.name)
        if (pokemonsLeftToInit > 0) {
            println("Fetching next pokemon (id: " +  currentPokemonId + " )")
            CreatePokemon(currentPokemonId)
          }
        else {
          println("No more pokemon to spawn !")
          unstashAll()
          context.become(receive)
        }
      case FetchPokemonByIdResponse(request : FetchPokemonByIdRequest, _, RequestStatus.Error) =>
        // Trying again. Not the best way to handle a failure, only serves as fail-safe for now
        CreatePokemon(request.id)
    }
  }


  var typeRouter : ActorRef = ActorRef.noSender

  def CreatePokemon(id: Int): Unit = {
    pokeApiClient.tell(FetchPokemonByIdRequest(id), self)
  }

  override def preStart(): Unit = {
    super.preStart()
    typeRouter = context.actorOf(Props[TypeRouter], "typeRouter")
    CreatePokemon(1)
  }

  // I could have stashed request until all pokemons are up before answering but I prefer to be as reactive as possible,
  // even if it means only having partial results
  override def receive: Receive = {
    case request : GetStatsRequest =>
      val path = context.self.path + "/" + request.pokemonName
      println("Received request for pokemon named " + request.pokemonName + ". Searching at path: " + path)
      context.actorSelection(path).forward(request)
    case response : FetchPokemonByIdResponse =>
      OnPokemonReceived(response)
  }
} 

//设置package包名称以及导入依赖的类
// Copyright (c) Microsoft. All rights reserved.

package it.helpers

import java.util.concurrent.Executors

import akka.actor.{Actor, Stash}

import scala.concurrent.ExecutionContext


class Counter extends Actor with Stash {

  implicit val executionContext = ExecutionContext
    .fromExecutorService(Executors.newFixedThreadPool(sys.runtime.availableProcessors))

  private[this] var count: Long = 0

  override def receive: Receive = ready

  def ready: Receive = {
    case "reset" ? {
      context.become(busy)
      count = 0
      context.become(ready)
      unstashAll()
    }
    case "inc"   ? {
      context.become(busy)
      count += 1
      context.become(ready)
      unstashAll()
    }
    case "get"   ? sender() ! count
  }

  def busy: Receive = {
    case _ ? stash()
  }
} 

//设置package包名称以及导入依赖的类
package fr.acinq.eclair

import akka.actor.{Actor, ActorLogging, ActorRef, Stash}
import fr.acinq.eclair.channel.Commitments.msg2String
import fr.acinq.eclair.channel.{INPUT_DISCONNECTED, INPUT_RECONNECTED}
import fr.acinq.eclair.wire.LightningMessage


class Pipe extends Actor with Stash with ActorLogging {

  def receive = {
    case (a: ActorRef, b: ActorRef) =>
      unstashAll()
      context become connected(a, b)

    case msg => stash()
  }

  def connected(a: ActorRef, b: ActorRef): Receive = {
    case msg: LightningMessage if sender() == a =>
      log.debug(f"A ---${msg2String(msg)}%-6s--> B")
      b forward msg
    case msg: LightningMessage if sender() == b =>
      log.debug(f"A <--${msg2String(msg)}%-6s--- B")
      a forward msg
    case [email protected]_DISCONNECTED =>
      log.debug("DISCONNECTED")
      // used for fuzzy testing (eg: send Disconnected messages)
      a forward msg
      b forward msg
      context become disconnected(a, b)
  }

  def disconnected(a: ActorRef, b: ActorRef): Receive = {
    case msg: LightningMessage if sender() == a =>
      // dropped
      log.info(f"A ---${msg2String(msg)}%-6s-X")
    case msg: LightningMessage if sender() == b =>
      // dropped
      log.debug(f"  X-${msg2String(msg)}%-6s--- B")
    case [email protected]_RECONNECTED(r) =>
      log.debug(s"RECONNECTED with $r")
      // used for fuzzy testing (eg: send Disconnected messages)
      a forward msg
      b forward msg
      r ! (a, b)

  }
} 

//设置package包名称以及导入依赖的类
package com.akkastarting.chapter5

import akka.actor.{ActorLogging, Actor, Props, Stash}

class Ch5PingActor extends Actor with ActorLogging with Stash {
  val child = context.system.actorOf(Ch5Ping1Actor.props, "Ping1Actor")

  
  def zeroDone: Receive = {
    case Done(message: String) =>
      log.info(s"Received the first done -> $message")
      context.become(oneDone)
    case _ =>
      stash()
  }

  def oneDone: Receive = {
    case Done(message: String) =>
      log.info(s"Received the second done -> $message")
      unstashAll()
      context.become(allDone)
    case _ =>
      stash()
  }

  def allDone: Receive = {
    case Reset(message: String) =>
      log.info(s"Received a reset -> $message")
      context.become(receive)
    case _ =>
      stash()
  }

  def receive: Receive = {
    case Work(message: String) =>
      (child ! message)(self)
      context.become(zeroDone)
    case _ =>
      stash()
  }
}

object Ch5PingActor {
  val props = Props[Ch5PingActor]
}