From c8be93b44bb0939dd512a72be578d42a4d7426b7 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Tue, 16 Feb 2016 12:35:28 -0800 Subject: Add RangePartitionAssignor (and use as default); add assignor tests --- kafka/coordinator/consumer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'kafka/coordinator/consumer.py') diff --git a/kafka/coordinator/consumer.py b/kafka/coordinator/consumer.py index a393d7e..515377a 100644 --- a/kafka/coordinator/consumer.py +++ b/kafka/coordinator/consumer.py @@ -8,6 +8,7 @@ import time import six from .base import BaseCoordinator +from .assignors.range import RangePartitionAssignor from .assignors.roundrobin import RoundRobinPartitionAssignor from .protocol import ( ConsumerProtocolMemberMetadata, ConsumerProtocolMemberAssignment, @@ -30,7 +31,7 @@ class ConsumerCoordinator(BaseCoordinator): 'enable_auto_commit': True, 'auto_commit_interval_ms': 5000, 'default_offset_commit_callback': lambda offsets, response: True, - 'assignors': (RoundRobinPartitionAssignor,), + 'assignors': (RangePartitionAssignor, RoundRobinPartitionAssignor), 'session_timeout_ms': 30000, 'heartbeat_interval_ms': 3000, 'retry_backoff_ms': 100, @@ -54,7 +55,7 @@ class ConsumerCoordinator(BaseCoordinator): trigger custom actions when a commit request completes. assignors (list): List of objects to use to distribute partition ownership amongst consumer instances when group management is - used. Default: [RoundRobinPartitionAssignor] + used. Default: [RangePartitionAssignor, RoundRobinPartitionAssignor] heartbeat_interval_ms (int): The expected time in milliseconds between heartbeats to the consumer coordinator when using Kafka's group management feature. Heartbeats are used to ensure -- cgit v1.2.1