浏览代码

install_rdb_handler now supports envvar keyword to specify the envvar used

Ask Solem 13 年之前
父节点
当前提交
17b0406513
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      celery/apps/worker.py

+ 2 - 2
celery/apps/worker.py

@@ -339,14 +339,14 @@ def install_cry_handler(logger):
         platforms.signals["SIGUSR1"] = cry_handler
         platforms.signals["SIGUSR1"] = cry_handler
 
 
 
 
-def install_rdb_handler():  # pragma: no cover
+def install_rdb_handler(envvar="CELERY_RDBSIG"):  # pragma: no cover
 
 
     def rdb_handler(signum, frame):
     def rdb_handler(signum, frame):
         """Signal handler setting a rdb breakpoint at the current frame."""
         """Signal handler setting a rdb breakpoint at the current frame."""
         from celery.contrib import rdb
         from celery.contrib import rdb
         rdb.set_trace(frame)
         rdb.set_trace(frame)
 
 
-    if os.environ.get("CELERY_RDBSIG"):
+    if os.environ.get(envvar):
         platforms.signals["SIGUSR2"] = rdb_handler
         platforms.signals["SIGUSR2"] = rdb_handler