diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/fatcat_tools/workers/worker_common.py | 6 |
1 files changed, 5 insertions, 1 deletions
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: |