raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Optimize Fold supervision

Browse source
This commit is contained in:
Endre Sándor Varga 2016-07-20 13:39:23 +02:00
parent 5382014133
commit a6cf6c646e
1 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

21
akka-stream/src/main/scala/akka/stream/impl/fusing/Ops.scala
View file

@ -379,19 +379,24 @@ final case class Fold[In, Out](zero: Out, f: (Out, In) ⇒ Out) extends GraphSta
override val initialAttributes = DefaultAttributes.fold override val initialAttributes = DefaultAttributes.fold
override def createLogic(inheritedAttributes: Attributes): GraphStageLogic = override def createLogic(inheritedAttributes: Attributes): GraphStageLogic =
new SupervisedGraphStageLogic(inheritedAttributes, shape) with InHandler with OutHandler { new GraphStageLogic(shape) with InHandler with OutHandler {
private var aggregator: Out = zero private var aggregator: Out = zero
override def onResume(t: Throwable): Unit = { private def decider =
aggregator = zero inheritedAttributes.get[SupervisionStrategy].map(_.decider).getOrElse(Supervision.stoppingDecider)
}
override def onPush(): Unit = withSupervision(() ⇒ grab(in)) match { override def onPush(): Unit = {
case Some(elem) ⇒ { try {
aggregator = f(aggregator, elem) aggregator = f(aggregator, grab(in))
pull(in) pull(in)
} catch {
case NonFatal(ex) ⇒ decider(ex) match {
case Supervision.Stop ⇒ failStage(ex)
case _ ⇒
aggregator = zero
pull(in)
}
} }
case None ⇒ pull(in)
} }
override def onPull(): Unit = { override def onPull(): Unit = {