RED-4564: POC-test to see if cancelling the consumer prevents messages from being stuck

pyinfra/queue/queue_manager/pika_queue_manager.py

@@ -143,8 +143,14 @@ class PikaQueueManager(QueueManager):
             message = (frame, properties, body)
             return self.publish_response(message, visitor)
 
-        self.channel.basic_consume(self._input_queue, callback)
+        consumer_tag = None
+
+        try:
+            consumer_tag = self.channel.basic_consume(self._input_queue, callback)
             self.channel.start_consuming()
+        finally:
+            if consumer_tag:
+                self.channel.basic_cancel(consumer_tag)
 
     def clear(self):
         try:

src/serve.py

@@ -40,6 +40,8 @@ def make_callback(analysis_endpoint):
 
 
 def main():
+    logger = logging.getLogger("main")
+
     show_banner()
 
     webserver = Process(target=run_probing_webserver, args=(set_up_probing_webserver(),))
@@ -58,6 +60,7 @@ def main():
             consumer = Consumer(visitor, queue_manager)
             consumer.basic_consume_and_publish()
         except Exception as err:
+            logger.exception(err)
             raise ConsumerError from err
 
     try: