Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

core&ui: add problem stat (#798)

Browse Source 
Co-authored-by: undefined <i@undefined.moe>
This commit is contained in:
pandaandundefined authored and GitHub committed 2024-05-11 00:11:27 +08:00
1 parent c51ff5ce3a
commit d582617aaf
11 files changed
+295 -74
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/src/handler/problem.ts
+29 -1
View File
@@ -667,6 +667,7 @@ export class ProblemFilesHandler extends ProblemDetailHandler {
@param('pjax', Types.Boolean)
@param('sidebar', Types.Boolean)
async get(domainId: string, d = ['testdata', 'additional_file'], pjax = false, sidebar = false) {
if (this.tdoc) throw new ContestNotEndedError();
this.response.body.testdata = d.includes('testdata') ? sortFiles(this.pdoc.data || []) : [];
this.response.body.reference = this.pdoc.reference;
this.response.body.additional_file = d.includes('additional_file') ? sortFiles(this.pdoc.additional_file || []) : [];
@@ -972,6 +973,32 @@ export class ProblemSolutionRawHandler extends ProblemDetailHandler {
}
}
export class ProblemStatisticsHandler extends ProblemDetailHandler {
@param('sort', Types.Range(Object.keys(record.STAT_QUERY)), true)
@param('direction', Types.Range([-1, 1]), true)
@param('lang', Types.String, true)
@param('page', Types.PositiveInt, true)
async get(domainId: string, sort = 'time', direction: 1 | -1 = 1, lang?: string, page = 1) {
if (this.tdoc) throw new ContestNotEndedError();
const [rsdocs, pcount, rscount] = await this.paginate(
record.getMultiStat(domainId, {
pid: this.pdoc.docId,
...lang ? { lang } : {},
}, record.STAT_QUERY[sort][Math.max(direction, 0)]),
page,
'record',
);
const [udict, udoc] = await Promise.all([
user.getListForRender(domainId, rsdocs.map((i) => i.uid), this.user.hasPerm(PERM.PERM_VIEW_DISPLAYNAME)),
user.getById(domainId, this.pdoc.owner),
]);
this.response.template = 'problem_statistics.html';
this.response.body = {
rsdocs, page, pcount, rscount, sort, direction, pdoc: this.pdoc, udict, types: Object.keys(record.STAT_QUERY), udoc,
};
}
}
export class ProblemCreateHandler extends Handler {
async get() {
this.response.template = 'problem_edit.html';
@@ -1014,7 +1041,7 @@ export class ProblemCreateHandler extends Handler {
export class ProblemPrefixListHandler extends Handler {
@param('prefix', Types.Name)
async get(domainId: string, prefix: string) {
const projection = ['domainId', 'docId', 'pid', 'title'] as const;
const projection = ['domainId', 'docId', 'pid', 'title', 'hidden'] as const;
const [pdocs, pdoc, apdoc] = await Promise.all([
problem.getPrefixList(domainId, prefix),
problem.get(domainId, Number.isSafeInteger(+prefix) ? +prefix : prefix, projection),
@@ -1047,6 +1074,7 @@ export async function apply(ctx) {
ctx.Route('problem_solution_detail', '/p/:pid/solution/:sid', ProblemSolutionHandler, PERM.PERM_VIEW_PROBLEM);
ctx.Route('problem_solution_raw', '/p/:pid/solution/:psid/raw', ProblemSolutionRawHandler, PERM.PERM_VIEW_PROBLEM);
ctx.Route('problem_solution_reply_raw', '/p/:pid/solution/:psid/:psrid/raw', ProblemSolutionRawHandler, PERM.PERM_VIEW_PROBLEM);
ctx.Route('problem_statistics', '/p/:pid/stat', ProblemStatisticsHandler, PERM.PERM_VIEW_PROBLEM);
ctx.Route('problem_create', '/problem/create', ProblemCreateHandler, PERM.PERM_CREATE_PROBLEM);
ctx.Route('problem_prefix_list', '/problem/list', ProblemPrefixListHandler, PERM.PERM_VIEW_PROBLEM);
}
packages/hydrooj/src/interface.ts
+11
View File
@@ -338,6 +338,16 @@ export interface RecordDoc {
subtasks?: Record<number, SubtaskResult>;
}
export interface RecordStatDoc {
_id: ObjectId;
domainId: string;
pid: number;
uid: number;
time: number;
memory: number;
code: number;
lang: string;
}
export interface JudgeMeta {
problemOwner: number;
hackRejudge?: string;
@@ -688,6 +698,7 @@ declare module './service/db' {
'domain': DomainDoc;
'domain.user': any;
'record': RecordDoc;
'record.stat': RecordStatDoc;
'document': any;
'document.status': StatusDocBase & {
[K in keyof DocStatusType]: { docType: K } & DocStatusType[K];
packages/hydrooj/src/model/record.ts
+44 -9
View File
@@ -19,6 +19,7 @@ import task from './task';
export default class RecordModel {
static coll = db.collection('record');
static collStat = db.collection('record.stat');
static PROJECTION_LIST: (keyof RecordDoc)[] = [
'_id', 'score', 'time', 'memory', 'lang',
'uid', 'pid', 'rejudged', 'progress', 'domainId',
@@ -26,6 +27,13 @@ export default class RecordModel {
'files',
];
static STAT_QUERY = {
time: [{ time: -1 }, { time: 1 }],
memory: [{ memory: -1 }, { memory: 1 }],
length: [{ length: -1 }, { length: 1 }],
date: [{ _id: -1 }, { _id: 1 }],
};
static RECORD_PRETEST = new ObjectId('000000000000000000000000');
static RECORD_GENERATE = new ObjectId('000000000000000000000001');
@@ -163,6 +171,10 @@ export default class RecordModel {
return RecordModel.coll.find(query);
}
static getMultiStat(domainId: string, query: any, sortBy: any = { _id: -1 }) {
return RecordModel.collStat.find({ domainId, ...query }).sort(sortBy);
}
static async update(
domainId: string, _id: MaybeArray<ObjectId>,
$set?: MatchKeysAndValues<RecordDoc>,
@@ -218,6 +230,7 @@ export default class RecordModel {
judger: null,
};
if (isRejudge) upd.rejudged = true;
await RecordModel.collStat.deleteOne(rid instanceof Array ? { _id: { $in: rid } } : { _id: rid });
await task.deleteMany(rid instanceof Array ? { rid: { $in: rid } } : { rid });
return RecordModel.update(domainId, rid, upd);
}
@@ -243,14 +256,36 @@ export function apply(ctx: Context) {
// Mark problem as deleted
ctx.on('problem/delete', (domainId, docId) => RecordModel.coll.deleteMany({ domainId, pid: docId }));
ctx.on('domain/delete', (domainId) => RecordModel.coll.deleteMany({ domainId }));
ctx.on('ready', () => db.ensureIndexes(
RecordModel.coll,
{ key: { domainId: 1, contest: 1, _id: -1 }, name: 'basic' },
{ key: { domainId: 1, contest: 1, uid: 1, _id: -1 }, name: 'withUser' },
{ key: { domainId: 1, contest: 1, pid: 1, _id: -1 }, name: 'withProblem' },
{ key: { domainId: 1, contest: 1, pid: 1, uid: 1, _id: -1 }, name: 'withUserAndProblem' },
{ key: { domainId: 1, contest: 1, status: 1, _id: -1 }, name: 'withStatus' },
));
ctx.on('record/judge', (rdoc, updated) => {
if (rdoc.status === STATUS.STATUS_ACCEPTED && updated) {
RecordModel.collStat.insertOne({
_id: rdoc._id,
domainId: rdoc.domainId,
pid: rdoc.pid,
uid: rdoc.uid,
time: rdoc.time,
memory: rdoc.memory,
code: rdoc.code?.length || 0,
lang: rdoc.lang,
});
}
});
ctx.on('ready', async () => {
await db.ensureIndexes(
RecordModel.coll,
{ key: { domainId: 1, contest: 1, _id: -1 }, name: 'basic' },
{ key: { domainId: 1, contest: 1, uid: 1, _id: -1 }, name: 'withUser' },
{ key: { domainId: 1, contest: 1, pid: 1, _id: -1 }, name: 'withProblem' },
{ key: { domainId: 1, contest: 1, pid: 1, uid: 1, _id: -1 }, name: 'withUserAndProblem' },
{ key: { domainId: 1, contest: 1, status: 1, _id: -1 }, name: 'withStatus' },
);
await db.ensureIndexes(
RecordModel.collStat,
{ key: { domainId: 1, pid: 1, uid: 1, _id: -1 }, name: 'basic' },
{ key: { domainId: 1, pid: 1, uid: 1, time: 1 }, name: 'time' },
{ key: { domainId: 1, pid: 1, uid: 1, memory: 1 }, name: 'memory' },
{ key: { domainId: 1, pid: 1, uid: 1, code: 1 }, name: 'code' },
);
});
}
global.Hydro.model.record = RecordModel;
packages/hydrooj/src/upgrade.ts
+22
View File
@@ -649,6 +649,28 @@ const scripts: UpgradeScript[] = [
await domain.setRoles(ddoc._id, ddoc.roles);
});
},
async function _88_89() {
const cursor = RecordModel.getMulti(undefined, { status: STATUS.STATUS_ACCEPTED });
let bulk = [];
for await (const doc of cursor) {
bulk.push({
_id: doc._id,
domainId: doc.domainId,
pid: doc.pid,
uid: doc.uid,
time: doc.time,
memory: doc.memory,
code: doc.code?.length || 0,
lang: doc.lang,
});
if (bulk.length > 500) {
await RecordModel.collStat.insertMany(bulk);
bulk = [];
}
}
if (bulk.length) await RecordModel.collStat.insertMany(bulk);
return true;
},
];
export default scripts;