在 Akka Streams Sink 之后获取原始元素的参考？

Question

我正在尝试使用 Amqp alpakka 连接器作为源和接收器。

Source<CommittableReadResult, NotUsed> AMQP_SOURCE  
      -> processAndGetResponse 
      -> Sink<ByteString, CompletionStage<Done>> AMQP_SINK 

Sink的操作成功后，我想确认从Amqp队列中获取的消息。 像这样：

amqp_source(committableReadResult) 
    -> processAndGetResponse 
    -> amqp_sink 
    -> IfSinkOperationSuccess.Then(committableReadResult.ack())

我怎样才能做到这一点？ 我基本上只想在接收器操作成功后才将消息标记为确认。

Answer 1

一般来说，如果你有一个Sink实现了Future[Done] （Scala API，我相信 Java 等价物是CompletionStage[Done] ），你可以在mapAsync阶段中使用该Sink运行一个子流。 数据只有在成功时才会通过。

通过快速阅读 Alpakka AMQP API， Sink都以这种方式实现，因此该方法将起作用。

假设您的doSomethingWithMessage函数导致CommittableReadResult和WriteMessage ，Scala 中的以下内容将起作用：

def doSomethingWithMessage(in: CommittableReadResult): (CommittableReadResult, WriteMessage) = ???

amqpSource
  .map(doSomethingWithMessage)
  .mapAsync(parallelism) { tup =>
    val (crr, wm) = tup
    val fut = Source.single(crr, wm).runWith(amqpSink)
    fut.flatMap(_ => crr.ack().map(_ => crr.message))
  }.runWith(Sink.ignore)