openstack/deb-python-kafka
Code Issues Proposed changes

Fixup iterators

Browse Source
This commit is contained in:
Dana Powers
2016-09-17 18:43:38 -07:00
parent b05c8da262
commit db49c8fbdd
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kafka/consumer/fetcher.py
View File

@@ -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]

2
kafka/consumer/group.py
View File

@@ -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]