我正在使用 Akka Streams 并使用 Alpakka 从文件中流式传输内容。一段时间后我需要停止流，所以我想使用KillSwitch。但是我不知道怎么用，因为我用的是graph DSL。
我的图表如下所示：
val graph = RunnableGraph.fromGraph(GraphDSL.create() { implicit builder: GraphDSL.Builder[NotUsed] =>
import GraphDSL.Implicits._
source ~> mainFlow ~> sink
ClosedShape
})
graph.run()
我在这里找到了解决方案：How to abruptly stop an akka stream Runnable Graph?
但是，如果我使用图形 DSL，我不知道如何应用它。你能给我一些建议吗？
【问题讨论】：
标签： scala stream akka-stream
要在 GraphDSL 中显示物化值，您可以将物化到该值的阶段传递给 create 方法。用一个例子更容易解释。在你的情况下：
val switch = KillSwitches.single[Int]
val graph: RunnableGraph[UniqueKillSwitch] =
RunnableGraph.fromGraph(GraphDSL.create(switch) { implicit builder: GraphDSL.Builder[UniqueKillSwitch] => sw =>
import GraphDSL.Implicits._
source ~> mainFlow ~> sw ~> sink
ClosedShape
})
val ks = graph.run()
ks.shutdown()
【讨论】：
.take(n)
take(n)，并选择一个在完成时实现为Future 的接收器（例如Sink.ignore）。然后，您可以将任何回调附加到您在运行图表时返回的未来。