Kafka支持以下几种分区策略:

1. 随机(Default)

生产者随机选择一个分区。
这种策略相对简单,但不能保证负载均衡。

举例

## www.itzhimei.com 代码段
public class DefaultPartitioner implements Partitioner {

   public void partition(String topic, Object key, byte[] keyBytes,
        Object value, byte[] valueBytes, Cluster cluster) {

       List<PartitionInfo> partitions = cluster.partitionsForTopic(topic);
       int numPartitions = partitions.size();
       return new Random().nextInt(numPartitions);  
   }
}

2. 轮询(RoundRobin)

生产者依次选择每个分区。
可以保证负载均衡,但不能考虑分区大小。

举例

## www.itzhimei.com 代码段
public class RoundRobinPartitioner implements Partitioner {

   private AtomicInteger nextIndex = new AtomicInteger(0);  

   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 = nextIndex.getAndIncrement() % numPartitions;
       return partitions.get(nextValue).partition();
   }
}

3. 哈希值

根据 key 的哈希值选择分区。
可以基于key实现部分负载均衡。

4.自定义

完全由用户自定义分区策略。
可以考虑很多方面如:分区大小、负载情况等。

以上是Kafka主要支持的分区策略:

1. 随机策略
2. 轮询策略
3. 根据key的哈希值
4. 完全自定义

自定义分区策略可以实现更高效的负载均衡。

Kafka支持随机、轮询、根据哈希值选择分区以及完全自定义分区策略。自定义策略可以实现更复杂的负载均衡。