File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed
Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change 2929from taskiq .acks import AckableMessage
3030from taskiq .decor import AsyncTaskiqDecoratedTask
3131from taskiq .events import TaskiqEvents
32+ from taskiq .exceptions import TaskRejectedError
3233from taskiq .formatters .proxy_formatter import ProxyFormatter
3334from taskiq .message import BrokerMessage
3435from taskiq .result_backends .dummy import DummyResultBackend
@@ -517,7 +518,13 @@ def _register_task(
517518 By default we register tasks in local task registry.
518519 But this behaviour can be changed in subclasses.
519520
521+ This method may raise TaskRejectedError if task has already been registered to a different broker.
522+
520523 :param task_name: Name of a task.
521524 :param task: Decorated task.
522525 """
526+ if task .broker != self :
527+ raise TaskRejectedError (
528+ f"Task already has a different broker ({ task .broker } )" ,
529+ )
523530 self .local_task_registry [task_name ] = task
You can’t perform that action at this time.
0 commit comments