Adding idle timeout to dask gateway

nebari-dev · Mar 18, 2022 · c240f67 · c240f67
1 parent bc0f85e
commit c240f67
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/...s/07-kubernetes-services/modules/kubernetes/services/dask-gateway/files/gateway_config.py b/...s/07-kubernetes-services/modules/kubernetes/services/dask-gateway/files/gateway_config.py
@@ -30,6 +30,7 @@ def dask_gateway_config(path="/var/lib/dask-gateway/config.json"):
 )
 c.KubeClusterConfig.image_pull_policy = config["cluster"]["image_pull_policy"]
 c.KubeClusterConfig.environment = config["cluster"]["environment"]
+c.KubeClusterConfig.idle_timeout = config["cluster"]["idle_timeout"]
 
 c.KubeClusterConfig.scheduler_cores = config["cluster"]["scheduler_cores"]
 c.KubeClusterConfig.scheduler_cores_limit = config["cluster"]["scheduler_cores_limit"]

diff --git a/...plate/stages/07-kubernetes-services/modules/kubernetes/services/dask-gateway/variables.tf b/...plate/stages/07-kubernetes-services/modules/kubernetes/services/dask-gateway/variables.tf
@@ -144,6 +144,7 @@ variable "cluster" {
     worker_extra_container_config = any
     worker_extra_pod_config       = any
     # additional fields
+    idle_timeout      = number
     image_pull_policy = string
     environment       = map(string)
   })
@@ -163,6 +164,7 @@ variable "cluster" {
     worker_extra_container_config = {}
     worker_extra_pod_config       = {}
     # additional fields
+    idle_timeout      = 1800 # 30 minutes
     image_pull_policy = "IfNotPresent"
     environment       = {}
   }