用于akka流Source.actorRef缓冲区和OverflowStrategy的接收器折叠

时间:2017-08-30 02:29:40

标签: scala akka akka-stream

以下是akka documentation

的代码段
val sinkUnderTest = Flow[Int].map(_.toString).toMat(Sink.fold("")(_ + _))(Keep.right)

val (ref, future) = Source.actorRef(3, OverflowStrategy.fail)
  .toMat(sinkUnderTest)(Keep.both).run()

ref ! 1
ref ! 2
ref ! 3
ref ! akka.actor.Status.Success("done")

val result = Await.result(future, 3.seconds)
assert(result == "123")

这是一个正常工作的代码段,但是,如果我使用ref来告诉ref ! 4之类的其他消息,我会遇到像akka.stream.BufferOverflowException: Buffer overflow (max capacity was: 3)这样的例外

我猜缓冲区大小3应该足够了。折叠操作的原因是(acc,ele)=> acc,所以它需要累加器和元素来返回新的值累加器。

所以我改变了代码让另一个演员告诉等待3秒。它再次运作。

  val sinkUnderTest = Flow[Int].map(_.toString).toMat(Sink.fold("")(_ + _))(Keep.right)

  private val (ref, future): (ActorRef, Future[String]) = Source.actorRef(3, OverflowStrategy.backpressure).toMat(sinkUnderTest)(Keep.both).run()

  ref ! 1
  ref ! 2
  ref ! 3
  Thread.sleep(3000)
  ref ! 4
  ref ! akka.actor.Status.Success("done")

  val result = Await.result(future, 10.seconds)

  println(result)

但是,我的问题是,我们可以告诉Akka流减速或等待接收器可用。我也在使用OverflowStrategy.backpressure,但它说Backpressure overflowStrategy not supported

还有其他选择吗?

1 个答案:

答案 0 :(得分:5)

您应该将Source.queue视为一种以背压方式从外部将元素注入流中的方法。

Source.queue将具体化为您可以提供元素的队列对象,但是当您提供元素时,您将获得一个Future,当流准备好接受该消息时,它将完成。

以下示例:

  val sinkUnderTest = Flow[Int].map(_.toString).toMat(Sink.fold("")(_ + _))(Keep.right)

  val (queue, future): (SourceQueueWithComplete[Int], Future[String]) =
    Source.queue(3, OverflowStrategy.backpressure).toMat(sinkUnderTest)(Keep.both).run()

  Future.sequence(Seq(
    queue.offer(1),
    queue.offer(2),
    queue.offer(3),
    queue.offer(4)
  ))

  queue.complete()

  val result = Await.result(future, 10.seconds)

  println(result)

docs中的更多信息。