summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHannu Valtonen <hannu.valtonen@aiven.io>2017-06-20 03:09:43 +0300
committerDana Powers <dana.powers@gmail.com>2017-06-19 17:09:43 -0700
commitb8da199b54b26728f302aaccc5dd0433a8671759 (patch)
tree431c8fd0038c06b50e735ffbfe22064ea2d5a6ab
parentb1cc966439a65f8be1b3973b16753dfba2b51c37 (diff)
downloadkafka-python-b8da199b54b26728f302aaccc5dd0433a8671759.tar.gz
producer: Set exit timeout to 0 for atexit handler to match __del__ (#1126)
Hit a problem with pytest hitting the atexit handler and waiting for close() timeout forever at teardown. This commit makes atexit close() equivalent to __del__ behavior, namely using timeout of 0 for close() completion. If you need a longer timeout you should be setting it explicitly.
-rw-r--r--kafka/producer/kafka.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/kafka/producer/kafka.py b/kafka/producer/kafka.py
index 1f78c89..bd891c9 100644
--- a/kafka/producer/kafka.py
+++ b/kafka/producer/kafka.py
@@ -371,7 +371,7 @@ class KafkaProducer(object):
_self = weakref.proxy(self)
def wrapper():
try:
- _self.close()
+ _self.close(timeout=0)
except (ReferenceError, AttributeError):
pass
return wrapper