Question

我试图了解一些较新的Akka Streams。我有一个自定义的FanOutShape2形状，它做了一些非常简单的事情：接受（Boolean，Option [A_Thing]）的输入并决定将流程路由到out0或out1（通过或失败），如下所示：

object PassFilter {
  type FilterShape = FanOutShape2[(Boolean, Option[OutputWrapper]), OutputWrapper, akka.NotUsed]
}
import PassFilter._

case class PassFilter()(implicit asys: ActorSystem) extends GraphStage[FilterShape] {
  val mergedIn: Inlet[(Boolean, Option[OutputWrapper])] = Inlet("Merged")
  val outPass: Outlet[OutputWrapper] = Outlet("Pass")
  val outFail: Outlet[akka.NotUsed] = Outlet("Fail")

  override val shape: FilterShape = new FanOutShape2(mergedIn, outPass, outFail)

  override def createLogic(inheritedAttributes: Attributes): GraphStageLogic = new GraphStageLogic(shape) {
    override def preStart(): Unit = pull(mergedIn)
    setHandler(mergedIn, new InHandler {
      override def onPush(): Unit = {
        val (passedPrivacy, outWrapper) = grab(mergedIn)
        if (!passedPrivacy || outWrapper.isEmpty)
          push(outFail, akka.NotUsed)
        else
          push(outPass, outWrapper.get)
        pull(mergedIn)
      }
      override def onUpstreamFinish(): Unit = {} // necessary for some reason!
    })
    setHandler(outPass, eagerTerminateOutput)
    setHandler(outFail, eagerTerminateOutput)
  }
}

这个基本思想有效，我可以看到它给了我完全控制过程的可能性，但对于这个超级平凡的决策逻辑，有一个更简单，更高层次的方法来创建一个＆＃34;小部件＆＃ 34;可以包含在我的GraphDSL中吗？

Answer 1

此答案基于akka-stream版本2.4.2-RC1。其他版本的API可能略有不同。 sbt可以使用依赖关系：

libraryDependencies += "com.typesafe.akka" %% "akka-stream" % "2.4.2-RC1"

您可以轻松创建自己的逻辑抽象。您可以将其添加为类型参数，而不是对PassFilter类的类型进行硬编码。而不是硬编码确定输出端口的函数，您可以将其作为构造函数参数传递。通过执行此操作，您将收到可以连接到任意流的可重用组件。幸运的是，Akka已经提供了这样一个组件。它被称为Partition：

val shape = GraphDSL.create() { implicit b ⇒
  import GraphDSL.Implicits._

  val first = b.add(Sink.foreach[Int](elem ⇒ println("even:\t" + elem)))
  val second = b.add(Sink.foreach[Int](elem ⇒ println("odd:\t" + elem)))
  val p = b.add(Partition[Int](2, elem ⇒ if (elem%2 == 0) 0 else 1))

  p ~> first
  p ~> second

  SinkShape(p.in)
}
Source(1 to 5).to(shape).run()

/*
This should print:
odd:    1
even:   2
odd:    3
even:   4
odd:    5
*/

Partition组件还将许多输出端口作为参数，以使其更具可重用性。使用~>符号，您可以将输出端口连接到其他组件，这是我在示例中所做的。您当然可以将其保留，并通过FanOutShape2组件返回两个输出端口。

是否有更高级别的方法来编写自定义GraphStage？

1 个答案: