Skip to content

Commit 41fe84a

Browse files
committed
feat(broker): allow to set queue name dynamically when kicking
1 parent 09e835e commit 41fe84a

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

taskiq_redis/redis_broker.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,9 @@ async def kick(self, message: BrokerMessage) -> None:
6060
6161
:param message: message to send.
6262
"""
63+
queue_name = message.labels.get("queue_name") or self.queue_name
6364
async with Redis(connection_pool=self.connection_pool) as redis_conn:
64-
await redis_conn.publish(self.queue_name, message.message)
65+
await redis_conn.publish(queue_name, message.message)
6566

6667
async def listen(self) -> AsyncGenerator[bytes, None]:
6768
"""
@@ -95,8 +96,9 @@ async def kick(self, message: BrokerMessage) -> None:
9596
9697
:param message: message to append.
9798
"""
99+
queue_name = message.labels.get("queue_name") or self.queue_name
98100
async with Redis(connection_pool=self.connection_pool) as redis_conn:
99-
await redis_conn.lpush(self.queue_name, message.message)
101+
await redis_conn.lpush(queue_name, message.message)
100102

101103
async def listen(self) -> AsyncGenerator[bytes, None]:
102104
"""

0 commit comments

Comments
 (0)