瀏覽代碼

Autoscale: Adds envvar AUTOSCALE_KEEPALIVE

Ask Solem 12 年之前
父節點
當前提交
9e6d44f8ce
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      celery/worker/autoscale.py

+ 4 - 1
celery/worker/autoscale.py

@@ -14,6 +14,7 @@
 from __future__ import absolute_import
 from __future__ import with_statement
 
+import os
 import threading
 
 from functools import partial
@@ -29,6 +30,8 @@ from .hub import DummyLock
 logger = get_logger(__name__)
 debug, info, error = logger.debug, logger.info, logger.error
 
+AUTOSCALE_KEEPALIVE = int(os.environ.get('AUTOSCALE_KEEPALIVE', 30))
+
 
 class WorkerComponent(StartStopComponent):
     name = 'worker.autoscaler'
@@ -65,7 +68,7 @@ class WorkerComponent(StartStopComponent):
 class Autoscaler(bgThread):
 
     def __init__(self, pool, max_concurrency,
-                 min_concurrency=0, keepalive=30, mutex=None):
+                 min_concurrency=0, keepalive=AUTOSCALE_KEEPALIVE, mutex=None):
         super(Autoscaler, self).__init__()
         self.pool = pool
         self.mutex = mutex or threading.Lock()