diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-02-06 11:57:49 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-02-09 10:34:19 -0800 |
| commit | fb03c69544221a25e925b8c6af45e44ac6944b2a (patch) | |
| tree | 3a4f99544c58863cd7ded04c88d4344179e75745 /kafka/partitioner/roundrobin.py | |
| parent | bb731eb89cc58c2e6799ff2223c47011594986f1 (diff) | |
| download | kafka-python-fb03c69544221a25e925b8c6af45e44ac6944b2a.tar.gz | |
make partitions arg optional in *Partitioner.partition, defaulting to self.partitions from init
Diffstat (limited to 'kafka/partitioner/roundrobin.py')
| -rw-r--r-- | kafka/partitioner/roundrobin.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/kafka/partitioner/roundrobin.py b/kafka/partitioner/roundrobin.py index 54d00da..6439e53 100644 --- a/kafka/partitioner/roundrobin.py +++ b/kafka/partitioner/roundrobin.py @@ -15,9 +15,9 @@ class RoundRobinPartitioner(Partitioner): self.partitions = partitions self.iterpart = cycle(partitions) - def partition(self, key, partitions): + def partition(self, key, partitions=None): # Refresh the partition list if necessary - if self.partitions != partitions: + if partitions and self.partitions != partitions: self._set_partitions(partitions) return next(self.iterpart) |