فهرست منبع

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)