Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

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

Browse Source
This commit is contained in:
Viktor Seifert 2022-07-08 15:46:11 +02:00
parent 12ebd186b2
commit bf4c85a0ab
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyinfra/queue/queue_manager/pika_queue_manager.py
View File

@ -143,8 +143,14 @@ class PikaQueueManager(QueueManager):
message = (frame, properties, body) message = (frame, properties, body)
return self.publish_response(message, visitor) return self.publish_response(message, visitor)
self.channel.basic_consume(self._input_queue, callback) consumer_tag = None
self.channel.start_consuming()
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): def clear(self):
try: try: