瀏覽代碼

Added TaskSet.Publisher: Can be used to override the publisher class used to send subtasks

Ask Solem 14 年之前
父節點
當前提交
19ad25dfa6
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      celery/task/sets.py

+ 3 - 1
celery/task/sets.py

@@ -125,6 +125,8 @@ class TaskSet(UserList):
         >>> list_of_return_values = taskset_result.join()
 
     """
+    Publisher = TaskPublisher
+
     _task = None                                                # compat
     _task_name = None                                           # compat
 
@@ -182,7 +184,7 @@ class TaskSet(UserList):
             return self.apply()
 
         taskset_id = gen_unique_id()
-        publisher = TaskPublisher(connection=connection)
+        publisher = self.Publisher(connection=connection)
         try:
             results = [task.apply_async(taskset_id=taskset_id,
                                         publisher=publisher)