diff --git a/pyinfra/queue/consumer.py b/pyinfra/queue/consumer.py index c21cb26..fec0ee4 100644 --- a/pyinfra/queue/consumer.py +++ b/pyinfra/queue/consumer.py @@ -1,3 +1,5 @@ +import logging + from pyinfra.queue.queue_manager.queue_manager import QueueManager @@ -10,4 +12,5 @@ class Consumer: self.queue_manager.consume_and_publish(self.callback) def consume(self, **kwargs): + logging.info(kwargs) return self.queue_manager.consume(**kwargs) diff --git a/pyinfra/queue/queue_manager/pika_queue_manager.py b/pyinfra/queue/queue_manager/pika_queue_manager.py index 7e80040..984b29f 100644 --- a/pyinfra/queue/queue_manager/pika_queue_manager.py +++ b/pyinfra/queue/queue_manager/pika_queue_manager.py @@ -44,7 +44,12 @@ def monkey_patch_queue_handle(channel, queue) -> QueueHandle: def get_connection_params(): credentials = pika.PlainCredentials(username=CONFIG.rabbitmq.user, password=CONFIG.rabbitmq.password) - kwargs = {"host": CONFIG.rabbitmq.host, "port": CONFIG.rabbitmq.port, "credentials": credentials} + kwargs = { + "host": CONFIG.rabbitmq.host, + "port": CONFIG.rabbitmq.port, + "credentials": credentials, + "heartbeat": CONFIG.rabbitmq.heartbeat, + } parameters = pika.ConnectionParameters(**kwargs) return parameters