Add alternative_endpoints to exporter configuration

Author: NickCao

packages/jumpstarter/jumpstarter/config/exporter.py

@@ -2,7 +2,7 @@
 
 from contextlib import asynccontextmanager, contextmanager, suppress
 from pathlib import Path
-from typing import Any, ClassVar, Literal, Optional, Self
+from typing import Any, ClassVar, List, Literal, Optional, Self
 
 import grpc
 import yaml
@@ -82,6 +82,8 @@ class ExporterConfigV1Alpha1(BaseModel):
     tls: TLSConfigV1Alpha1 = Field(default_factory=TLSConfigV1Alpha1)
     token: str
 
+    alternative_endpoints: List[str] = Field(default_factory=list)
+
     export: dict[str, ExporterConfigV1Alpha1DriverInstance] = Field(default_factory=dict)
 
     path: Path | None = Field(default=None)
@@ -169,6 +171,7 @@ def channel_factory():
             channel_factory=channel_factory,
             device_factory=ExporterConfigV1Alpha1DriverInstance(children=self.export).instantiate,
             tls=self.tls,
+            alternative_endpoints=self.alternative_endpoints,
         ) as exporter:
             await exporter.serve()
 

packages/jumpstarter/jumpstarter/exporter/exporter.py

@@ -25,6 +25,7 @@ class Exporter(AbstractAsyncContextManager, Metadata):
     channel_factory: Callable[[], grpc.aio.Channel]
     device_factory: Callable[[], Driver]
     lease_name: str = field(init=False, default="")
+    alternative_endpoints: list[str] = field(default_factory=list)
     tls: TLSConfigV1Alpha1 = field(default_factory=TLSConfigV1Alpha1)
 
     async def __aexit__(self, exc_type, exc_value, traceback):
@@ -49,7 +50,7 @@ async def session(self):
             labels=self.labels,
             root_device=self.device_factory(),
         ) as session:
-            async with session.serve_unix_async() as path:
+            async with session.serve_unix_async(alternative_endpoints=self.alternative_endpoints) as path:
                 async with grpc.aio.secure_channel(
                     f"unix://{path}", grpc.local_channel_credentials(grpc.LocalConnectionType.UDS)
                 ) as channel: