Skip to content

Commit b983751

Browse files
warn if task broker doesn't match self
1 parent 4967239 commit b983751

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

taskiq/schedule_sources/label_based.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ async def get_schedules(self) -> List["ScheduledTask"]:
3030
for task_name, task in self.broker.get_all_tasks().items():
3131
if task.broker != self.broker:
3232
# if task broker doesn't match self, something is probably wrong
33+
logger.warning(f"Broker for {task_name} ({task.broker}) doesn't match self ({self.broker})")
3334
continue
3435
for schedule in task.labels.get("schedule", []):
3536
if "cron" not in schedule and "time" not in schedule:
@@ -62,7 +63,10 @@ def post_send(self, scheduled_task: ScheduledTask) -> None:
6263
return # it's scheduled task with cron label, do not remove this trigger.
6364

6465
for task_name, task in self.broker.get_all_tasks().items():
65-
if task.broker != self.broker or scheduled_task.task_name != task_name:
66+
if task.broker != self.broker:
67+
# if task broker doesn't match self, something is probably wrong
68+
logger.warning(f"Broker for {task_name} ({task.broker}) doesn't match self ({self.broker})")
69+
if scheduled_task.task_name != task_name:
6670
continue
6771

6872
schedule_list = task.labels.get("schedule", []).copy()

0 commit comments

Comments
 (0)