소스 검색

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)