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

set AMPQ heartbeat

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-24 13:19:10 +01:00
parent 9f4dfb9781
commit 301ba05d66
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyinfra/queue/consumer.py
View File

@ -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)

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

@ -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