Skip to content

Commit 6341cf2

Browse files
committed
feat: support custom queue_name per message for all kick methods
1 parent e1eca8a commit 6341cf2

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

taskiq_redis/redis_broker.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,9 +251,10 @@ async def kick(self, message: BrokerMessage) -> None:
251251
252252
:param message: message to append.
253253
"""
254+
queue_name = message.labels.get("queue_name") or self.queue_name
254255
async with Redis(connection_pool=self.connection_pool) as redis_conn:
255256
await redis_conn.xadd(
256-
self.queue_name,
257+
queue_name,
257258
{b"data": message.message},
258259
maxlen=self.maxlen,
259260
approximate=self.approximate,

taskiq_redis/redis_sentinel_broker.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,9 +230,10 @@ async def kick(self, message: BrokerMessage) -> None:
230230
231231
:param message: message to append.
232232
"""
233+
queue_name = message.labels.get("queue_name") or self.queue_name
233234
async with self._acquire_master_conn() as redis_conn:
234235
await redis_conn.xadd(
235-
self.queue_name,
236+
queue_name,
236237
{b"data": message.message},
237238
maxlen=self.maxlen,
238239
approximate=self.approximate,

0 commit comments

Comments
 (0)