diff options
| author | Zack Dever <zack.dever@rd.io> | 2013-12-12 17:03:35 -0800 |
|---|---|---|
| committer | Zack Dever <zack.dever@rd.io> | 2013-12-12 17:03:35 -0800 |
| commit | 8d26368433dffb65ff8d144203fc59d8162974fe (patch) | |
| tree | 55fbc1b127f65646318684c4fef5925bf09c4923 /kafka/consumer.py | |
| parent | cc96e5c89aa65427cb0851e21a47cffa675987dc (diff) | |
| download | kafka-python-8d26368433dffb65ff8d144203fc59d8162974fe.tar.gz | |
allow for timeout to be None in SimpleConsumer.get_messages
Diffstat (limited to 'kafka/consumer.py')
| -rw-r--r-- | kafka/consumer.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index f2898ad..57b5b97 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -294,7 +294,8 @@ class SimpleConsumer(Consumer): iterator = self.__iter__() # HACK: This splits the timeout between available partitions - timeout = timeout * 1.0 / len(self.offsets) + if timeout: + timeout = timeout * 1.0 / len(self.offsets) with FetchContext(self, block, timeout): while count > 0: |