diff options
| author | Martin Olveyra <molveyra@gmail.com> | 2015-06-29 13:28:12 -0200 |
|---|---|---|
| committer | Martin Olveyra <molveyra@gmail.com> | 2015-06-29 13:28:12 -0200 |
| commit | 43164ad096cf3dc9f252d0c6e4377ff025bb01f0 (patch) | |
| tree | 72d31fe3c05b2a33c96b21d9068237154ae43523 /kafka/consumer/multiprocess.py | |
| parent | adbd4ac052e4a5b40cfc2a3589b7adbcb656afe5 (diff) | |
| download | kafka-python-43164ad096cf3dc9f252d0c6e4377ff025bb01f0.tar.gz | |
allow to retrieve partition info in mp consumer
Diffstat (limited to 'kafka/consumer/multiprocess.py')
| -rw-r--r-- | kafka/consumer/multiprocess.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/kafka/consumer/multiprocess.py b/kafka/consumer/multiprocess.py index d03eb95..bd784cf 100644 --- a/kafka/consumer/multiprocess.py +++ b/kafka/consumer/multiprocess.py @@ -257,7 +257,8 @@ class MultiProcessConsumer(Consumer): except Empty: break - messages.append(message) + _msg = (partition, message) if self.partition_info else message + messages.append(_msg) new_offsets[partition] = message.offset + 1 count -= 1 if timeout is not None: |