alex/assets-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

rabbit: Fixed auto_ack with generator consumer

Browse Source
This commit is contained in:
Aleksey Lobanov
2021-01-03 23:35:04 +03:00
parent 668e4b5044
commit 13d8911220
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/rabbit.py
View File

@@ -91,5 +91,7 @@ def consume_task(channel, queue: str, timeout=None, auto_ack=True, max_count=Non
yield body yield body
if max_count and method.delivery_tag == max_count: if max_count and method.delivery_tag == max_count:
break break
if not auto_ack:
channel.basic_ack(delivery_tag=method.delivery_tag)
requeued_messages = channel.cancel() requeued_messages = channel.cancel()
_logger.info(f"Requeued {requeued_messages} messages") _logger.info(f"Requeued {requeued_messages} messages")