0

我正在使用Akka Streams Kafka，并且正在寻找一种方法来执行以下操作：

  • 从偏移量启动流x
  • 依次消费项目x, x+1, x+2.. 直到最后一个项目
  • 消费完最后一项后，完成流。

代码看起来像

Consumer
  .plainSource(consumerSettings, subscription)
  .runForeach(println("got record!"))
  .onComplete {
    case Success(_) => // all items read
    case Failure(error) => // error
  }

它会在最后一个元素被读取后完成。也许这不是这个库的使用方式。我怎样才能做到这一点？

4

1 回答 1

3

Akka Consumer 以“拉”的方式工作，除非发生与代理连接的错误，否则它将永远存在。但是，你什么时候认为流结束了？Kafka 可以被视为一个分布式日志，您可以从其中读取给定偏移量的消息。只要您的客户端连接到代理，您的客户端就会启动并运行......如果您考虑在一段时间内没有来自 Kafka 的事件（例如）时您的流终止，您可以使用idleTimeout：

  Consumer
    .plainSource(consumerSettings, subscription)
    .idleTimeout(10 seconds)
    .runForeach(e => println("E"))
    .onComplete {
      case Success(_) => // all items read
      case Failure(error) =>
      // TimeoutException if no element in ten seconds the stream stops throwing this exception
    }

另一种可能性是使用扇入阶段，特别是MergePreferred。我们可以创建另一个按时间间隔发出事件的 Tick Source。Kafka 源将具有优先权，因此只要元素来自 Kafka，舞台总是会从该源中提取元素。如果某个时间间隔内没有元素，则会将“超时”字符串推送到下游。就像是：

  implicit val actorSystem = ActorSystem("test-actor-system")
  implicit val streamMaterializer = ActorMaterializer()
  implicit val ec = actorSystem.dispatcher

  val consumer =
  Consumer
    .plainSource(consumerSettings, subscription)
    .map(_.value())

  val tick = Source.tick(50 millis, 30 seconds, "Timeout")

  val source = GraphDSL.create(consumer, tick)(Keep.both) { implicit b ⇒
    (r1, r2) ⇒
      val merge = b.add(MergePreferred[String](1, false))
      r2 ~> merge.in(0)
      r1 ~> merge.preferred
      SourceShape(merge.out)
  }

  Source
    .fromGraph(source)
    .takeWhile(el => el != "Timeout")
    .runForeach(msg => println(msg))
  .onComplete{
    case Success(_) => println("Stream ended")
    case Failure(error) => println("There was an error")
  }

使用 takeWhile 流将处于活动状态，同时有来自 Kafka 的元素。

这只是一种方法。Akka Stream 有许多不同的阶段，而 Graph Api 可能以更优雅的方式面对这些情况。

于 2018-04-03T20:44:16.773 回答