Make scheduler configurable

superduper-io · Jul 5, 2024 · f9581a0 · f9581a0
1 parent 064ed9b
commit f9581a0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/superduperdb/base/config.py b/superduperdb/base/config.py
@@ -155,6 +155,15 @@ class CronTab(BaseConfig):
 
     uri: t.Optional[str] = None
 
+@dc.dataclass
+class SchedulerConfig(BaseConfig):
+    """Describes the configuration for scheduler service
+
+    :param uri: The URI for the scheduler service.
+    """
+
+    uri: t.Optional[str] = None # None implies local mode
+
 
 @dc.dataclass
 class Compute(BaseConfig):
@@ -196,7 +205,7 @@ class Cluster(BaseConfig):
     vector_search: VectorSearch = dc.field(default_factory=VectorSearch)
     rest: Rest = dc.field(default_factory=Rest)
     cdc: CDCConfig = dc.field(default_factory=CDCConfig)
-
+    scheduler: SchedulerConfig = dc.field(default_factory=SchedulerConfig)
 
 class LogLevel(str, Enum):
     """Enumerate log severity level # noqa."""

diff --git a/superduperdb/misc/server.py b/superduperdb/misc/server.py
@@ -42,7 +42,7 @@ def _request_server(
     elif service == 'vector_search':
         service_uri = CFG.cluster.vector_search.uri
     elif service == 'scheduler':
-        service_uri = 'http://localhost:8181'
+        service_uri = CFG.cluster.scheduler.uri
     else:
         raise NotImplementedError(f'Unknown service {service}')