diff options
| -rw-r--r-- | kafka/conn.py | 1 | ||||
| -rw-r--r-- | kafka/partitioner.py | 3 |
2 files changed, 3 insertions, 1 deletions
diff --git a/kafka/conn.py b/kafka/conn.py index 7798a75..29efbf1 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -17,6 +17,7 @@ class KafkaConnection(local): since the Kafka API includes a correlation id. """ def __init__(self, host, port, bufsize=4096): + super(KafkaConnection, self).__init__() self.host = host self.port = port self.bufsize = bufsize diff --git a/kafka/partitioner.py b/kafka/partitioner.py index 84db4d5..817e416 100644 --- a/kafka/partitioner.py +++ b/kafka/partitioner.py @@ -31,7 +31,8 @@ class RoundRobinPartitioner(Partitioner): in a round robin fashion """ def __init__(self, partitions): - self._set_partitions(partitions) + super(RoundRobinPartitioner, self).__init__(partitions) + self.iterpart = cycle(partitions) def _set_partitions(self, partitions): self.partitions = partitions |