在流内发送actorRefWithAck

时间:2018-06-17 07:04:03

标签: scala akka akka-stream

我使用this线程的答案,因为我需要特别对待第一个元素。问题是,我需要将这些数据发送到另一个Actor或本地持久化(这是不可能的)。

所以,我的流看起来像这样:

val flow: Flow[Message, Message, (Future[Done], Promise[Option[Message]])] = Flow.fromSinkAndSourceMat(
  Flow[Message].mapAsync[Trade](1) {
    case TextMessage.Strict(text) =>
      Unmarshal(text).to[Trade]
    case streamed: TextMessage.Streamed =>
      streamed.textStream.runFold("")(_ ++ _).flatMap(Unmarshal(_).to[Trade])
  }.groupBy(pairs.size, _.s).prefixAndTail(1).flatMapConcat {
    case (head, tail) =>
      // sending first element here
      val result = Source(head).to(Sink.actorRefWithAck(
        ref = actor,
        onInitMessage = Init,
        ackMessage = Ack,
        onCompleteMessage = "done"
      )).run()
      // some kind of operation on the result
    Source(head).concat(tail)
  }.mergeSubstreams.toMat(sink)(Keep.right),
  Source.maybe[Message])(Keep.both)

这是一个好习惯吗?它会产生意想不到的后果吗?不幸的是,我无法在流内调用persist,因此我想将此数据发送到外部系统。

1 个答案:

答案 0 :(得分:1)

您当前的方法不会以任何方式使用result,因此更简单的替代方法是触发并忘记演员的第一个Message

groupBy(pairs.size, _.s).prefixAndTail(1).flatMapConcat {
  case (head, tail) =>
    // sending first element here
    actor ! head.head

    Source(head).concat(tail)
}

演员不必担心处理Init和发送Ack消息,而只关心持续Message个实例。