AlpakkaKafkaConsumer.committablePartitionedSource是否使用每个分区单独的Kafka消费者?
创始人
2024-08-07 08:00:29
0

通过Alpakka Kafka Consumer提供的committablePartitionedSource方法创建的流在默认情况下将为每个分区创建单独的Kafka消费者。示例如下:

import akka.kafka.scaladsl.Consumer
import akka.kafka.scaladsl.Consumer.Control
import akka.kafka.{ConsumerSettings, Subscriptions}
import akka.stream.scaladsl.{Keep, Sink}
import akka.stream.{ActorMaterializer, Materializer}
import org.apache.kafka.clients.consumer.ConsumerConfig
import org.apache.kafka.common.serialization.{StringDeserializer, StringSerializer}
import akka.kafka.scaladsl.Consumer.Control
import akka.kafka.scaladsl._
import akka.stream.scaladsl._
import org.apache.kafka.common.TopicPartition

import scala.concurrent.duration._
import scala.concurrent.{Await, ExecutionContextExecutor, Future}

object Example extends App {
  implicit val ec: ExecutionContextExecutor = scala.concurrent.ExecutionContext.global
  implicit val system = akka.actor.ActorSystem("example")
  implicit val mat: Materializer = ActorMaterializer.create(system)

  // ----------------------------------------
  // Setup

  val config = system.settings.config.getConfig("akka.kafka.consumer")
  val consumerSettings =
    ConsumerSettings(config, new StringDeserializer, new StringDeserializer)
      .withBootstrapServers("localhost:9092")
      .withGroupId("group1")
      .withProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest")

  val topics = List("test")

  // ----------------------------------------
  // Source creation

  val partition = new TopicPartition("test", /* partition = */ 0)
  val source: Source[(TopicPartition, Source[ConsumerMessage.CommittableMessage[String, String], Consumer.Control]), Consumer.Control] =
    Consumer.committablePartitionedSource(consumerSettings, Subscriptions.assignment(partition))

  // ----------------------------------------
  // Sink creation

  val sinkPartition: ((TopicPartition, Source[ConsumerMessage.CommittableMessage[String, String], Control]), Future[Done]) => Future[Done] = {
    case ((topicPartition, source), streamCompletion) =>
      source.runWith(Sink.ignore)
  }

  // ----------------------------------------
  // Run

  val (consumerControl, streamDone) = source
    .map {
      case (part, source) => (part, source.mapAsync(1)(msg => {
        msg.committableOffset
          .commitScaladsl()
          .map(_ => msg)
      }))
    }
    .toMat(BroadcastHub.sink(bufferSize = 10))(Keep.both)
    .run()

  val sinks = consumerControl.partitionedSink(sinkPartition)
  val subscriptions = consumerControl.subscriptions

  // ----------------------------------------
  // Examples

  // 1. Run stream in background with cancellable control
  val control = consumerControl
    .indexWhere(_.topicPartition == partition)
    .filter(_ >= 0)
    .toMat(Sink.foreach(idx => println(s"Commencing stream for partition $partition ($idx)")))(Keep.right)
    .run()
  // Wait for acknowledgement
  Thread.sleep(1000)
  // Cancel the stream
  control.cancel()

  // 2. Listen for 10 seconds and shut down
  val _ = consumerControl
    .throttle(1, 1.second)
    .takeWithin(10.seconds)
    .toMat(Sink.foreach(println))(Keep.both)
    .run()

  // ----------------------------------------
  // Shutdown

  Await.ready(system.terminate(), Duration.Inf)
}

相关内容

热门资讯

安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
安卓系统拦截短信在哪,安卓系统... 你是不是也遇到了这种情况:手机里突然冒出了很多垃圾短信,烦不胜烦?别急,今天就来教你怎么在安卓系统里...
app安卓系统登录不了,解锁登... 最近是不是你也遇到了这样的烦恼:手机里那个心爱的APP,突然就登录不上了?别急,让我来帮你一步步排查...
安卓系统要维护多久,安卓系统维... 你有没有想过,你的安卓手机里那个陪伴你度过了无数日夜的安卓系统,它究竟要陪伴你多久呢?这个问题,估计...
windows官网系统多少钱 Windows官网系统价格一览:了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌... 手机里的App越来越多,是不是感觉内存不够用了?别急,今天就来教你怎么轻松卸载安卓系统里的App,让...
怎么复制照片安卓系统,操作步骤... 亲爱的手机控们,是不是有时候想把自己的手机照片分享给朋友,或者备份到电脑上呢?别急,今天就来教你怎么...
安卓系统应用怎么重装,安卓应用... 手机里的安卓应用突然罢工了,是不是让你头疼不已?别急,今天就来手把手教你如何重装安卓系统应用,让你的...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...