From 9f817c6c70a749f2ac449ab4edfd26c6dd8a7410 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Thu, 15 Nov 2018 12:22:11 -0800 Subject: have recent message helper cleanup consumer --- python/fatcat_tools/workers/worker_common.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'python/fatcat_tools/workers') diff --git a/python/fatcat_tools/workers/worker_common.py b/python/fatcat_tools/workers/worker_common.py index daae91b3..e400e815 100644 --- a/python/fatcat_tools/workers/worker_common.py +++ b/python/fatcat_tools/workers/worker_common.py @@ -27,13 +27,17 @@ def most_recent_message(topic): for p, op in consumer._partitions.items()] offsets = [(p, (o if o > -1 else -2)) for p, o in offsets] if -2 in [o for p, o in offsets]: + consumer.stop() return None else: consumer.reset_offsets(offsets) msg = islice(consumer, 1) if msg: - return list(msg)[0].value + val = list(msg)[0].value + consumer.stop() + return val else: + consumer.stop() return None class FatcatWorker: -- cgit v1.2.3