lix-project/buildbot-nix
15
0
Fork 0
Code Issues 6 Pull requests 2 Packages Projects Releases Wiki Activity

pre-filter drv_info into all_deps

Browse source 
otherwise failure reporting is *enormous* with the entirety of a full
derivation info dump in there
This commit is contained in:
eldritch horrors 2024-03-11 13:07:35 +01:00
parent 13a67b483a
commit 51f7b52149
1 changed files with 20 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
buildbot_nix/__init__.py
View file

@ -61,11 +61,11 @@ class BuildTrigger(steps.BuildStep):
self,
builds_scheduler: str,
jobs: list[dict[str, Any]],
drv_info: dict[str, Any],
all_deps: dict[str, Any],
**kwargs: Any,
) -> None:
self.jobs = jobs
self.drv_info = drv_info
self.all_deps = all_deps
self.config = None
self.builds_scheduler = builds_scheduler
self._result_list = []
@ -159,20 +159,9 @@ class BuildTrigger(steps.BuildStep):
build_props = self.build.getProperties()
source = f"nix-eval-lix"
all_deps = dict()
for drv, info in self.drv_info.items():
all_deps[drv] = set(info.get("inputDrvs").keys())
def closure_of(key, deps):
r, size = set([key]), 0
while len(r) != size:
size = len(r)
r.update(*[ deps[k] for k in r ])
return r.difference([key])
job_set = set(( drv for drv in ( job.get("drvPath") for job in self.jobs ) if drv ))
all_deps = { k: list(closure_of(k, all_deps).intersection(job_set)) for k in job_set }
builds_to_schedule = list(self.jobs)
build_schedule_order = []
sorter = graphlib.TopologicalSorter(all_deps)
sorter = graphlib.TopologicalSorter(self.all_deps)
for item in sorter.static_order():
i = 0
while i < len(builds_to_schedule):
@ -191,7 +180,7 @@ class BuildTrigger(steps.BuildStep):
print('Scheduling..')
schedule_now = []
for build in list(build_schedule_order):
if all_deps.get(build.get("drvPath"), []) == []:
if self.all_deps.get(build.get("drvPath"), []) == []:
build_schedule_order.remove(build)
schedule_now.append(build)
if len(schedule_now) == 0:
@ -238,7 +227,7 @@ class BuildTrigger(steps.BuildStep):
while True:
old_paths = list(failed_paths)
for build in list(build_schedule_order):
deps = all_deps.get(build.get("drvPath"), [])
deps = self.all_deps.get(build.get("drvPath"), [])
for path in old_paths:
if path in deps:
failed_checks.append(build)
@ -252,9 +241,9 @@ class BuildTrigger(steps.BuildStep):
print(' Removed jobs: ' + ', '.join(removed))
all_results = worst_status(result, all_results)
print(f' New result: {util.Results[all_results].upper()}')
for dep in all_deps:
if job.get("drvPath") in all_deps[dep]:
all_deps[dep].remove(job.get("drvPath"))
for dep in self.all_deps:
if job.get("drvPath") in self.all_deps[dep]:
self.all_deps[dep].remove(job.get("drvPath"))
print('Done!')
if self.ended:
return util.CANCELLED
@ -328,6 +317,17 @@ class NixEvalCommand(buildstep.ShellMixin, steps.BuildStep):
except json.JSONDecodeError as e:
msg = f"Failed to parse `nix derivation show` output for {cmd.command}"
raise BuildbotNixError(msg) from e
all_deps = dict()
for drv, info in drv_info.items():
all_deps[drv] = set(info.get("inputDrvs").keys())
def closure_of(key, deps):
r, size = set([key]), 0
while len(r) != size:
size = len(r)
r.update(*[ deps[k] for k in r ])
return r.difference([key])
job_set = set(( drv for drv in ( job.get("drvPath") for job in filtered_jobs ) if drv ))
all_deps = { k: list(closure_of(k, all_deps).intersection(job_set)) for k in job_set }
self.build.addStepsAfterCurrentStep(
[
@ -335,7 +335,7 @@ class NixEvalCommand(buildstep.ShellMixin, steps.BuildStep):
builds_scheduler=f"lix-nix-build",
name="build flake",
jobs=filtered_jobs,
drv_info=drv_info,
all_deps=all_deps,
),
],
)