ops/nix-builder-autoscaler
4
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add remote autoscaler daemon endpoint support
All checks were successful
buildbot/nix-eval Build done.
Details
buildbot/nix-build Build done.
Details
buildbot/nix-effects Build done.
Details

Browse source
This commit is contained in:
Abel Luck 2026-03-05 15:47:57 +01:00
parent 95021a4253
commit 679b5c8d07
11 changed files with 291 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

16
buildbot-ext/buildbot_autoscale_ext/steps.py
View file

@ -20,7 +20,9 @@ class CapacityGateStep(buildstep.BuildStep):
def __init__(
self,
*,
daemon_socket: str,
daemon_socket: str | None = None,
daemon_url: str | None = None,
daemon_auth_token: str | None = None,
system_property: str,
default_system: str,
reserve_timeout_seconds: int,
@ -36,12 +38,14 @@ class CapacityGateStep(buildstep.BuildStep):
self._reserve_timeout_seconds = reserve_timeout_seconds
self._poll_interval_seconds = poll_interval_seconds
self._client = DaemonClient(
socket_path=daemon_socket,
retry_policy=RetryPolicy(
max_attempts=retry_max_attempts,
base_seconds=retry_base_seconds,
max_seconds=retry_max_seconds,
),
socket_path=daemon_socket,
base_url=daemon_url,
auth_token=daemon_auth_token,
)
def _determine_system(self) -> str:
@ -155,7 +159,9 @@ class CapacityReleaseStep(buildstep.BuildStep):
def __init__(
self,
*,
daemon_socket: str,
daemon_socket: str | None = None,
daemon_url: str | None = None,
daemon_auth_token: str | None = None,
retry_max_attempts: int,
retry_base_seconds: float,
retry_max_seconds: float = 5.0,
@ -163,12 +169,14 @@ class CapacityReleaseStep(buildstep.BuildStep):
) -> None:
super().__init__(**kwargs)
self._client = DaemonClient(
socket_path=daemon_socket,
retry_policy=RetryPolicy(
max_attempts=retry_max_attempts,
base_seconds=retry_base_seconds,
max_seconds=retry_max_seconds,
),
socket_path=daemon_socket,
base_url=daemon_url,
auth_token=daemon_auth_token,
)
def run(self) -> defer.Deferred[int]: