public class RangeAssignor
extends org.apache.kafka.clients.consumer.internals.AbstractPartitionAssignor
The range assignor works on a per-topic basis. For each topic, we lay out the available partitions in numeric order and the consumers in lexicographic order. We then divide the number of partitions by the total number of consumers to determine the number of partitions to assign to each consumer. If it does not evenly divide, then the first few consumers will have one extra partition.
For example, suppose there are two consumers C0
and C1
, two topics t0
and
t1
, and each topic has 3 partitions, resulting in partitions t0p0
, t0p1
,
t0p2
, t1p0
, t1p1
, and t1p2
.
The assignment will be:
C0: [t0p0, t0p1, t1p0, t1p1]
C1: [t0p2, t1p2]
group.instance.id
to make the assignment behavior more sticky.
For the above example, after one rolling bounce, group coordinator will attempt to assign new member.id
towards consumers,
for example C0
-> C3
C1
-> C2
.
The assignment could be completely shuffled to:
C3 (was C0): [t0p2, t1p2] (before was [t0p0, t0p1, t1p0, t1p1])
C2 (was C1): [t0p0, t0p1, t1p0, t1p1] (before was [t0p2, t1p2])
member.id
relative order, and
can be avoided by setting the group.instance.id.
Consumers will have individual instance ids I1
, I2
. As long as
1. Number of members remain the same across generation
2. Static members' identities persist across generation
3. Subscription pattern doesn't change for any member
The assignment will always be:
I0: [t0p0, t0p1, t1p0, t1p1]
I1: [t0p2, t1p2]
Rack-aware assignment is used if both consumer and partition replica racks are available and some partitions have replicas only on a subset of racks. We attempt to match consumer racks with partition replica racks on a best-effort basis, prioritizing balanced assignment over rack alignment. Topics with equal partition count and same set of subscribers guarantee co-partitioning by prioritizing co-partitioning over rack alignment. In this case, aligning partition replicas of these topics on the same racks will improve locality for consumers. For example, if partitions 0 of all topics have a replica on rack 'a', partition 1 on rack 'b' etc., partition 0 of all topics can be assigned to a consumer on rack 'a', partition 1 to a consumer on rack 'b' and so on.
Note that rack-aware assignment currently takes all replicas into account, including any offline replicas and replicas that are not in the ISR. This is based on the assumption that these replicas are likely to join the ISR relatively soon. Since consumers don't rebalance on ISR change, this avoids unnecessary cross-rack traffic for long durations after replicas rejoin the ISR. In the future, we may consider rebalancing when replicas are added or removed to improve consumer rack alignment.
org.apache.kafka.clients.consumer.internals.AbstractPartitionAssignor.MemberInfo
ConsumerPartitionAssignor.Assignment, ConsumerPartitionAssignor.GroupAssignment, ConsumerPartitionAssignor.GroupSubscription, ConsumerPartitionAssignor.RebalanceProtocol, ConsumerPartitionAssignor.Subscription
Modifier and Type | Field and Description |
---|---|
static String |
RANGE_ASSIGNOR_NAME |
Constructor and Description |
---|
RangeAssignor() |
Modifier and Type | Method and Description |
---|---|
Map<String,List<TopicPartition>> |
assign(Map<String,Integer> partitionsPerTopic,
Map<String,ConsumerPartitionAssignor.Subscription> subscriptions) |
Map<String,List<TopicPartition>> |
assignPartitions(Map<String,List<PartitionInfo>> partitionsPerTopic,
Map<String,ConsumerPartitionAssignor.Subscription> subscriptions)
Performs range assignment of the specified partitions for the consumers with the provided subscriptions.
|
String |
name()
Unique name for this assignor (e.g.
|
assign, partitionInfosWithoutRacks, partitions, put, useRackAwareAssignment
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getAssignorInstances, onAssignment, subscriptionUserData, supportedProtocols, version
public static final String RANGE_ASSIGNOR_NAME
public String name()
ConsumerPartitionAssignor
ConsumerConfig.PARTITION_ASSIGNMENT_STRATEGY_CONFIG
public Map<String,List<TopicPartition>> assignPartitions(Map<String,List<PartitionInfo>> partitionsPerTopic, Map<String,ConsumerPartitionAssignor.Subscription> subscriptions)
assignPartitions
in class org.apache.kafka.clients.consumer.internals.AbstractPartitionAssignor
public Map<String,List<TopicPartition>> assign(Map<String,Integer> partitionsPerTopic, Map<String,ConsumerPartitionAssignor.Subscription> subscriptions)
assign
in class org.apache.kafka.clients.consumer.internals.AbstractPartitionAssignor