你好,我是胡夕。今天我们来聊聊 Kafka 中位移提交的那些事。
之前我们说过,Consumer 端有个位移的概念,它和消息在分区中的位移不是一回事儿,虽然它们的英文都是 Offset。今天我们要聊的位移是 Consumer 的消费位移,它记录了 Consumer 要消费的下一条消息的位移。这可能和你以前了解的有些出入,不过切记是下一条消息的位移,而不是目前最新消费消息的位移。
我来举个例子说明一下。假设一个分区中有 10 条消息,位移分别是 0 到 9。某个 Consumer 应用已消费了 5 条消息,这就说明该 Consumer 消费了位移为 0 到 4 的 5 条消息,此时 Consumer 的消费位移是 5,指向了下一条消息的位移。


Consumer 需要向 Kafka 汇报自己的消费位移数据,这个汇报过程被称为提交位移(Committing Offsets)。因为 Consumer 能够同时消费多个分区的数据,所以位移的提交实际上是在分区粒度上进行的,即 Consumer 需要为分配给它的每个分区提交各自的消费位移数据。

提交位移主要是为了表征 Consumer 的消费进度,这样当 Consumer 发生故障重启之后,就能够从 Kafka 中读取之前提交的位移值,然后从相应的位移处继续消费,从而避免整个消费过程重来一遍。换句话说,位移提交是 Kafka 提供给你的一个工具或语义保障,你负责维持这个语义保障,即如果你提交了位移 X,那么 Kafka 会认为所有位移值小于 X 的消息你都已经成功消费了。
这一点特别关键。因为位移提交非常灵活,你完全可以提交任何位移值,但由此产生的后果你也要一并承担。假设你的 Consumer 消费了 10 条消息,

  • 如果你提交的位移值却是 20,那么从理论上讲,位移介于 11~19 之间的消息是有可能丢失的;
  • 相反地,如果你提交的位移值是 5,那么位移介于 5~9 之间的消息就有可能被重复消费。

所以,我想再强调一下,位移提交的语义保障是由你来负责的,Kafka 只会“无脑”地接受你提交的位移。你对位移提交的管理直接影响了你的 Consumer 所能提供的消息语义保障。
鉴于位移提交甚至是位移管理对 Consumer 端的巨大影响,Kafka,特别是 KafkaConsumer API,提供了多种提交位移的方法。

  • 从用户的角度来说,位移提交分为自动提交和手动提交;
  • 从 Consumer 端的角度来说,位移提交分为同步提交和异步提交。

    自动提交 & 手动提交

    我们先来说说自动提交和手动提交。所谓自动提交,就是指 Kafka Consumer 在后台默默地为你提交位移,作为用户的你完全不必操心这些事;而手动提交,则是指你要自己提交位移,Kafka Consumer 压根不管。
    开启自动提交位移的方法很简单。Consumer 端有个参数 enable.auto.commit,把它设置为 true 或者压根不设置它就可以了。因为它的默认值就是 true,即 Java Consumer 默认就是自动提交位移的。如果启用了自动提交,Consumer 端还有个参数就派上用场了:auto.commit.interval.ms。它的默认值是 5 秒,表明 Kafka 每 5 秒会为你自动提交一次位移。
    为了把这个问题说清楚,我给出了完整的 Java 代码。这段代码展示了设置自动提交位移的方法。有了这段代码做基础,今天后面的讲解我就不再展示完整的代码了。
    1. Properties props = new Properties();
    2. props.put("bootstrap.servers", "localhost:9092");
    3. props.put("group.id", "test");
    4. props.put("enable.auto.commit", "true");
    5. props.put("auto.commit.interval.ms", "2000");
    6. props.put("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
    7. props.put("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
    8. KafkaConsumer<String, String> consumer = new KafkaConsumer<>(props);
    9. consumer.subscribe(Arrays.asList("foo", "bar"));
    10. while (true) {
    11. ConsumerRecords<String, String> records = consumer.poll(100);
    12. for (ConsumerRecord<String, String> record : records)
    13. System.out.printf("offset = %d, key = %s, value = %s%n", record.offset(), record.key(), record.value());
    14. }
    上面的橙色粗体部分,就是开启自动提交位移的方法。总体来说,还是很简单的吧。

和自动提交相反的,就是手动提交了。开启手动提交位移的方法就是设置 enable.auto.commit 为 false。但是,仅仅设置它为 false 还不够,因为你只是告诉 Kafka Consumer 不要自动提交位移而已,你还需要调用相应的 API 手动提交位移。
最简单的 API 就是 KafkaConsumer#commitSync()。该方法会提交 KafkaConsumer#poll() 返回的最新位移。从名字上来看,它是一个同步操作,即该方法会一直等待,直到位移被成功提交才会返回。如果提交过程中出现异常,该方法会将异常信息抛出。下面这段代码展示了 commitSync() 的使用方法:

  1. while (true) {
  2. ConsumerRecords<String, String> records =
  3. consumer.poll(Duration.ofSeconds(1));
  4. // 处理消息
  5. process(records);
  6. try {
  7. consumer.commitSync();
  8. } catch (CommitFailedException e) {
  9. // 处理提交失败异常
  10. handle(e);
  11. }
  12. }

可见,调用 consumer.commitSync() 方法的时机,是在你处理完了 poll() 方法返回的所有消息之后。如果你莽撞地过早提交了位移,就可能会出现消费数据丢失的情况。那么你可能会问,自动提交位移就不会出现消费数据丢失的情况了吗?自动提交位移能恰到好处地把握时机进行位移提交吗?为了搞清楚这个问题,我们必须要深入地了解一下自动提交位移的顺序。
一旦设置了 enable.auto.commit 为 true,Kafka 会保证在开始调用 poll 方法时,提交上次 poll 返回的所有消息。从顺序上来说,poll 方法的逻辑是:先提交上一批消息的位移,再处理下一批消息,因此它能保证不出现消费丢失的情况。但自动提交位移的一个问题在于,它可能会出现重复消费。

提交频率指的是:最小的提交间隔。比如默认值是 5s,那么 Kafka 保证的是至少等待 5s 才会自动提交一次,

  • 不可能出现 3s 时提交的情况;
  • 但是可能出现由于 5s 时消息还没有处理完,仍在执行业务代码,那么只有处理完消息后,后续再一次执行 poll 时才会自动提交消费位移的情况,此时自动提交的时间间隔超过 5 s。

自动提交消费位移的代码逻辑是在 KafkaConsumer#poll() 里实现的,

  • 如果时间间隔 > 最小提交间隔,就会进行 offset 提交逻辑
  • 如果时间间隔 < 最小提交间隔,则忽略 offset 提交逻辑,等待下一次执行 poll 再检查时间间隔判断是否执行自动提交消费位移的代码逻辑

上次 poll 的数据即便处理结束,没有调用下一次 poll,那么也不会进行自动提交消费位移。

在默认情况下,Consumer 每 5 秒自动提交一次位移。现在,我们假设提交位移之后的 3 秒发生了 Rebalance 操作。在 Rebalance 之后,所有 Consumer 从上一次提交的位移处继续消费,但该位移已经是 3 秒前的位移数据了,故在 Rebalance 发生前 3 秒消费的所有数据都要重新再消费一次。虽然你能够通过减少 auto.commit.interval.ms 的值来提高提交频率,但这么做只能缩小重复消费的时间窗口,不可能完全消除它。这是自动提交机制的一个缺陷。

反观手动提交位移,它的好处就在于更加灵活,你完全能够把控位移提交的时机和频率。但是,它也有一个缺陷,就是在调用 commitSync() 时,Consumer 程序会处于阻塞状态,直到远端的 Broker 返回提交结果,这个阻塞状态才会结束。在任何系统中,因为程序而非资源限制而导致的阻塞都可能是系统的瓶颈,会影响整个应用程序的 TPS。当然,你可以选择拉长提交间隔,但这样做的后果是 Consumer 的提交频率下降,在下次 Consumer 重启回来后,会有更多的消息被重新消费。

同步提交 & 异步提交

鉴于这个问题,Kafka 社区为手动提交位移提供了另一个 API 方法:KafkaConsumer#commitAsync()。从名字上来看它就不是同步的,而是一个异步操作。调用 commitAsync() 之后,它会立即返回,不会阻塞,因此不会影响 Consumer 应用的 TPS。由于它是异步的,Kafka 提供了回调函数(callback),供你实现提交之后的逻辑,比如记录日志或处理异常等。下面这段代码展示了调用 commitAsync() 的方法:

我的疑问:commitAsync 是异步的,那如果还没有提交消费位移,就 poll 拉取了下一批,为什么不会拉取到重复消息呢?

之后的了解:不会拉取到重复消息,即使不 commit,循环执行 poll 也不会拉取到重复消息。 只要 Consumer 没有重启,就不会发生重复消费。 如果 Consumer 重启了,那么它首先会从位移主题 __consumer_offsets 获取消费位移的位置;之后 Consumer 会维护一个消费位移(和位移主题中的消费位移不同),在 poll 的过程中不断更新;Consumer 在发送拉取消息请求的时候告诉 Broker 它想要的消息的消息起始位移(把它维护的消费位移发送给 Broker)。

使用 KafkaConsumer#commitsync() 或者 commitAsync() 提交的消费位移主要是为了:保证 Consumer 重启之后恢复消费进度。

  1. while (true) {
  2. ConsumerRecords<String, String> records =
  3. consumer.poll(Duration.ofSeconds(1));
  4. // 处理消息
  5. process(records);
  6. consumer.commitAsync((offsets, exception) -> {
  7. if (exception != null)
  8. handle(exception);
  9. });
  10. }

commitAsync 能否替代 commitSync

commitAsync 是否能够替代 commitSync 呢?答案是不能。
commitAsync 的问题在于,出现问题时 commitAsync 不会自动重试。因为它是异步操作,倘若提交失败后自动重试,那么它重试时提交的位移值可能早已经“过期”或不是最新值了。因此,异步提交的重试其实没有意义,所以 commitAsync 是不会重试的。

我的想法:并且还有可能出现新值被旧值覆盖的情况。

显然,如果是手动提交,我们需要将 commitSync 和 commitAsync 组合使用才能到达最理想的效果,原因有两个:

  1. 我们可以利用 commitSync 的自动重试来规避那些瞬时错误,比如网络的瞬时抖动,Broker 端 GC 等。因为这些问题都是短暂的,自动重试通常都会成功,因此,我们不想自己重试,而是希望 Kafka Consumer 帮我们做这件事。
  2. 我们不希望程序总处于阻塞状态,影响 TPS。

我们来看一下下面这段代码,它展示的是如何将两个 API 方法结合使用进行手动提交。

  1. try {
  2. while (true) {
  3. ConsumerRecords<String, String> records =
  4. consumer.poll(Duration.ofSeconds(1));
  5. // 处理消息
  6. process(records);
  7. // 使用异步提交规避阻塞
  8. commitAysnc();
  9. }
  10. } catch (Exception e) {
  11. // 处理异常
  12. handle(e);
  13. } finally {
  14. try {
  15. // 最后一次提交使用同步阻塞式提交
  16. consumer.commitSync();
  17. } finally {
  18. consumer.close();
  19. }
  20. }

这段代码同时使用了 commitSync() 和 commitAsync()。对于常规性、阶段性的手动提交,我们调用 commitAsync() 避免程序阻塞,而在 Consumer 要关闭前,我们调用 commitSync() 方法执行同步阻塞式的位移提交,以确保 Consumer 关闭前能够保存正确的位移数据。将两者结合后,我们既实现了异步无阻塞式的位移管理,也确保了 Consumer 位移的正确性,所以,如果你需要自行编写代码开发一套 Kafka Consumer 应用,那么我推荐你使用上面的代码范例来实现手动的位移提交。
我们说了自动提交和手动提交,也说了同步提交和异步提交,这些就是 Kafka 位移提交的全部了吗?其实,我们还差一部分。

更加细粒度化地提交位移

实际上,Kafka Consumer API 还提供了一组更为方便的方法,可以帮助你实现更精细化的位移管理功能。刚刚我们聊到的所有位移提交,都是提交 poll 方法返回的所有消息的位移,比如 poll 方法一次返回了 500 条消息,当你处理完这 500 条消息之后,前面我们提到的各种方法会一次性地将这 500 条消息的位移一并处理。简单来说,就是直接提交最新一条消息的位移。但如果我想更加细粒度化地提交位移,该怎么办呢?


设想这样一个场景:你的 poll 方法返回的不是 500 条消息,而是 5000 条。那么,你肯定不想把这 5000 条消息都处理完之后再提交位移,因为一旦中间出现差错,之前处理的全部都要重来一遍。这类似于我们数据库中的事务处理。很多时候,我们希望将一个大事务分割成若干个小事务分别提交,这能够有效减少错误恢复的时间。
在 Kafka 中也是相同的道理。对于一次要处理很多消息的 Consumer 而言,它会关心社区有没有方法允许它在消费的中间进行位移提交。比如前面这个 5000 条消息的例子,你可能希望每处理完 100 条消息就提交一次位移,这样能够避免大批量的消息重新消费。
庆幸的是,Kafka Consumer API 为手动提交提供了这样的方法:commitSync(Map) 和 commitAsync(Map)。它们的参数是一个 Map 对象,键就是 TopicPartition,即消费的分区,而值是一个 OffsetAndMetadata 对象,保存的主要是位移数据。
就拿刚刚提过的那个例子来说,如何每处理 100 条消息就提交一次位移呢?在这里,我以 commitAsync 为例,展示一段代码,实际上,commitSync 的调用方法和它是一模一样的。

  1. private Map<TopicPartition, OffsetAndMetadata> offsets = new HashMap<>();
  2. int count = 0;
  3. // ……
  4. while (true) {
  5. ConsumerRecords<String, String> records =
  6. consumer.poll(Duration.ofSeconds(1));
  7. for (ConsumerRecord<String, String> record : records) {
  8. // 处理消息
  9. process(record);
  10. ifcount % 100 == 0){
  11. offsets.put(new TopicPartition(record.topic(), record.partition()),
  12. new OffsetAndMetadata(record.offset() + 1);
  13. // 回调处理逻辑是 null
  14. consumer.commitAsync(offsets, null);
  15. }
  16. count++;
  17. }
  18. }

简单解释一下这段代码。程序先是创建了一个 Map 对象,用于保存 Consumer 消费处理过程中要提交的分区消费位移,之后开始逐条处理消息,并构造要提交的位移值。还记得之前我说过要提交下一条消息的位移吗?这就是这里构造 OffsetAndMetadata 对象时,使用当前消息位移加 1 的原因。代码的最后部分是做位移的提交。我在这里设置了一个计数器,每累计 100 条消息就统一提交一次位移。与调用无参的 commitAsync 不同,这里调用了带 Map 对象参数的 commitAsync 进行细粒度的位移提交。这样,这段代码就能够实现每处理 100 条消息就提交一次位移,不用再受 poll 方法返回的消息总数的限制了。

小结

好了,我们来总结一下今天的内容。Kafka Consumer 的位移提交,是实现 Consumer 端语义保障的重要手段。位移提交分为自动提交和手动提交,而手动提交又分为同步提交和异步提交。
在实际使用过程中,推荐你使用手动提交机制,因为它更加可控,也更加灵活。另外,建议你同时采用同步提交和异步提交两种方式,这样既不影响 TPS,又支持自动重试,改善 Consumer 应用的高可用性。总之,Kafka Consumer API 提供了多种灵活的提交方法,方便你根据自己的业务场景定制你的提交策略。
image.png

开放讨论

实际上,手动提交也不能避免消息重复消费。假设 Consumer 在处理完消息和提交位移前出现故障,下次重启后依然会出现消息重复消费的情况。请你思考一下,如何实现你的业务场景中的去重逻辑呢?