Make workers in fast api service configurable

nebari-dev · Jan 9, 2024 · 31e7b5e · 31e7b5e
1 parent 43a29ce
commit 31e7b5e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/jhub_apps/config_utils.py b/jhub_apps/config_utils.py
@@ -1,4 +1,4 @@
-from traitlets import Unicode, Union, List, Callable
+from traitlets import Unicode, Union, List, Callable, Integer
 from traitlets.config import SingletonConfigurable, Enum
 
 
@@ -43,3 +43,8 @@ class JAppsConfig(SingletonConfigurable):
         "127.0.0.1",
         help="Hub Host name, in k8s environment it would be the container name, e.g. 'hub'",
     ).tag(config=True)
+
+    service_workers = Integer(
+        2,
+        help="The number of workers to create for the JHub Apps FastAPI service",
+    ).tag(config=True)
diff --git a/jhub_apps/configuration.py b/jhub_apps/configuration.py
@@ -33,6 +33,10 @@ def install_jhub_apps(c, spawner_to_subclass):
         c.JAppsConfig.jupyterhub_config_path = (
             JAppsConfig.jupyterhub_config_path.default_value
         )
+    if not isinstance(c.JAppsConfig.service_workers, int):
+        c.JAppsConfig.service_workers = (
+            JAppsConfig.service_workers.default_value
+        )
 
     if not isinstance(bind_url, str):
         raise ValueError(f"c.JupyterHub.bind_url is not set: {c.JupyterHub.bind_url}")