diff options
author | Omar Ghishan <omar.ghishan@rd.io> | 2013-12-18 17:56:29 -0800 |
---|---|---|
committer | Omar Ghishan <omar.ghishan@rd.io> | 2014-01-06 15:14:50 -0800 |
commit | 5dd8d81c9e47ee21c22945b90221c67baa7852b9 (patch) | |
tree | 2d6826cab807242327d302755940e59fad2f077a /kafka/consumer.py | |
parent | 4d6bafae7ece327ccccc8f2e42ffef1eed061096 (diff) | |
download | kafka-python-5dd8d81c9e47ee21c22945b90221c67baa7852b9.tar.gz |
Reset consumer fields to original values rather than defaults in FetchContext
Diffstat (limited to 'kafka/consumer.py')
-rw-r--r-- | kafka/consumer.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index 5d2775d..600c8c7 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -40,6 +40,8 @@ class FetchContext(object): def __enter__(self): """Set fetch values based on blocking status""" + self.orig_fetch_max_wait_time = self.consumer.fetch_max_wait_time + self.orig_fetch_min_bytes = self.consumer.fetch_min_bytes if self.block: self.consumer.fetch_max_wait_time = self.timeout self.consumer.fetch_min_bytes = 1 @@ -47,9 +49,9 @@ class FetchContext(object): self.consumer.fetch_min_bytes = 0 def __exit__(self, type, value, traceback): - """Reset values to default""" - self.consumer.fetch_max_wait_time = FETCH_MAX_WAIT_TIME - self.consumer.fetch_min_bytes = FETCH_MIN_BYTES + """Reset values""" + self.consumer.fetch_max_wait_time = self.orig_fetch_max_wait_time + self.consumer.fetch_min_bytes = self.orig_fetch_min_bytes class Consumer(object): |