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

Convert Completed from PushPullStage to GraphStage (#20454)

Browse source
This commit is contained in:
Priyanka Chordia 2016-05-05 02:49:02 -07:00 • committed by Konrad Malawski
parent b511508825
commit 1f8225087f
1 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -568,10 +568,23 @@ private[akka] final case class Buffer[T](size: Int, overflowStrategy: OverflowSt
/**
* INTERNAL API
*/
private[akka] final case class Completed[T]() extends PushPullStage[T, T] {
override def onPush(elem: T, ctx: Context[T]): SyncDirective = ctx.finish()
private[akka] final case class Completed[T]() extends GraphStage[FlowShape[T, T]] {
val out: Outlet[T] = Outlet("Completed.out")
val in: Inlet[T] = Inlet("Completed.in")
override def createLogic(inheritedAttributes: Attributes): GraphStageLogic = new GraphStageLogic(shape) with InHandler with OutHandler {
override def onPush(): Unit = { completeStage() }
override def onPull(): Unit = { completeStage() }
setHandlers(in, out, this)
}
val shape: FlowShape[T, T] = FlowShape.of(in, out)
override def onPull(ctx: Context[T]): SyncDirective = ctx.finish()
}
/**