* Basic job info in the database.
This commit is contained in:
161
src/scheduler.pl
161
src/scheduler.pl
@ -2,37 +2,19 @@
|
||||
|
||||
use strict;
|
||||
use XML::Simple;
|
||||
use DBI;
|
||||
use File::Basename;
|
||||
use HydraFrontend::Schema;
|
||||
|
||||
|
||||
my $jobsFile = "../test.nix";
|
||||
my $db = HydraFrontend::Schema->connect("dbi:SQLite:dbname=hydra.sqlite", "", "", {});
|
||||
|
||||
|
||||
my $dbh = DBI->connect("dbi:SQLite:dbname=hydra.sqlite", "", "");
|
||||
sub buildJob {
|
||||
my ($project, $jobset, $jobName, $drvPath, $outPath) = @_;
|
||||
|
||||
|
||||
my $jobsXml = `nix-env -f $jobsFile --query --available "*" --attr-path --out-path --drv-path --meta --xml --system-filter "*"`
|
||||
or die "cannot evaluate the Nix expression containing the job definitions: $?";
|
||||
|
||||
print "$jobsXml";
|
||||
|
||||
|
||||
my $jobs = XMLin($jobsXml, KeyAttr => ['attrPath', 'name'])
|
||||
or die "cannot parse XML output";
|
||||
|
||||
|
||||
foreach my $jobName (keys %{$jobs->{item}}) {
|
||||
my $job = $jobs->{item}->{$jobName};
|
||||
my $description = defined $job->{meta}->{description} ? $job->{meta}->{description}->{value} : "";
|
||||
print "JOB: $jobName ($description)\n";
|
||||
|
||||
my $outPath = $job->{outPath};
|
||||
my $drvPath = $job->{drvPath};
|
||||
|
||||
if (scalar(@{$dbh->selectall_arrayref("select * from builds where jobName = ? and outPath = ?", {}, $jobName, $outPath)}) > 0) {
|
||||
print " already done\n";
|
||||
next;
|
||||
if (scalar($db->resultset('Builds')->search({project => $project->name, jobset => $jobset->name, attrname => $jobName, outPath => $outPath})) > 0) {
|
||||
print " already done\n";
|
||||
return;
|
||||
}
|
||||
|
||||
my $isCachedBuild = 1;
|
||||
@ -45,13 +27,34 @@ foreach my $jobName (keys %{$jobs->{item}}) {
|
||||
|
||||
$startTime = time();
|
||||
|
||||
my $res = system("nix-build $jobsFile --attr $jobName");
|
||||
print " BUILDING\n";
|
||||
|
||||
my $res = system("nix-store --realise $drvPath");
|
||||
|
||||
$stopTime = time();
|
||||
|
||||
$buildStatus = $res == 0 ? 0 : 1;
|
||||
}
|
||||
|
||||
$db->txn_do(sub {
|
||||
my $build = $db->resultset('Builds')->create(
|
||||
{ timestamp => time()
|
||||
, project => $project->name
|
||||
, jobset => $jobset->name
|
||||
, attrname => $jobName
|
||||
, drvpath => $drvPath
|
||||
, outpath => $outPath
|
||||
, iscachedbuild => $isCachedBuild
|
||||
, buildstatus => $buildStatus
|
||||
, starttime => $startTime
|
||||
, stoptime => $stopTime
|
||||
});
|
||||
});
|
||||
|
||||
return 0;
|
||||
|
||||
my $dbh, my $description, my $jobName;
|
||||
|
||||
$dbh->begin_work;
|
||||
|
||||
$dbh->prepare("insert into builds(timestamp, jobName, description, drvPath, outPath, isCachedBuild, buildStatus, errorMsg, startTime, stopTime) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
|
||||
@ -83,3 +86,109 @@ foreach my $jobName (keys %{$jobs->{item}}) {
|
||||
|
||||
$dbh->commit;
|
||||
}
|
||||
|
||||
|
||||
sub fetchInput {
|
||||
my ($input, $inputInfo) = @_;
|
||||
my $type = $input->type;
|
||||
my $uri = $input->uri;
|
||||
|
||||
if ($type eq "path") {
|
||||
my $storePath = `nix-store --add "$uri"`
|
||||
or die "cannot copy path $uri to the Nix store";
|
||||
chomp $storePath;
|
||||
print " copied to $storePath\n";
|
||||
$$inputInfo{$input->name} = {storePath => $storePath};
|
||||
}
|
||||
|
||||
else {
|
||||
die "input `" . $input->type . "' has unknown type `$type'";
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
sub checkJobSet {
|
||||
my ($project, $jobset) = @_;
|
||||
|
||||
my $inputInfo = {};
|
||||
|
||||
foreach my $input ($jobset->jobsetinputs) {
|
||||
print " INPUT ", $input->name, " (", $input->type, " ", $input->uri, ")\n";
|
||||
fetchInput($input, $inputInfo);
|
||||
}
|
||||
|
||||
die unless defined $inputInfo->{$jobset->nixexprinput};
|
||||
|
||||
my $nixExprPath = $inputInfo->{$jobset->nixexprinput}->{storePath} . "/" . $jobset->nixexprpath;
|
||||
|
||||
print " EVALUATING $nixExprPath\n";
|
||||
|
||||
my $jobsXml = `nix-instantiate $nixExprPath --eval-only --strict --xml`
|
||||
or die "cannot evaluate the Nix expression containing the jobs: $?";
|
||||
|
||||
#print "$jobsXml";
|
||||
|
||||
my $jobs = XMLin($jobsXml,
|
||||
ForceArray => [qw(value)],
|
||||
KeyAttr => ['name'],
|
||||
SuppressEmpty => '',
|
||||
ValueAttr => [value => 'value'])
|
||||
or die "cannot parse XML output";
|
||||
|
||||
die unless defined $jobs->{attrs};
|
||||
|
||||
foreach my $jobName (keys(%{$jobs->{attrs}->{attr}})) {
|
||||
print " JOB $jobName\n";
|
||||
|
||||
my $jobExpr = $jobs->{attrs}->{attr}->{$jobName};
|
||||
|
||||
my $extraArgs = "";
|
||||
|
||||
# If the expression is a function, then look at its formal
|
||||
# arguments and see if we can supply them.
|
||||
if (defined $jobExpr->{function}->{attrspat}) {
|
||||
foreach my $argName (keys(%{$jobExpr->{function}->{attrspat}->{attr}})) {
|
||||
print " needs input $argName\n";
|
||||
die "missing input `$argName'" if !defined $inputInfo->{$argName};
|
||||
$extraArgs .= " --arg $argName '{path = " . $inputInfo->{$argName}->{storePath} . ";}'";
|
||||
}
|
||||
}
|
||||
|
||||
# Instantiate the store derivation.
|
||||
my $drvPath = `nix-instantiate $nixExprPath --attr $jobName $extraArgs`
|
||||
or die "cannot evaluate the Nix expression containing the job definitions: $?";
|
||||
chomp $drvPath;
|
||||
|
||||
# Call nix-env --xml to get info about this job (drvPath, outPath, meta attributes, ...).
|
||||
my $infoXml = `nix-env -f $nixExprPath --query --available "*" --attr-path --out-path --drv-path --meta --xml --system-filter "*" --attr $jobName $extraArgs`
|
||||
or die "cannot get information about the job: $?";
|
||||
|
||||
my $info = XMLin($infoXml, KeyAttr => ['attrPath', 'name'])
|
||||
or die "cannot parse XML output";
|
||||
|
||||
my $job = $info->{item};
|
||||
die unless !defined $job || $job->{system} ne $jobName;
|
||||
|
||||
my $description = defined $job->{meta}->{description} ? $job->{meta}->{description}->{value} : "";
|
||||
die unless $job->{drvPath} eq $drvPath;
|
||||
my $outPath = $job->{outPath};
|
||||
|
||||
buildJob($project, $jobset, $jobName, $drvPath, $outPath);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
sub checkJobs {
|
||||
|
||||
foreach my $project ($db->resultset('Projects')->all) {
|
||||
print "PROJECT ", $project->name, "\n";
|
||||
foreach my $jobset ($project->jobsets->all) {
|
||||
print " JOBSET ", $jobset->name, "\n";
|
||||
checkJobSet($project, $jobset);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
checkJobs;
|
||||
|
Reference in New Issue
Block a user