diff options
| -rw-r--r-- | kafka/producer/future.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/kafka/producer/future.py b/kafka/producer/future.py index 52c4ffc..5a7a9dc 100644 --- a/kafka/producer/future.py +++ b/kafka/producer/future.py @@ -25,7 +25,8 @@ class FutureProduceResult(Future): return ret def await(self, timeout=None): - return self._latch.wait(timeout) + # wait() on python2.6 returns None instead of the flag value + return self._latch.wait(timeout) or self._latch.is_set() class FutureRecordMetadata(Future): |