Add configurable user-visible log message for memory errors

ecmwf-projects · Feb 3, 2025 · a8cd3d6 · a8cd3d6
1 parent 755cba3
commit a8cd3d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/cads_broker/config.py b/cads_broker/config.py
@@ -42,6 +42,7 @@ class BrokerConfig(pydantic_settings.BaseSettings):
     broker_max_dismissed_requests: int = 100
     broker_cancel_stuck_requests_cache_ttl: int = 60
     broker_stuck_requests_limit_minutes: int = 15
+    broker_memory_error_user_visible_log: str = "Worker has been killed due to memory usage."
 
 
 class SqlalchemySettings(pydantic_settings.BaseSettings):

diff --git a/cads_broker/dispatcher.py b/cads_broker/dispatcher.py
@@ -378,7 +378,7 @@ def set_request_error_status(
                         db.add_event(
                             event_type="user_visible_error",
                             request_uid=request_uid,
-                            message="Worker has been killed due to memory usage. ",
+                            message=CONFIG.broker_memory_error_user_visible_log,
                             session=session,
                         )
                         request = db.set_request_status(