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

use SimpleLinearGraphStage to reduce the boilerplate code #21830

Browse source
This commit is contained in:
Hawstein 2016-11-18 17:13:15 +08:00 • committed by Johan Andrén
parent e80d7d7a71
commit c373cef20f
4 changed files with 9 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

10
akka-stream-tests/src/test/scala/akka/stream/impl/fusing/InterpreterSpec.scala
View file

@ -577,9 +577,7 @@ class InterpreterSpec extends StreamSpec with GraphInterpreterSpecKit {
}
private[akka] final case class Doubler[T]() extends GraphStage[FlowShape[T, T]] {
val out: Outlet[T] = Outlet("Doubler.out")
val in: Inlet[T] = Inlet("Doubler.in")
private[akka] final case class Doubler[T]() extends SimpleLinearGraphStage[T] {
override def createLogic(inheritedAttributes: Attributes): GraphStageLogic =
new GraphStageLogic(shape) with InHandler with OutHandler {
@ -608,12 +606,9 @@ class InterpreterSpec extends StreamSpec with GraphInterpreterSpecKit {
setHandlers(in, out, this)
}
override val shape: FlowShape[T, T] = FlowShape(in, out)
}
private[akka] final case class KeepGoing[T]() extends GraphStage[FlowShape[T, T]] {
val in = Inlet[T]("KeepGoing.in")
val out = Outlet[T]("KeepGoing.out")
private[akka] final case class KeepGoing[T]() extends SimpleLinearGraphStage[T] {
override def createLogic(inheritedAttributes: Attributes): GraphStageLogic =
new GraphStageLogic(shape) with InHandler with OutHandler {
@ -638,7 +633,6 @@ class InterpreterSpec extends StreamSpec with GraphInterpreterSpecKit {
setHandlers(in, out, this)
}
override val shape: FlowShape[T, T] = FlowShape(in, out)
}
private[akka] class PushFinishStage(onPostStop: () ⇒ Unit = () ⇒ ()) extends SimpleLinearGraphStage[Any] {