From 122af3428238412aec84c0e2a48b735dff73f441 Mon Sep 17 00:00:00 2001 From: GefMar Date: Wed, 16 Apr 2025 17:04:27 +0200 Subject: [PATCH] add: delay Float type --- taskiq_aio_pika/broker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/taskiq_aio_pika/broker.py b/taskiq_aio_pika/broker.py index b6aede9..35422a9 100644 --- a/taskiq_aio_pika/broker.py +++ b/taskiq_aio_pika/broker.py @@ -250,7 +250,7 @@ async def kick(self, message: BrokerMessage) -> None: ), } - delay: Optional[int] = parse_val(int, message.labels.get("delay")) + delay: Optional[float] = parse_val(float, message.labels.get("delay")) rmq_message: Message = Message(**message_base_params) if delay is None: @@ -260,7 +260,7 @@ async def kick(self, message: BrokerMessage) -> None: ) await exchange.publish(rmq_message, routing_key=message.task_name) elif self._delayed_message_exchange_plugin: - rmq_message.headers["x-delay"] = delay * 1000 + rmq_message.headers["x-delay"] = int(delay * 1000) exchange = await self.write_channel.get_exchange( self._delay_plugin_exchange_name, )