fix(worker): added worker lost wait in worker

hotosm · Feb 10, 2025 · d0d0034 · d0d0034
1 parent 26538a8
commit d0d0034
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/API/api_worker.py b/API/api_worker.py
@@ -20,6 +20,7 @@
 from src.config import CELERY_BROKER_URL as celery_broker_uri
 from src.config import CELERY_RESULT_BACKEND as celery_backend
 from src.config import (
+    CELERY_WORKER_LOST_WAIT,
     DEFAULT_HARD_TASK_LIMIT,
     DEFAULT_README_TEXT,
     DEFAULT_SOFT_TASK_LIMIT,
@@ -51,6 +52,7 @@
 celery.conf.broker_url = celery_broker_uri
 celery.conf.result_backend = celery_backend
 celery.conf.broker_heartbeat = CELERY_BROKER_HEARTBEAT
+celery.conf.worker_lost_wait = CELERY_WORKER_LOST_WAIT
 # celery.conf.task_serializer = "pickle"
 # celery.conf.result_serializer = "json"
 # celery.conf.accept_content = ["application/json", "application/x-python-serialize"]

diff --git a/src/config.py b/src/config.py
@@ -55,6 +55,10 @@ def get_bool_env_var(key, default=False):
 CELERY_BROKER_HEARTBEAT = os.environ.get("CELERY_BROKER_HEARTBEAT") or config.get(
     "CELERY", "CELERY_BROKER_HEARTBEAT", fallback=120
 )
+CELERY_WORKER_LOST_WAIT = os.environ.get("CELERY_WORKER_LOST_WAIT") or config.get(
+    "CELERY", "CELERY_WORKER_LOST_WAIT ", fallback=10
+)
+
 WORKER_PREFETCH_MULTIPLIER = int(
     os.environ.get("WORKER_PREFETCH_MULTIPLIER")
     or config.get("CELERY", "WORKER_PREFETCH_MULTIPLIER", fallback=1)