Merge pull request #34 from Matter-Tech/enable-result-backend

Enable Result Backend

Author: ts-matter

matter_task_queue/celery_config.py

@@ -90,7 +90,7 @@ def format(self, record):
 
 def build_celery_config(create_dead_letter_queue: bool):
     sqs_attributes = get_sqs_attributes() if create_dead_letter_queue else {}
-    return {
+    celery_config = {
         "broker_url": Config.CELERY_BROKER_URL,
         "broker_transport_options": {
             "region": Config.AWS_REGION,
@@ -104,13 +104,18 @@ def build_celery_config(create_dead_letter_queue: bool):
         ),
         "task_routes": (route_task,),
         "worker_pool_restarts": True,
-        "task_ignore_result": True,
+        "task_ignore_result": True if not Config.CELERY_RESULT_BACKEND_URL else False,
         "worker_hijack_root_logger": False,
         "worker_enable_remote_control": False,
         "worker_send_task_events": False,
         "task_always_eager": Config.DEBUG and Config.IS_ENV_LOCAL_OR_TEST,
     }
 
+    if Config.CELERY_RESULT_BACKEND_URL:
+        celery_config["result_backend"] = Config.CELERY_RESULT_BACKEND_URL
+
+    return celery_config
+
 
 CELERY_BEAT_CONFIG = {
     "traceback": True,

matter_task_queue/config.py

@@ -19,6 +19,7 @@ def load_variables(cls):
         cls.SENTRY_DSN = os.getenv("SENTRY_DSN")
         cls.AWS_ENDPOINT_URL = os.getenv("AWS_ENDPOINT_URL")
         cls.CELERY_BROKER_URL = os.environ["CELERY_BROKER_URL"] if cls.ENV != "test" else os.getenv("CELERY_BROKER_URL")
+        cls.CELERY_RESULT_BACKEND_URL = os.getenv("CELERY_RESULT_BACKEND_URL")
 
 
 Config()