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

set dead letter queue related params

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-15 09:46:39 +01:00
parent f8b60aad16
commit 4c7c4b466e
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pyinfra/pyampq/queue_manager/pika_queue_manager.py
View File

@ -63,12 +63,19 @@ def attempts_remain(n_attempts, max_attempts):
class PikaQueueManager(QueueManager):
def __init__(self, input_queue, output_queue):
def __init__(self, input_queue, output_queue, dead_letter_queue=None):
super().__init__(input_queue, output_queue)
self.connection = get_connection()
self.channel = self.connection.channel()
self.channel.queue_declare(input_queue)
self.channel.queue_declare(output_queue)
if not dead_letter_queue:
dead_letter_queue = CONFIG.rabbitmq.queues.dead_letter
args = {"x-dead-letter-exchange": "", "x-dead-letter-routing-key": dead_letter_queue}
self.channel.queue_declare(input_queue, arguments=args, auto_delete=False, durable=False)
self.channel.queue_declare(output_queue, arguments=args, auto_delete=False, durable=False)
def republish(self, body, n_current_attempts, frame):
self.channel.basic_publish(