在 Akka 流中连接两个流

Question

我试图连接两个流，但我无法解释我的实现的输出。

val source = Source(1 to 10)
val sink = Sink.foreach(println)

val flow1 = Flow[Int].map(s => s + 1)
val flow2 = Flow[Int].map(s => s * 10)

val flowGraph = Flow.fromGraph(
    GraphDSL.create() { implicit builder =>
      import GraphDSL.Implicits._

      val concat = builder.add(Concat[Int](2))
      val broadcast = builder.add(Broadcast[Int](2))

      broadcast ~> flow1 ~> concat.in(0)
      broadcast ~> flow2 ~> concat.in(1)

      FlowShape(broadcast.in, concat.out)
    }
  )

source.via(flowGraph).runWith(sink)

我希望此代码有以下输出。

2
3
4
.
.
.
11
10
20
.
.
.
100

相反，我只看到打印了“2”。 你能解释一下我的实现有什么问题，我应该如何更改程序以获得所需的输出。

Answer 1

来自 Akka Stream 的 API 文档：

Concat ：

当当前流有可用元素时发出； 如果当前输入完成，它会尝试下一个

Broadcast ：

当所有输出停止反压并且有可用的输入元素时发出

这两个运算符不会一起工作，因为它们的工作方式存在冲突Concat尝试在切换到另一个输出之前从Broadcast的输出中提取所有元素，而Broadcast除非有需求否则不会发出对于它的所有输出。

对于您需要的内容，您可以按照评论者的建议使用concat连接：

source.via(flow1).concat(source.via(flow2)).runWith(sink)

或等效地，使用Source.combine如下：

Source.combine(source.via(flow1), source.via(flow2))(Concat[Int](_)).runWith(sink)

Answer 2

使用GraphDSL ，它是Source.combine实现的简化版本：

val sg = Source.fromGraph(
  GraphDSL.create(){ implicit builder =>
    import GraphDSL.Implicits._

    val concat = builder.add(Concat[Int](2))

    source ~> flow1 ~> concat
    source ~> flow2 ~> concat

    SourceShape(concat.out)
  }
)

sg.runWith(sink)