59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import multiprocessing
|
|
import os
|
|
import socket
|
|
from pathlib import Path
|
|
|
|
from buildbot_worker.bot import Worker
|
|
from twisted.application import service
|
|
|
|
|
|
def require_env(key: str) -> str:
|
|
val = os.environ.get(key)
|
|
assert val is not None, "val is not set"
|
|
return val
|
|
|
|
|
|
PASSWD = Path(require_env("WORKER_PASSWORD_FILE")).read_text().strip("\r\n")
|
|
BUILDBOT_DIR = require_env("BUILDBOT_DIR")
|
|
MASTER_URL = require_env("MASTER_URL")
|
|
|
|
|
|
def setup_worker(application: service.Application, id: int) -> None:
|
|
basedir = f"{BUILDBOT_DIR}-{id}"
|
|
os.makedirs(basedir, mode=0o700, exist_ok=True)
|
|
|
|
hostname = socket.gethostname()
|
|
workername = f"{hostname}-{id}"
|
|
keepalive = 600
|
|
umask = None
|
|
maxdelay = 300
|
|
numcpus = None
|
|
allow_shutdown = None
|
|
|
|
s = Worker(
|
|
None,
|
|
None,
|
|
workername,
|
|
PASSWD,
|
|
basedir,
|
|
keepalive,
|
|
connection_string=MASTER_URL,
|
|
umask=umask,
|
|
maxdelay=maxdelay,
|
|
numcpus=numcpus,
|
|
allow_shutdown=allow_shutdown,
|
|
)
|
|
# defaults to 4096, bump to 10MB for nix-eval-jobs
|
|
s.bot.max_line_length = 10485760
|
|
s.setServiceParent(application)
|
|
|
|
|
|
# note: this line is matched against to check that this is a worker
|
|
# directory; do not edit it.
|
|
application = service.Application("buildbot-worker")
|
|
|
|
for i in range(multiprocessing.cpu_count()):
|
|
setup_worker(application, i)
|