diff --git a/src/rabbitmq_handler.py b/src/rabbitmq_handler.py index 1d32649..5cf8c0e 100644 --- a/src/rabbitmq_handler.py +++ b/src/rabbitmq_handler.py @@ -18,8 +18,9 @@ RABBITMQ_URL = f"amqp://{rmq_username}:{rmq_password}@{rmq_host}/{rmq_vhost}" class RabbitMQProducer: - def __init__(self, url=RABBITMQ_URL, exchange_name=rmq_exchange): + def __init__(self, url=RABBITMQ_URL, exchange_name=rmq_exchange,rmq_host=rmq_host): self.url = url + self.rmq_host = rmq_host self.exchange_name = exchange_name self.connection: aio_pika.RobustConnection | None = None self.channel: aio_pika.RobustChannel | None = None @@ -28,10 +29,9 @@ class RabbitMQProducer: self._flush_task: asyncio.Task | None = None self._closing = False self._ready = asyncio.Event() - logger.debug(f"[RabbitMQ] Initialized producer for exchange '{self.exchange_name}'") async def connect(self): - logger.info(f"[RabbitMQ] Connecting to {self.url}...") + logger.info(f"[RabbitMQ] Connecting to {self.rmq_host}...") self.connection = await aio_pika.connect_robust(self.url) self.channel = await self.connection.channel(publisher_confirms=True) self.exchange = await self.channel.declare_exchange(