|
|
|
@ -1,4 +1,6 @@
|
|
|
|
|
const yaml = require('js-yaml');
|
|
|
|
|
const { PERM_JUDGE } = require('../permission');
|
|
|
|
|
const { parseTimeMS, parseMemoryMB } = require('../utils');
|
|
|
|
|
const record = require('../model/record');
|
|
|
|
|
const problem = require('../model/problem');
|
|
|
|
|
const builtin = require('../model/builtin');
|
|
|
|
@ -13,6 +15,7 @@ const {
|
|
|
|
|
async function _postJudge(rdoc) {
|
|
|
|
|
const accept = rdoc.status === builtin.STATUS_ACCEPTED;
|
|
|
|
|
bus.publish('record_change', rdoc);
|
|
|
|
|
if (rdoc.type === 'run') return;
|
|
|
|
|
const tasks = [];
|
|
|
|
|
if (rdoc.tid) {
|
|
|
|
|
tasks.push(
|
|
|
|
@ -26,7 +29,7 @@ async function _postJudge(rdoc) {
|
|
|
|
|
if (accept && !rdoc.rejudged) {
|
|
|
|
|
tasks.push(
|
|
|
|
|
problem.inc(rdoc.domainId, rdoc.pid, 'nAccept', 1),
|
|
|
|
|
user.inc(rdoc.domainId, rdoc.uid, 'nAccept', 1),
|
|
|
|
|
user.incDomain(rdoc.domainId, rdoc.uid, 'nAccept', 1),
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -78,6 +81,8 @@ async function end(body) {
|
|
|
|
|
}
|
|
|
|
|
if (body.status) $set.status = body.status;
|
|
|
|
|
if (body.score) $set.score = body.score;
|
|
|
|
|
if (body.stdout) $set.stdout = body.stdout;
|
|
|
|
|
if (body.stderr) $set.stderr = body.stderr;
|
|
|
|
|
if (body.time_ms) $set.time = body.time_ms;
|
|
|
|
|
if (body.memory_kb) $set.memory = body.memory_kb;
|
|
|
|
|
$set.judgeAt = new Date();
|
|
|
|
@ -87,6 +92,34 @@ async function end(body) {
|
|
|
|
|
rdoc = await record.update(body.domainId, body.rid, $set, $push);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
class PretestHandler extends Handler {
|
|
|
|
|
async post({
|
|
|
|
|
domainId, pid, code, lang, time = '1s', memory = '256m', input = '',
|
|
|
|
|
}) {
|
|
|
|
|
if (pid) {
|
|
|
|
|
const pdoc = await problem.get(domainId, pid);
|
|
|
|
|
if (pdoc.config) {
|
|
|
|
|
const config = yaml.safeLoad(pdoc.config);
|
|
|
|
|
if (config.time) time = config.time;
|
|
|
|
|
if (config.memory) memory = config.memory;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
const rid = await record.add(domainId, {
|
|
|
|
|
pid: pid || String.random(16),
|
|
|
|
|
uid: this.user._id,
|
|
|
|
|
type: 'run',
|
|
|
|
|
time: parseTimeMS(time),
|
|
|
|
|
memory: parseMemoryMB(memory),
|
|
|
|
|
input,
|
|
|
|
|
lang,
|
|
|
|
|
code,
|
|
|
|
|
hidden: true,
|
|
|
|
|
});
|
|
|
|
|
this.response.body = { rid };
|
|
|
|
|
this.response.redirect = this.url('record_detail', { rid });
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
class JudgeHandler extends Handler {
|
|
|
|
|
async get({ check = false }) {
|
|
|
|
|
if (check) return;
|
|
|
|
@ -130,6 +163,7 @@ class JudgeConnectionHandler extends ConnectionHandler {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async function apply() {
|
|
|
|
|
Route('pretest', '/pretest', PretestHandler);
|
|
|
|
|
Route('judge', '/judge', JudgeHandler, PERM_JUDGE);
|
|
|
|
|
Connection('judge_conn', '/judge/conn', JudgeConnectionHandler, PERM_JUDGE);
|
|
|
|
|
}
|
|
|
|
|