summaryrefslogtreecommitdiff
path: root/kafka/partitioner/hashed.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2015-02-19 23:25:19 -0800
committerDana Powers <dana.powers@gmail.com>2015-02-19 23:25:19 -0800
commit9ad0be662d388b47aadf04d712f5744add6456e3 (patch)
treef520b34f440b7d2e9513b8802841c32b1126a106 /kafka/partitioner/hashed.py
parent60a73788ee9036a79078193dfab892c6e6ef8f9b (diff)
parent6ed6ad53f23fae8dd7ccadc15a7c84f67334d0c0 (diff)
downloadkafka-python-9ad0be662d388b47aadf04d712f5744add6456e3.tar.gz
Merge pull request #314 from dpkp/keyed_producer_failover
Handle keyed producer failover
Diffstat (limited to 'kafka/partitioner/hashed.py')
-rw-r--r--kafka/partitioner/hashed.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/kafka/partitioner/hashed.py b/kafka/partitioner/hashed.py
index 587a3de..fb5e598 100644
--- a/kafka/partitioner/hashed.py
+++ b/kafka/partitioner/hashed.py
@@ -5,7 +5,9 @@ class HashedPartitioner(Partitioner):
Implements a partitioner which selects the target partition based on
the hash of the key
"""
- def partition(self, key, partitions):
+ def partition(self, key, partitions=None):
+ if not partitions:
+ partitions = self.partitions
size = len(partitions)
idx = hash(key) % size