1 package Gruntmaster
::Daemon
;
7 our $VERSION = '5999.000_004';
9 use Gruntmaster
::Daemon
::Constants qw
/ERR/;
10 use Gruntmaster
::Daemon
::Format qw
/prepare_files stopvms/;
12 use File
::Slurp qw
/read_file/;
13 use File
::Temp qw
/tempdir/;
14 use JSON qw
/decode_json encode_json/;
15 use Sys
::Hostname qw
/hostname/;
16 use Time
::HiRes qw
/time/;
18 use Log
::Log4perl qw
/get_logger/;
20 use constant FORMAT_EXTENSION
=> {
36 ##################################################
41 my ($type, $sub, $name) = @_;
43 get_logger
->logdie("Invalid \l$type: '$name'") unless $name =~ /^\w+$/s;
44 my $pkg = __PACKAGE__
. "::${type}::${name}";
45 eval "require $pkg" or get_logger
->warn("Error while requiring $pkg: $@");
46 $pkg->can($sub) or get_logger
->logdie("No such \l$type: '$name'");
53 my @full_results = ();
59 my ($files, $generator, $runner, $judge, $testcnt) = map { $meta->{$_} or die "Required parameter missing: $_\n"} qw
/files generator runner judge testcnt/;
61 $generator = safe_can Generator
=> generate
=> $generator;
62 $runner = safe_can Runner
=> run
=> $runner;
63 $judge = safe_can Judge
=> judge
=> $judge;
65 for my $test (1 .. $testcnt) {
66 my $start_time = time;
70 $generator->($test, $meta);
71 $result = $runner->($test, $meta);
74 unless (ref $result) {
76 $result = [ERR
, $result];
81 my $result_text = $result->[1];
82 $result_text .= ': ' . read_file
'result' if -s
'result';
83 get_logger
->trace("Test $test result is " . $result_text);
84 push @full_results, {id
=> $test, result
=> $result->[0], result_text
=> $result_text, time => time - $start_time}
86 get_logger
->trace("Test $test result is $result");
87 push @full_results, {id
=> $test, result
=> 0, result_text
=> $result, time => time - $start_time}
89 push @results, $result;
90 last if $meta->{judge
} eq 'Absolute' && ref $result
93 my %results = $judge->(@results);
94 $meta->{$_} = $results{$_} for keys %results;
95 $meta->{results
} = \
@full_results
100 $meta->{result_text
} = $_;
103 $meta->{errors
} = $errors;
105 get_logger
->info('Job result: ' . $meta->{result_text
});
111 problem
=> $job->problem->id,
114 name
=> 'prog.' . $job->extension,
115 format
=> $job->format,
116 content
=> $job->source,
119 map { $_ => $job->problem->get_column($_) } qw
/generator runner judge precnt testcnt timeout olimit/
121 my $limit_override = $job->result_source->schema->limits->find($meta->{problem
}, $job->format);
122 $meta->{timeout
} = $limit_override->timeout if $limit_override;
123 $meta->{tests
} = decode_json
$job->problem->tests if $meta->{runner
} eq 'File';
124 delete $meta->{precnt
} unless $job->contest && $job->contest->is_running;
125 $meta->{testcnt
} = $meta->{precnt
} if $meta->{precnt
};
127 $meta->{files
}{ver
} = {
128 name
=> 'ver.' . FORMAT_EXTENSION
->{$job->problem->verformat},
129 format
=> $job->problem->verformat,
130 content
=> $job->problem->versource,
131 } if $job->problem->verformat;
135 $meta->{result_text
} .= ' (pretests)' if $meta->{precnt
};
137 result
=> $meta->{result
},
138 result_text
=> $meta->{result_text
},
139 ($meta->{results
} ?
(results
=> encode_json
$meta->{results
}) : ()),
140 $meta->{errors
} ?
(errors
=> $meta->{errors
}) : ()
143 my $solved = $meta->{result
} ?
0 : 1;
144 my $status = $db->problem_statuses->find($job->owner, $job->problem);
145 $db->problem_statuses->update_or_create({
146 problem
=> $job->problem,
147 owner
=> $job->owner,
149 solved
=> $solved}) if !$job->problem->private;
155 get_logger
->debug("Taking job $id...");
156 my $daemon = hostname
. ":$$";
157 $job->update({daemon
=> $daemon}, {where
=> {daemon
=> undef}})->discard_changes;
158 if ($job->daemon eq $daemon) {
159 get_logger
->debug("Succesfully taken job $id");
161 get_logger
->debug("Job $id done");
163 get_logger
->debug("Job $id already taken by " . $job->daemon);
168 require Gruntmaster
::Data
;
169 $db = Gruntmaster
::Data
->connect($ENV{GRUNTMASTER_DSN
} // 'dbi:Pg:');
170 Log
::Log4perl
->init('/etc/gruntmasterd/gruntmasterd-log.conf');
171 get_logger
->info("gruntmasterd $VERSION started");
172 my $dir = tempdir
'gruntmasterd.XXXX', CLEANUP
=> 1, TMPDIR
=> 1;
176 my $job = $db->jobs->search({daemon
=> undef}, {rows
=> 1})->first;
177 got_job
$job if defined $job;
178 sleep 2 unless defined $job;
187 Gruntmaster::Daemon - Gruntmaster 6000 Online Judge -- daemon
191 use Gruntmaster::Daemon;
192 Gruntmaster::Daemon->run;
196 Gruntmaster::Daemon is the daemon component of the Gruntmaster 6000 online judge.
200 Marius Gavrilescu E<lt>marius@ieval.roE<gt>
202 =head1 COPYRIGHT AND LICENSE
204 Copyright (C) 2014 by Marius Gavrilescu
206 This library is free software: you can redistribute it and/or modify
207 it under the terms of the GNU Affero General Public License as published by
208 the Free Software Foundation, either version 3 of the License, or
209 (at your option) any later version.