Kafka 0.9版本之前,offset存储在zookeeper,0.9版本之后,默认将offset存储在Kafka的一个内置的topic中。除此之外,Kafka还可以选择自定义存储offset。
    offset的维护是相当繁琐的,因为需要考虑到消费者的Rebalace。
    当有新的消费者加入消费者组、已有的消费者推出消费者组或者所订阅的主题的分区发生变化,就会触发到分区的重新分配,重新分配的过程叫做Rebalance。
    消费者发生Rebalance之后,每个消费者消费的分区就会发生变化。因此消费者要首先获取到自己被重新分配到的分区,并且定位到每个分区最近提交的offset位置继续消费。
    要实现自定义存储offset,需要借助ConsumerRebalanceListener,以下为示例代码,其中提交和获取offset的方法,需要根据所选的offset存储系统自行实现。

    1. package com.consumer;
    2. import org.apache.kafka.clients.consumer.ConsumerRebalanceListener;
    3. import org.apache.kafka.clients.consumer.ConsumerRecord;
    4. import org.apache.kafka.clients.consumer.ConsumerRecords;
    5. import org.apache.kafka.clients.consumer.KafkaConsumer;
    6. import org.apache.kafka.common.TopicPartition;
    7. import java.util.*;
    8. public class CustomConsumer4 {
    9. // 记录分区的信息,这个也可以用redis等等存储. key是主体的名字和分区号
    10. private static Map<TopicPartition, Long> currentOffset = new HashMap<>();
    11. public static void main(String[] args) {
    12. //创建配置信息
    13. Properties props = new Properties();
    14. //Kafka集群
    15. props.put("bootstrap.servers", "zjj101:9092");
    16. //消费者组,只要group.id相同,就属于同一个消费者组
    17. props.put("group.id", "test");
    18. //关闭自动提交offset
    19. props.put("enable.auto.commit", "false");
    20. //Key和Value的反序列化类
    21. props.put("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
    22. props.put("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
    23. //创建一个消费者
    24. KafkaConsumer<String, String> consumer = new KafkaConsumer<>(props);
    25. //消费者订阅主题
    26. consumer.subscribe(Arrays.asList("first"), new ConsumerRebalanceListener() {
    27. /*该方法会在分区之前调用
    28. 分区之前各个消费者应该把offset提交*/
    29. @Override
    30. public void onPartitionsRevoked(Collection<TopicPartition> partitions) {
    31. // 提交旧的offset
    32. commitOffset(currentOffset);
    33. }
    34. /*该方法会在分区之后调用
    35. 分区之后应该给你offset信息重新给你*/
    36. @Override
    37. public void onPartitionsAssigned(Collection<TopicPartition> partitions) {
    38. //获取新的offset
    39. currentOffset.clear();
    40. for (TopicPartition partition : partitions) {
    41. consumer.seek(partition, getOffset(partition));//定位到最近提交的offset位置继续消费
    42. }
    43. }
    44. });
    45. while (true) {
    46. ConsumerRecords<String, String> records = consumer.poll(100);//消费者拉取数据
    47. for (ConsumerRecord<String, String> record : records) {
    48. System.out.printf("offset = %d, key = %s, value = %s%n", record.offset(), record.key(), record.value());
    49. currentOffset.put(new TopicPartition(record.topic(), record.partition()), record.offset());
    50. }
    51. commitOffset(currentOffset);//异步提交
    52. }
    53. }
    54. //获取某分区的最新offset
    55. private static long getOffset(TopicPartition partition) {
    56. return 0;
    57. }
    58. //提交该消费者所有分区的offset
    59. private static void commitOffset(Map<TopicPartition, Long> currentOffset) {
    60. }
    61. }