summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2018-02-27 09:30:47 -0800
committerGitHub <noreply@github.com>2018-02-27 09:30:47 -0800
commit793dc4dd4f1fdce9167a23e08ddb3841ea69805b (patch)
tree559ddb77ee0d6906c33b6df846f1c7edf937e508
parent4cadaafb24c2bdad475a68e3df5a4e19ce043ce7 (diff)
downloadkafka-python-793dc4dd4f1fdce9167a23e08ddb3841ea69805b.tar.gz
Validate max_records in KafkaConsumer.poll (#1398)
-rw-r--r--kafka/consumer/group.py2
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()