diff options
| author | Dana Powers <dana.powers@gmail.com> | 2016-09-17 18:43:38 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2016-09-19 21:07:14 -0700 |
| commit | db49c8fbdd657b652d1c3715044e6b10af6ec77b (patch) | |
| tree | 9d0cd488cf9d013635c3f3791259386e7d23021b | |
| parent | b05c8da2626bd639455f1ab3c25b84c59ad43690 (diff) | |
| download | kafka-python-max_poll_records.tar.gz | |
Fixup iteratorsmax_poll_records
| -rw-r--r-- | kafka/consumer/fetcher.py | 4 | ||||
| -rw-r--r-- | kafka/consumer/group.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/kafka/consumer/fetcher.py b/kafka/consumer/fetcher.py index 62190fb..15fa1c9 100644 --- a/kafka/consumer/fetcher.py +++ b/kafka/consumer/fetcher.py @@ -448,10 +448,10 @@ class Fetcher(six.Iterator): return self def __next__(self): - ret = self.fetched_records(max_records=1) + ret, _ = self.fetched_records(max_records=1) if not ret: raise StopIteration - assert len(ret.keys()) == 1 + assert len(ret) == 1 (messages,) = ret.values() assert len(messages) == 1 return messages[0] diff --git a/kafka/consumer/group.py b/kafka/consumer/group.py index 5eeaddd..efadde1 100644 --- a/kafka/consumer/group.py +++ b/kafka/consumer/group.py @@ -842,7 +842,7 @@ class KafkaConsumer(six.Iterator): ret = self.poll(timeout_ms=self.config['consumer_timeout_ms'], max_records=1) if not ret: raise StopIteration - assert len(ret.keys()) == 1 + assert len(ret) == 1 (messages,) = ret.values() assert len(messages) == 1 return messages[0] |