From 13d8911220b544de3d4b3dbfac0253023ebf2a4d Mon Sep 17 00:00:00 2001
From: Aleksey Lobanov <i@likemath.ru>
Date: Sun, 3 Jan 2021 23:35:04 +0300
Subject: [PATCH] rabbit: Fixed auto_ack with generator consumer

---
 src/rabbit.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/rabbit.py b/src/rabbit.py
index 84fac4c..c90862f 100644
--- a/src/rabbit.py
+++ b/src/rabbit.py
@@ -91,5 +91,7 @@ def consume_task(channel, queue: str, timeout=None, auto_ack=True, max_count=Non
         yield body
         if max_count and method.delivery_tag == max_count:
             break
+        if not auto_ack:
+            channel.basic_ack(delivery_tag=method.delivery_tag)
     requeued_messages = channel.cancel()
     _logger.info(f"Requeued {requeued_messages} messages")