瀏覽代碼

TaskPool: Support taking the initargs argument to Pool()

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

+ 3 - 1
celery/concurrency/processes/__init__.py

@@ -36,10 +36,11 @@ class TaskPool(object):
 
     def __init__(self, limit, logger=None, initializer=None,
             maxtasksperchild=None, timeout=None, soft_timeout=None,
-            putlocks=True):
+            putlocks=True, initargs=None):
         self.limit = limit
         self.logger = logger or log.get_default_logger()
         self.initializer = initializer
+        self.initargs = initargs
         self.maxtasksperchild = maxtasksperchild
         self.timeout = timeout
         self.soft_timeout = soft_timeout
@@ -54,6 +55,7 @@ class TaskPool(object):
         """
         self._pool = self.Pool(processes=self.limit,
                                initializer=self.initializer,
+                               initargs=self.initargs,
                                timeout=self.timeout,
                                soft_timeout=self.soft_timeout,
                                maxtasksperchild=self.maxtasksperchild)