diff options
Diffstat (limited to 'kafka/producer.py')
| -rw-r--r-- | kafka/producer.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/kafka/producer.py b/kafka/producer.py index 7ef7896..8275701 100644 --- a/kafka/producer.py +++ b/kafka/producer.py @@ -198,6 +198,9 @@ class SimpleProducer(Producer): partition = self.next_partition.next() return super(SimpleProducer, self).send_messages(partition, *msg) + def __repr__(self): + return '<SimpleProducer topic=%s, batch=%s>' % (self.topic, self.async) + class KeyedProducer(Producer): """ @@ -239,3 +242,6 @@ class KeyedProducer(Producer): partitions = self.client.topic_partitions[self.topic] partition = self.partitioner.partition(key, partitions) return self.send_messages(partition, msg) + + def __repr__(self): + return '<KeyedProducer topic=%s, batch=%s>' % (self.topic, self.async) |