Kafka：指定 offset 消费

超、凢脫俗 2024-03-25 22:49 80阅读 0赞

**注意：每次执行完，需要修改消费者组名；**

import org.apache.kafka.clients.consumer.ConsumerConfig;
    import org.apache.kafka.clients.consumer.ConsumerRecord;
    import org.apache.kafka.clients.consumer.ConsumerRecords;
    import org.apache.kafka.clients.consumer.KafkaConsumer;
    import org.apache.kafka.common.TopicPartition;
    import org.apache.kafka.common.serialization.StringDeserializer;
    
    import java.time.Duration;
    import java.util.ArrayList;
    import java.util.Properties;
    import java.util.Set;
    
    public class CustomConsumerSeek {
    
        public static void main(String[] args) {
    
            // 0 配置信息
            Properties properties = new Properties();
    
            // 连接
            properties.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG,"192.168.25.129:9092,192.168.25.129:9092");
    
            // 反序列化
            properties.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
            properties.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
    
            // 组id
            properties.put(ConsumerConfig.GROUP_ID_CONFIG,"test3");
    
            // 1 创建消费者
            // KafkaConsumer<K, V> 
            // 由于消息形式是 key value 为 "", "hello"
            // 所以泛型K为key为String类型 泛型V为传递消息的类型，此处发送字符串用String类型
            KafkaConsumer<String, String> kafkaConsumer = new KafkaConsumer<>(properties);
    
            // 2 订阅主题
            ArrayList<String> topics = new ArrayList<>();
            topics.add("test");
            kafkaConsumer.subscribe(topics);
    
            // ===================================此处==============================================
    
            // 指定位置进行消费
            // 获取所有分区信息
            Set<TopicPartition> assignment = kafkaConsumer.assignment();
    
            //  保证分区分配方案已经制定完毕 
            // 因为消费者初始化与broker通信进行分区分配需要一定的时间
            while (assignment.size() == 0){
                kafkaConsumer.poll(Duration.ofSeconds(1));
    
                assignment = kafkaConsumer.assignment();
            }
    
            // 指定消费的offset 从600开始
            for (TopicPartition topicPartition : assignment) {
                kafkaConsumer.seek(topicPartition,600);
            }
    
            // ==============================止此================================================
    
            // 3  消费数据
            while (true){
    
                ConsumerRecords<String, String> consumerRecords = kafkaConsumer.poll(Duration.ofSeconds(1));
    
                for (ConsumerRecord<String, String> consumerRecord : consumerRecords) {
    
                    System.out.println(consumerRecord);
                }
            }
        }
    }