Kafka使用详解-Consumer API（手动提交offset）-手动维护offset(事务维护)

手动维护offset

需要依赖

    <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-tx</artifactId>
            <version>5.0.8.RELEASE</version>
        </dependency>

通过事务的方式维护消费者消费的数据的offset和提交offset的统一性

package com.ln.kafka.custom;

import org.apache.kafka.clients.consumer.*;
import org.apache.kafka.common.TopicPartition;
import org.apache.kafka.common.serialization.StringDeserializer;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;

import java.util.Arrays;
import java.util.Collection;
import java.util.Properties;

/**
 * @ProjectName: kafka
 * @Package: com.ln.kafka.custom
 * @Name:CustomOffsetCustomer
 * @Author:linianest
 * @CreateTime:2021/1/8 11:22
 * @version:1.0
 * @Description TODO: 手动维护offset
 */
public class CustomOffsetCustomer {
    public static void main(String[] args) {
        Properties props = new Properties();
        props.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG, "hadoop101:9092");
        props.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
        props.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
        props.put(ConsumerConfig.GROUP_ID_CONFIG, "1234");
        // 自动提交offset,关闭默认,自动提交效率可能慢
        props.put(ConsumerConfig.ENABLE_AUTO_COMMIT_CONFIG, "false");
        // 默认自动提交offset的时间是5秒
//        props.put(ConsumerConfig.AUTO_COMMIT_INTERVAL_MS_CONFIG, 1000);

        // 创建对象
        KafkaConsumer<String, String> consumer = new KafkaConsumer<>(props);
        // 自定义维护offset
        consumer.subscribe(Arrays.asList("first"), new ConsumerRebalanceListener() {
            // todo 重新分配分区前，提交当前负责的分区的offset
            @Override
            public void onPartitionsRevoked(Collection<TopicPartition> collection) {
                System.out.println("=======回收的分区============");
                for (TopicPartition partition : collection) {
                    System.out.println("partition=" + partition);
                }
            }

            // todo 重新分配分区后，定位新分配的分区的offset
            @Override
            public void onPartitionsAssigned(Collection<TopicPartition> partitions) {
                System.out.println("=======重新得到的分区============");
                for (TopicPartition partition : partitions) {
                    System.out.println("partition=" + partition);
                    // todo 从offset的存储地方获取值,redis或者mysql
                    Long offset = getPartitionOffset(partition);
                    consumer.seek(partition, offset);
                }
            }


        });
        while (true) {
            // 拉取数据
            ConsumerRecords<String, String> records = consumer.poll(100);
            for (ConsumerRecord record : records) {
                // todo 为了避免漏消费或者重复消费的问题，开启个事务
                //===================================
                // 消费数据
                ConsumMessage(record);
                //===================================
            }

        }
    }

    // 消费数据
    @Transactional(rollbackFor = Exception.class)
    private static void ConsumMessage(ConsumerRecord record) {
        System.out.printf("topic=%d,key=%s,value=%s%n", record.topic(), record.offset(), record.value());
        try{
            // 自己维护offset
            TopicPartition topicPartition = new TopicPartition(record.topic(), record.partition());
            // todo 提交offset
            commitOffset(topicPartition, record.offset() + 1);
        }catch (Exception e){
            //设置手动回滚
            TransactionAspectSupport.currentTransactionStatus()
                    .setRollbackOnly();
        }
    }

    /**
     * todo 提交offset,可以将offset维护到其他设备，如mysql或者redis中
     *
     * @param topicPartition
     * @param l
     */
    private static void commitOffset(TopicPartition topicPartition, long l) {
    }

    /**
     * todo 获取分区offset
     *
     * @param partition
     * @return
     */
    private static Long getPartitionOffset(TopicPartition partition) {
        return null;
    }
    // 关闭数据流
}