Kafka 实战 - Kafka Consumer 重置 Offset

在开发测试过程中，可能需要消费一段时间的消息，来验证数据的可靠性，这里需要消费者（Consumer）重置其消费的偏移量（Offset）。

以下是几种常用的方法来重置Kafka Consumer的Offset：

方法一：使用命令行工具（kafka-consumer-groups.sh）

适用于快速手动干预或脚本自动化。

bin/kafka-consumer-groups.sh --bootstrap-server localhost:9092 --group my-group --reset-offsets --to-datetime YYYY-MM-DDTHH:mm:ss.sssZ --all-topics --execute

--bootstrap-server: 指定Kafka集群的地址。
--group: 消费者组的名称。
--reset-offsets: 表示要执行偏移量重置操作。
--to-datetime: 设置重置偏移量的目标时间点。所有在该时间点之前的消息都将被重新消费。
--all-topics: 重置该消费者组订阅的所有Topic的偏移量。
--execute: 直接执行重置操作，不进行交互式确认。

方法二：使用Java AdminClient API

适用于在应用程序代码中动态调整偏移量。

import org.apache.kafka.clients.admin.AdminClient;
import org.apache.kafka.clients.admin.OffsetSpec;
import org.apache.kafka.common.TopicPartition;
import org.apache.kafka.common.serialization.StringDeserializer;import java.time.Instant;
import java.util.*;
import java.util.concurrent.ExecutionException;public class OffsetResetExample {public static void main(String[] args) throws ExecutionException, InterruptedException {Properties adminProps = new Properties();adminProps.put("bootstrap.servers", "localhost:9092");try (AdminClient adminClient = AdminClient.create(adminProps)) {String groupId = "my-group";Instant targetTimestamp = Instant.parse("2024-04-0½T12:00:00Z"); // 替换为目标时间List<TopicPartition> partitions = new ArrayList<>();// 添加需要重置偏移量的Topic和分区，例如：partitions.add(new TopicPartition("my-topic", 0));Map<TopicPartition, OffsetSpec> offsetSpecs = new HashMap<>();for (TopicPartition partition : partitions) {offsetSpecs.put(partition, OffsetSpec.forTimestamp(targetTimestamp));}adminClient.resetOffsets(groupId, offsetSpecs).all().get();System.out.println("Offsets have been reset.");}}
}

创建AdminClient实例，连接到Kafka集群。
定义消费者组ID、目标时间点以及需要重置偏移量的TopicPartition列表。
使用AdminClient.resetOffsets()方法，指定消费者组、偏移量规格（基于目标时间点）以及受影响的TopicPartition，执行偏移量重置操作。

方法三：通过编程方式手动设置偏移量

适用于在消费者代码中直接控制消费起始位置。

import org.apache.kafka.clients.consumer.Consumer;
import org.apache.kafka.clients.consumer.ConsumerRecords;
import org.apache.kafka.clients.consumer.KafkaConsumer;
import org.apache.kafka.common.TopicPartition;import java.time.Duration;
import java.util.Collections;
import java.util.Properties;public class ManualOffsetResetExample {public static void main(String[] args) {Properties props = new Properties();props.put("bootstrap.servers", "localhost:9092");props.put("group.id", "my-group");props.put("key.deserializer", StringDeserializer.class.getName());props.put("value.deserializer", StringDeserializer.class.getName());try (Consumer<String, String> consumer = new KafkaConsumer<>(props)) {TopicPartition tp = new TopicPartition("my-topic", 0);long targetOffset = 12345L; // 替换为目标偏移量consumer.assign(Collections.singletonList(tp));consumer.seek(tp, targetOffset);while (true) {ConsumerRecords<String, String> records = consumer.poll(Duration.ofMillis(100));// 处理记录...}}}
}