diff options
Diffstat (limited to 'kafka/consumer.py')
| -rw-r--r-- | kafka/consumer.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index f2898ad..8a04c2a 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -230,6 +230,10 @@ class SimpleConsumer(Consumer): auto_commit_every_n=auto_commit_every_n, auto_commit_every_t=auto_commit_every_t) + def __repr__(self): + return '<SimpleConsumer group=%s, topic=%s, partitions=%s>' % \ + (self.group, self.topic, str(self.offsets.keys())) + def provide_partition_info(self): """ Indicates that partition info must be returned by the consumer @@ -535,6 +539,10 @@ class MultiProcessConsumer(Consumer): proc.start() self.procs.append(proc) + def __repr__(self): + return '<MultiProcessConsumer group=%s, topic=%s, consumers=%d>' % \ + (self.group, self.topic, len(self.procs)) + def stop(self): # Set exit and start off all waiting consumers self.exit.set() |