diff options
author | Dana Powers <dana.powers@gmail.com> | 2018-02-27 09:30:47 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-02-27 09:30:47 -0800 |
commit | 793dc4dd4f1fdce9167a23e08ddb3841ea69805b (patch) | |
tree | 559ddb77ee0d6906c33b6df846f1c7edf937e508 | |
parent | 4cadaafb24c2bdad475a68e3df5a4e19ce043ce7 (diff) | |
download | kafka-python-793dc4dd4f1fdce9167a23e08ddb3841ea69805b.tar.gz |
Validate max_records in KafkaConsumer.poll (#1398)
-rw-r--r-- | kafka/consumer/group.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/kafka/consumer/group.py b/kafka/consumer/group.py index debbd2d..110df55 100644 --- a/kafka/consumer/group.py +++ b/kafka/consumer/group.py @@ -595,6 +595,8 @@ class KafkaConsumer(six.Iterator): assert timeout_ms >= 0, 'Timeout must not be negative' if max_records is None: max_records = self.config['max_poll_records'] + assert isinstance(max_records, int), 'max_records must be an integer' + assert max_records > 0, 'max_records must be positive' # Poll for new data until the timeout expires start = time.time() |