Kafka分区分配策略（3）自定义分区分配策略

原 Kafka分区分配策略（3）——自定义分区分配策略https://blog.csdn.net/u013256816/article/details/81123858版权声明：本文为博主原创文章，未经博主朱小厮允许不得转载。 https://blog.csdn.net/u013256816/article/details/81123858
接上文：
1.【Kafka分区分配策略（1）——RangeAssignor】
2.【Kafka分区分配策略（2）——RoundRobinAssignor和StickyAssignor】
欢迎支持笔者新作：《深入理解Kafka:核心设计与实践原理》和《RabbitMQ实战指南》，同时欢迎关注笔者的微信公众号：朱小厮的博客。
自定义分区分配策略
读者不仅可以任意选用Kafka所提供的3种分配策略，还可以自定义分配策略来实现更多可选的功能。自定义的分配策略必须要实现org.apache.kafka.clients.consumer.internals.PartitionAssignor接口。PartitionAssignor接口的定义如下：
Subscription subscription(Set<String> topics);
String name();
Map<String, Assignment> assign(Cluster metadata, 
                               Map<String, Subscription> subscriptions);
void onAssignment(Assignment assignment);
class Subscription {
    private final List<String> topics;
    private final ByteBuffer userData;
（省略若干方法……）
}
class Assignment {
    private final List<TopicPartition> partitions;
    private final ByteBuffer userData;
（省略若干方法……）
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
PartitionAssignor接口中定义了两个内部类：Subscription和Assignment。
Subscription类用来表示消费者的订阅信息，类中有两个属性