普通生产者Kafka流的自定义分割器

Question

我有一个kafka streams应用程序

props.put(ProducerConfig.PARTITIONER_CLASS_CONFIG, MyPartitioner.class);

或

props.put(ProducerConfig.PARTITIONER_CLASS_CONFIG, RoundRobinPartitioner.class);

这是一个类，用于将消息分发到不同的分区，即使在kafka 2.4版本中使用相同的键

RoundRobinPartitioner具有以下实现：

public class RoundRobinPartitioner implements Partitioner {
    private final ConcurrentMap<String, AtomicInteger> topicCounterMap = new ConcurrentHashMap();

    public RoundRobinPartitioner() {
    }

    public void configure(Map<String, ?> configs) {
    }

    public int partition(String topic, Object key, byte[] keyBytes, Object value, byte[] valueBytes, Cluster cluster) {
        List<PartitionInfo> partitions = cluster.partitionsForTopic(topic);
        int numPartitions = partitions.size();
        int nextValue = this.nextValue(topic);
        List<PartitionInfo> availablePartitions = cluster.availablePartitionsForTopic(topic);
        if (!availablePartitions.isEmpty()) {
            int part = Utils.toPositive(nextValue) % availablePartitions.size();
            return ((PartitionInfo)availablePartitions.get(part)).partition();
        } else {
            return Utils.toPositive(nextValue) % numPartitions;
        }
    }

    private int nextValue(String topic) {
        AtomicInteger counter = (AtomicInteger)this.topicCounterMap.computeIfAbsent(topic, (k) -> {
            return new AtomicInteger(0);
        });
        return counter.getAndIncrement();
    }

    public void close() {
    }
}

我的分区器由完全相同的代码组成，但分区方法实现不同，我的代码块是：

    public int partition(String topic, Object key, byte[] keyBytes, Object value, byte[] valueBytes, Cluster cluster) {
        List<PartitionInfo> partitions = cluster.partitionsForTopic(topic);

        int numPartitions = partitions.size();

        int nextValue = nextValue(topic);

        return Utils.toPositive(nextValue) % numPartitions;

    }

当我这样配置时，消息在两种实现中都被分发到不同的分区，但决不使用某些分区。

我有50个分区，而分区14和34从未收到消息。我的分区不是没有价值的。它们是可用的。当我将返回分区方法更改为14或34时，所有消息都会转到该分区。会有什么问题？这两种实现都没有按预期工作。

编辑1：我试过普通制作人的RoundRobinPartitioner。结果是一样的。生产者不能在多个分区之间平等地生成消息，有些分区从未使用过。原因可能是什么？它不像是缺少的配置。

编辑2：我有调试RoundRobin分区，并把一个断点在返回。当我只产生一条消息时，生产者产生两次消息。第一次尝试总是不成功，并且消息不会进入任何分区。当我在调试时点击继续时，并发映射的索引会增加1。制作人的第二次尝试是成功的。

分区（）方法被调用的东西，我还找不到。

编辑3：这可能与我没有覆盖的onNewBatch方法有关吗？

编辑4：这个实现适用于kafka客户端2.2，但不适用于2.4。分区接口没有onNewBatch方法。当键为空2.2 vs 2.4时，默认分区程序实现会更改。它能与棍子隔断有关吗？

Answer 1

使用UniformStickyPartitioner。在kafka 2.4客户端版本中初始化。圆机器人切割器。类适用于Kafka2.2或更低版本。在2.4版本中

props.put(ProducerConfig.PARTITIONER_CLASS_CONFIG, UniformStickyPartitioner.class);

应该使用。我认为这与新的StickPartitioner有关。