瀏覽代碼

Envvar C_BENCH_EVERY alias to CELERY_BENCH_EVERY

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

+ 3 - 1
celery/worker/state.py

@@ -65,6 +65,8 @@ def task_ready(request):
 
 
 C_BENCH = os.environ.get('C_BENCH') or os.environ.get('CELERY_BENCH')
+C_BENCH_EVERY = int(os.environ.get('C_BENCH_EVERY') or
+                    os.environ.get('CELERY_BENCH_EVERY') or 1000)
 if C_BENCH:  # pragma: no cover
     import atexit
 
@@ -76,7 +78,7 @@ if C_BENCH:  # pragma: no cover
     bench_first = None
     bench_start = None
     bench_last = None
-    bench_every = int(os.environ.get('CELERY_BENCH_EVERY', 1000))
+    bench_every = C_BENCH_EVERY
     bench_sample = []
     __reserved = task_reserved
     __ready = task_ready