diff options
Diffstat (limited to 'kafka/producer.py')
| -rw-r--r-- | kafka/producer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/kafka/producer.py b/kafka/producer.py index e1f4a22..fb83da6 100644 --- a/kafka/producer.py +++ b/kafka/producer.py @@ -258,9 +258,9 @@ class SimpleProducer(Producer): if self.random_start: num_partitions = len(self.client.topic_partitions[topic]) for _ in xrange(random.randint(0, num_partitions-1)): - self.partition_cycles[topic].next() + next(self.partition_cycles[topic]) - return self.partition_cycles[topic].next() + return next(self.partition_cycles[topic]) def send_messages(self, topic, *msg): partition = self._next_partition(topic) |