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.
119 lines
3.7 KiB
TypeScript
119 lines
3.7 KiB
TypeScript
/* eslint-disable no-await-in-loop */
|
|
import db from '../service/db';
|
|
import { STATUS } from '../model/builtin';
|
|
import * as document from '../model/document';
|
|
|
|
export const description = 'Recalculates nSubmit and nAccept in problem status.';
|
|
|
|
export async function udoc() {
|
|
const pipeline = [
|
|
{ $match: { hidden: false, type: { $ne: 'run' } } },
|
|
{
|
|
$group: {
|
|
_id: { domainId: '$domainId', pid: '$pid', uid: '$uid' },
|
|
nSubmit: { $sum: 1 },
|
|
nAccept: {
|
|
$sum: {
|
|
$cond: [{ $eq: ['$status', STATUS.STATUS_ACCEPTED] }, 1, 0],
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
$group: {
|
|
_id: { domainId: '$_id.domainId', uid: '$_id.uid' },
|
|
nSubmit: { $sum: '$nSubmit' },
|
|
nAccept: { $sum: { $min: ['$nAccept', 1] } },
|
|
},
|
|
},
|
|
];
|
|
let bulk = db.collection('domain.user').initializeUnorderedBulkOp();
|
|
const cursor = db.collection('record').aggregate(pipeline);
|
|
while (await cursor.hasNext()) {
|
|
const adoc = await cursor.next() as any;
|
|
bulk.find({
|
|
domainId: adoc._id.domainId,
|
|
uid: adoc._id.uid,
|
|
}).updateOne({
|
|
$set: {
|
|
nSubmit: adoc.nSubmit,
|
|
nAccept: adoc.nAccept,
|
|
},
|
|
});
|
|
if (bulk.length > 100) {
|
|
await bulk.execute();
|
|
bulk = db.collection('domain.user').initializeUnorderedBulkOp();
|
|
}
|
|
}
|
|
if (bulk.length) await bulk.execute();
|
|
}
|
|
|
|
export async function psdoc() {
|
|
const pipeline = [
|
|
{ $match: { hidden: false, type: { $ne: 'run' } } },
|
|
{
|
|
$group: {
|
|
_id: { domainId: '$domainId', pid: '$pid', uid: '$uid' },
|
|
nSubmit: { $sum: 1 },
|
|
},
|
|
},
|
|
];
|
|
const data = db.collection('record').aggregate(pipeline);
|
|
while (await data.hasNext()) {
|
|
const adoc = await data.next() as any;
|
|
await document.setStatus(adoc._id.domainId, document.TYPE_PROBLEM, adoc._id.pid, adoc._id.uid, { nSubmit: adoc.nSubmit });
|
|
}
|
|
}
|
|
|
|
export async function pdoc() {
|
|
const pipeline = [
|
|
{ $match: { hidden: false, type: { $ne: 'run' }, effective: true } },
|
|
{
|
|
$group: {
|
|
_id: { domainId: '$domainId', pid: '$pid', uid: '$uid' },
|
|
nSubmit: { $sum: 1 },
|
|
nAccept: {
|
|
$sum: {
|
|
$cond: [{ $eq: ['$status', STATUS.STATUS_ACCEPTED] }, 1, 0],
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
$group: {
|
|
_id: { domainId: '$_id.domainId', pid: '$_id.pid' },
|
|
nSubmit: { $sum: '$nSubmit' },
|
|
nAccept: { $sum: { $min: ['$nAccept', 1] } },
|
|
},
|
|
},
|
|
];
|
|
let bulk = db.collection('document').initializeUnorderedBulkOp();
|
|
const data = db.collection('record').aggregate(pipeline);
|
|
while (await data.hasNext()) {
|
|
const adoc = await data.next() as any;
|
|
bulk.find({
|
|
domainId: adoc._id.domainId,
|
|
docType: document.TYPE_PROBLEM,
|
|
docId: adoc._id.pid,
|
|
}).updateOne({
|
|
$set: {
|
|
nSubmit: adoc.nSubmit,
|
|
nAccept: adoc.nAccept,
|
|
},
|
|
});
|
|
if (bulk.length > 100) {
|
|
await bulk.execute();
|
|
bulk = db.collection('domain.user').initializeUnorderedBulkOp();
|
|
}
|
|
}
|
|
if (bulk.length) await bulk.execute();
|
|
}
|
|
|
|
export async function run() {
|
|
return await Promise.all([udoc(), pdoc(), psdoc()]);
|
|
}
|
|
|
|
export const validate = {};
|
|
|
|
global.Hydro.script.problemStat = { run, description, validate };
|