diff options
Diffstat (limited to 'kafka/producer/kafka.py')
| -rw-r--r-- | kafka/producer/kafka.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/kafka/producer/kafka.py b/kafka/producer/kafka.py index d5a94ad..f137b4e 100644 --- a/kafka/producer/kafka.py +++ b/kafka/producer/kafka.py @@ -400,8 +400,12 @@ class KafkaProducer(object): log.info('Kafka producer closed') return if timeout is None: - timeout = 999999999 - assert timeout >= 0 + # threading.TIMEOUT_MAX is available in Python3.3+ + timeout = getattr(threading, 'TIMEOUT_MAX', 999999999) + if getattr(threading, 'TIMEOUT_MAX', False): + assert 0 <= timeout <= getattr(threading, 'TIMEOUT_MAX') + else: + assert timeout >= 0 log.info("Closing the Kafka producer with %s secs timeout.", timeout) #first_exception = AtomicReference() # this will keep track of the first encountered exception |