feat(dask): update info command to include autoscaler info (reanahub#701

) closes (reanahub#834)
Alputer · Sep 20, 2024 · 189b8d8 · 189b8d8
1 parent 73ab0ea
commit 189b8d8
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/reana_server/config.py b/reana_server/config.py
@@ -61,6 +61,9 @@
 DASK_ENABLED = strtobool(os.getenv("DASK_ENABLED", "true"))
 """Whether dask is enabled in the cluster or not"""
 
+DASK_AUTOSCALER_ENABLED = strtobool(os.getenv("DASK_AUTOSCALER_ENABLED", "true"))
+"""Whether dask autoscaler is enabled in the cluster or not"""
+
 REANA_DASK_CLUSTER_MAX_CORES_LIMIT = float(
     os.getenv("REANA_DASK_CLUSTER_MAX_CORES_LIMIT", 8)
 )

diff --git a/reana_server/rest/info.py b/reana_server/rest/info.py
@@ -25,6 +25,7 @@
     REANA_KUBERNETES_JOBS_MAX_USER_TIMEOUT_LIMIT,
     REANA_INTERACTIVE_SESSION_MAX_INACTIVITY_PERIOD,
     DASK_ENABLED,
+    DASK_AUTOSCALER_ENABLED,
     REANA_DASK_CLUSTER_MAX_CORES_LIMIT,
     REANA_DASK_CLUSTER_MAX_MEMORY_LIMIT,
     REANA_DASK_CLUSTER_DEFAULT_CORES_LIMIT,
@@ -139,6 +140,13 @@ def info(user, **kwargs):  # noqa
                   value:
                     type: string
                 type: object
+              dask_autoscaler_enabled:
+                properties:
+                  title:
+                    type: string
+                  value:
+                    type: string
+                type: object
               dask_cluster_default_cores_limit:
                 properties:
                   title:
@@ -236,6 +244,10 @@ def info(user, **kwargs):  # noqa
                     "title": "Dask workflows allowed in the cluster",
                     "value": "False"
                 },
+                "dask_autoscaler_enabled": {
+                    "title": "Dask autoscaler is used for the dask clusters",
+                    "value": "False"
+                },
                 "dask_cluster_default_cores_limit": {
                     "title": "Default cores limit for dask clusters",
                     "value": "4"
@@ -318,6 +330,10 @@ def info(user, **kwargs):  # noqa
             ),
         )
         if DASK_ENABLED:
+            cluster_information["dask_autoscaler_enabled"] = dict(
+                title="Dask autoscaler is used for the dask clusters",
+                value=bool(DASK_AUTOSCALER_ENABLED),
+            )
             cluster_information["dask_cluster_default_cores_limit"] = dict(
                 title="Default cores limit for dask clusters",
                 value=REANA_DASK_CLUSTER_DEFAULT_CORES_LIMIT,
@@ -388,6 +404,7 @@ class InfoSchema(Schema):
     kubernetes_max_memory_limit = fields.Nested(StringInfoValue)
     dask_enabled = fields.Nested(StringInfoValue)
     if DASK_ENABLED:
+        dask_autoscaler_enabled = fields.Nested(StringInfoValue)
         dask_cluster_default_cores_limit = fields.Nested(StringInfoValue)
         dask_cluster_max_cores_limit = fields.Nested(StringInfoValue)
         dask_cluster_default_memory_limit = fields.Nested(StringInfoValue)