经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » Java相关 » Scala » 查看文章
alpakka-kafka(2)-consumer
来源:cnblogs  作者:雪川大虫  时间:2021/3/1 9:16:14  对本文有异议

   alpakka-kafka-consumer的功能描述很简单:向kafka订阅某些topic然后把读到的消息传给akka-streams做业务处理。在kafka-consumer的实现细节上,为了达到高可用、高吞吐的目的,topic又可用划分出多个分区partition。分区是分布在kafka集群节点broker上的。由于一个topic可能有多个partition,对应topic就会有多个consumer,形成一个consumer组,共用统一的groupid。一个partition只能对应一个consumer、而一个consumer负责从多个partition甚至多个topic读取消息。kafka会根据实际情况将某个partition分配给某个consumer,即partition-assignment。所以一般来说我们会把topic订阅与consumer-group挂钩。这个可以在典型的ConsumerSettings证实:

  1. val system = ActorSystem("kafka-sys")
  2. val config = system.settings.config.getConfig("akka.kafka.consumer")
  3. val bootstrapServers = "localhost:9092"
  4. val consumerSettings =
  5. ConsumerSettings(config, new StringDeserializer, new ByteArrayDeserializer)
  6. .withBootstrapServers(bootstrapServers)
  7. .withGroupId("group1")
  8. .withProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest")

我们先用一个简单的consumer plainSource试试把前一篇示范中producer写入kafka的消息读出来: 

  1. import akka.actor.ActorSystem
  2. import akka.kafka._
  3. import akka.kafka.scaladsl._
  4. import akka.stream.{RestartSettings, SystemMaterializer}
  5. import akka.stream.scaladsl.{Keep, RestartSource, Sink}
  6. import org.apache.kafka.clients.consumer.{ConsumerConfig, ConsumerRecord}
  7. import org.apache.kafka.common.serialization.{ByteArrayDeserializer, StringDeserializer}
  8.  
  9. import scala.concurrent._
  10. import scala.concurrent.duration._
  11. object plain_source extends App {
  12. val system = ActorSystem("kafka-sys")
  13. val config = system.settings.config.getConfig("akka.kafka.consumer")
  14. implicit val mat = SystemMaterializer(system).materializer
  15. implicit val ec: ExecutionContext = mat.executionContext
  16. val bootstrapServers = "localhost:9092"
  17. val consumerSettings =
  18. ConsumerSettings(config, new StringDeserializer, new StringDeserializer)
  19. .withBootstrapServers(bootstrapServers)
  20. .withGroupId("group1")
  21. .withProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest")
  22.  
  23. val subscription = Subscriptions.topics("greatings")
  24. Consumer.plainSource(consumerSettings, subscription)
  25. .runWith(Sink.foreach(msg => println(msg.value())))
  26.  
  27. scala.io.StdIn.readLine()
  28. system.terminate()
  29.  
  30. }

以上我们没有对读出的消息做任何的业务处理,直接显示出来。注意每次都会从头完整读出,因为设置了 .withProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest"),也就是kafka-consumer的auto.offset.reset = "earliest" 。那么如果需要用读出的数据进行业务处理的话,每次开始运行应用时都会重复从头执行这些业务。所以需要某种机制来标注已经读取的消息,也就是需要记住当前读取位置offset。

Consumer.plainSource输入ConsumerRecord类型:

  1. public ConsumerRecord(String topic,
  2. int partition,
  3. long offset,
  4. K key,
  5. V value) {
  6. this(topic, partition, offset, NO_TIMESTAMP, TimestampType.NO_TIMESTAMP_TYPE,
  7. NULL_CHECKSUM, NULL_SIZE, NULL_SIZE, key, value);
  8. }

这个ConsumerRecord类型里包括了offset,用户可以自行commit这个位置参数,也就是说用户可以选择把这个offset存储在kafka或者其它的数据库里。说到commit-offset,offset管理机制在kafka-consumer业务应用中应该属于关键技术。kafka-consumer方面的业务流程可以简述为:从kafka读出业务指令,执行指令并更新业务状态,然后再从kafka里读出下一批指令。为了实现业务状态的准确性,无论错过一些指令或者重复执行一些指令都是不能容忍的。所以,必须准确的标记每次从kafka读取数据后的指针位置,commit-offset。但是,如果读出数据后即刻commit-offset,那么在执行业务指令时如果系统发生异常,那么下次再从标注的位置开始读取数据时就会越过一批业务指令。这种情况称为at-most-once,即可能会执行一次,但绝不会重复。另一方面:如果在成功改变业务状态后再commit-offset,那么,一旦执行业务指令时发生异常而无法进行commit-offset,下次读取的位置将使用前一次的标注位置,就会出现重复改变业务状态的情况,这种情况称为at-least-once,即一定会执行业务指令,但可能出现重复更新情况。如果涉及资金、库存等业务,两者皆不可接受,只能采用exactly-once保证一次这种模式了。不过也有很多业务要求没那么严格,比如某个网站统计点击量,只需个约莫数,无论at-least-once,at-most-once都可以接受。

kafka-consumer-offset是一个Long类型的值,可以存放在kafka内部或者外部的数据库里。如果选择在kafka内部存储offset, kafka配置里可以设定按时间间隔自动进行位置标注,自动把当前offset存入kafka里。当我们在上面例子的ConsumerSettings里设置自动commit后,多次重新运行就不会出现重复数据的情况了:

  1. val consumerSettings =
  2. ConsumerSettings(config, new StringDeserializer, new StringDeserializer)
  3. .withBootstrapServers(bootstrapServers)
  4. .withGroupId("group1")
  5. .withProperty(ConsumerConfig.ENABLE_AUTO_COMMIT_CONFIG, "true") //自动commit
  6. .withProperty(ConsumerConfig.AUTO_COMMIT_INTERVAL_MS_CONFIG, "1000") //自动commit间隔
  7. .withProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest")

alpakka-kafka提供了Committer类型,是akka-streams的Sink或Flow组件,负责把offset写入kafka。如果用Committer的Sink或Flow就可以按用户的需要控制commit-offset的发生时间。如下面这段示范代码: 

 

  1. val committerSettings = CommitterSettings(system)
  2.  
  3. val control: DrainingControl[Done] =
  4. Consumer
  5. .committableSource(consumerSettings, Subscriptions.topics("greatings"))
  6. .mapAsync(10) { msg =>
  7. BusinessLogic.runBusiness(msg.record.key, msg.record.value)
  8. .map(_ => msg.committableOffset)
  9. }
  10. .toMat(Committer.sink(committerSettings))(DrainingControl.apply)
  11. .run()
  12.  
  13. control.drainAndShutdown();
  14.  
  15. scala.io.StdIn.readLine()
  16. system.terminate()
  17.  
  18. }
  19. object BusinessLogic {
  20. def runBusiness(key: String, value: String): Future[Done] = Future.successful(Done)
  21. }

上面这个例子里BusinessLogic.runBusiess()模拟一段业务处理代码,也就是说完成了业务处理之后就用Committer.sink进行了commit-offset。这是一种at-least-once模式,因为runBusiness可能会发生异常失败,所以有可能出现重复运算的情况。Consumer.committableSource输出CommittableMessage: 

  1. def committableSource[K, V](settings: ConsumerSettings[K, V],
  2. subscription: Subscription): Source[CommittableMessage[K, V], Control] =
  3. Source.fromGraph(new CommittableSource[K, V](settings, subscription))
  4.  
  5.  
  6.  
  7. final case class CommittableMessage[K, V](
  8. record: ConsumerRecord[K, V],
  9. committableOffset: CommittableOffset
  10. )
  11.  
  12. @DoNotInherit sealed trait CommittableOffset extends Committable {
  13. def partitionOffset: PartitionOffset
  14. }

Committer.sink接受输入Committable类型并将之写入kafka,上游的CommittableOffset 继承了 Committable。另外,这个DrainingControl类型结合了Control类型和akka-streams终结信号可以有效控制整个consumer-streams安全终结。

alpakka-kafka还有一个atMostOnceSource。这个Source组件每读一条数据就会立即自动commit-offset:

  1. def atMostOnceSource[K, V](settings: ConsumerSettings[K, V],
  2. subscription: Subscription): Source[ConsumerRecord[K, V], Control] =
  3. committableSource[K, V](settings, subscription).mapAsync(1) { m =>
  4. m.committableOffset.commitInternal().map(_ => m.record)(ExecutionContexts.sameThreadExecutionContext)
  5. }

可以看出来,atMostOnceSource在输出ConsumerRecord之前就进行了commit-offset。atMostOnceSource的一个具体使用示范如下:

  1. import scala.collection.immutable
  2. val control: DrainingControl[immutable.Seq[Done]] =
  3. Consumer
  4. .atMostOnceSource(consumerSettings, Subscriptions.topics("greatings"))
  5. .mapAsync(1)(record => BussinessLogic.runBusiness(record.key, record.value()))
  6. .toMat(Sink.seq)(DrainingControl.apply)
  7. .run()
  8.  
  9. control.drainAndShutdown();
  10. scala.io.StdIn.readLine()
  11. system.terminate()

所以,使用atMostOnceSource后是不需要任何Committer来进行commit-offset的了。值得注意的是atMostOnceSource是对每一条数据进行位置标注的,所以运行效率必然会受到影响,如果要求不是那么严格的话还是启动自动commit比较合适。

对于任何类型的交易业务系统来说,无论at-least-once或at-most-once都是不可接受的,只有exactly-once才妥当。实现exactly-once的其中一个方法是把offset与业务数据存放在同一个外部数据库中。如果在外部数据库通过事务处理机制(transaction-processing)把业务状态更新与commit-offset放在一个事务单元中同进同退就能实现exactly-once模式了。下面这段是官方文档给出的一个示范:

  1. val db = new mongoldb
  2. val control = db.loadOffset().map { fromOffset =>
  3. Consumer
  4. .plainSource(
  5. consumerSettings,
  6. Subscriptions.assignmentWithOffset(
  7. new TopicPartition(topic, /* partition = */ 0) -> fromOffset
  8. )
  9. )
  10. .mapAsync(1)(db.businessLogicAndStoreOffset)
  11. .toMat(Sink.seq)(DrainingControl.apply)
  12. .run()
  13. }
  14.  
  15. class mongoldb {
  16. def businessLogicAndStoreOffset(record: ConsumerRecord[String, String]): Future[Done] = // ...
  17. def loadOffset(): Future[Long] = // ...
  18. }

在上面这段代码里:db.loadOffset()从mongodb里取出上一次读取位置,返回Future[Long],然后用Subscriptions.assignmentWithOffset把这个offset放在一个tuple (TopicPartition,Long)里。TopicPartition定义如下: 

  1. public TopicPartition(String topic, int partition) {
  2. this.partition = partition;
  3. this.topic = topic;
  4. }

这样Consumer.plainSource就可以从offset开始读取数据了。plainSource输出ConsumerRecord类型:

  1. public ConsumerRecord(String topic,
  2. int partition,
  3. long offset,
  4. K key,
  5. V value) {
  6. this(topic, partition, offset, NO_TIMESTAMP, TimestampType.NO_TIMESTAMP_TYPE,
  7. NULL_CHECKSUM, NULL_SIZE, NULL_SIZE, key, value);
  8. }

这里面除业务指令value外还提供了当前offset。这些已经足够在businessLogicAndStoreOffset()里运算一个单独的business+offset事务了(transaction)。 

 

 

 

 

 

 

 

 

原文链接:http://www.cnblogs.com/tiger-xc/p/14432566.html

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号