Try with global variable

nsoranzo · Sep 17, 2024 · aa9ac04 · aa9ac04
1 parent 4e521fa
commit aa9ac04
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/lib/galaxy/celery/__init__.py b/lib/galaxy/celery/__init__.py
@@ -10,6 +10,7 @@
     Any,
     Callable,
     Dict,
+    Optional,
 )
 
 import pebble
@@ -201,20 +202,26 @@ def wrapper(*args, **kwds):
         return decorate
 
 
-@lru_cache(maxsize=1)
+_celery_app: Optional[GalaxyCelery] = None
+
+
 def get_celery_app() -> GalaxyCelery:
-    celery_app = GalaxyCelery(
+    global _celery_app
+
+    if _celery_app is not None:
+        return _celery_app
+    _celery_app = GalaxyCelery(
         "galaxy",
         include=TASKS_MODULES,
         task_default_queue=DEFAULT_TASK_QUEUE,
         task_create_missing_queues=True,
         timezone="UTC",
     )
-    celery_app.set_default()
+    _celery_app.set_default()
     config = get_config()
-    config_celery_app(config, celery_app)
-    setup_periodic_tasks(config, celery_app)
-    return celery_app
+    config_celery_app(config, _celery_app)
+    setup_periodic_tasks(config, _celery_app)
+    return _celery_app
 
 
 def config_celery_app(config, celery_app):