You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.4 KiB
JavaScript
79 lines
2.4 KiB
JavaScript
const
|
|
{ GET, POST } = require('../service/server'),
|
|
queue = require('../service/queue'),
|
|
record = require('../model/record'),
|
|
{ requirePerm } = require('./tools'),
|
|
{ PERM_JUDGE } = require('../permission');
|
|
|
|
queue.assert('judge');
|
|
|
|
GET('/judge/noop', requirePerm(PERM_JUDGE), async ctx => {
|
|
ctx.body = {};
|
|
});
|
|
GET('/judge/fetch', requirePerm(PERM_JUDGE), async ctx => {
|
|
let rid = await queue.get('judge', false);
|
|
if (rid) {
|
|
let rdoc = await record.get(rid);
|
|
let task = {
|
|
event: 'judge',
|
|
rid, type: 0,
|
|
pid: rdoc.pid,
|
|
data: rdoc.data,
|
|
lang: rdoc.lang,
|
|
code: rdoc.code
|
|
};
|
|
ctx.body = { task };
|
|
}
|
|
else ctx.body = {};
|
|
});
|
|
POST('/judge/next', requirePerm(PERM_JUDGE), async ctx => {
|
|
console.log(ctx.request.body);
|
|
let body = ctx.request.body;
|
|
let rdoc = await record.get(body.rid);
|
|
let $set = {};
|
|
if (body.case) {
|
|
rdoc.testCases.push(body.case);
|
|
$set.testCases = rdoc.testCases;
|
|
}
|
|
if (body.judge_text) {
|
|
rdoc.judgeTexts.push(body.judge_text);
|
|
$set.judgeTexts = rdoc.judgeTexts;
|
|
}
|
|
if (body.compiler_text) {
|
|
rdoc.compilerTexts.push(body.compilerTexts);
|
|
$set.compilerTexts = rdoc.compilerTexts;
|
|
}
|
|
if (body.status) $set.status = body.status;
|
|
if (body.score) $set.score = body.score;
|
|
if (body.time_ms) $set.time = body.time_ms;
|
|
if (body.memory_kb) $set.memory = body.memory_kb;
|
|
await record.update(body.rid, $set);
|
|
ctx.body = {};
|
|
});
|
|
POST('/judge/end', requirePerm(PERM_JUDGE), async ctx => {
|
|
console.log(ctx.request.body);
|
|
let body = ctx.request.body;
|
|
let rdoc = await record.get(body.rid);
|
|
let $set = {};
|
|
if (body.case) {
|
|
rdoc.testCases.push(body.case);
|
|
$set.testCases = rdoc.testCases;
|
|
}
|
|
if (body.judge_text) {
|
|
rdoc.judgeTexts.push(body.judge_text);
|
|
$set.judgeTexts = rdoc.judgeTexts;
|
|
}
|
|
if (body.compiler_text) {
|
|
rdoc.compilerTexts.push(body.compilerTexts);
|
|
$set.compilerTexts = rdoc.compilerTexts;
|
|
}
|
|
if (body.status) $set.status = body.status;
|
|
if (body.score) $set.score = body.score;
|
|
if (body.time_ms) $set.time = body.time_ms;
|
|
if (body.memory_kb) $set.memory = body.memory_kb;
|
|
$set.judgeAt = new Date();
|
|
$set.judger = ctx.state.user._id;
|
|
await record.update(body.rid, $set);
|
|
ctx.body = {};
|
|
});
|