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..20d67fa 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 batch=%s, topic=%s>' % (self.async, self.topic) + 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 batch=%s, topic=%s>' % (self.async, self.topic) |