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

core: fix problemStat aggr executing

Browse Source
This commit is contained in:
undefined committed 2021-03-12 16:55:42 +08:00
1 parent be0b90dcd4
commit 139c1e2f59
1 file changed
+38 -33
Download Patch File Download Diff File Expand all files Collapse all files
packages/hydrooj/src/script/problemStat.ts
+38 -33
View File
@@ -1,3 +1,4 @@
/* eslint-disable no-await-in-loop */
import db from '../service/db';
import { STATUS } from '../model/builtin';
import * as document from '../model/document';
@@ -26,22 +27,24 @@ export async function udoc() {
},
},
];
const bulk = db.collection('domain.user').initializeUnorderedBulkOp();
db.collection('record').aggregate(pipeline).each(
(err, adoc: any) => {
if (adoc) {
bulk.find({
domainId: adoc._id.domainId,
uid: adoc._id.uid,
}).updateOne({
$set: {
nSubmit: adoc.nSubmit,
nAccept: adoc.nAccept,
},
});
}
},
);
let bulk = db.collection('domain.user').initializeUnorderedBulkOp();
const cursor = db.collection('record').aggregate(pipeline);
while (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();
}
@@ -67,23 +70,25 @@ export async function pdoc() {
},
},
];
const bulk = db.collection('document').initializeUnorderedBulkOp();
db.collection('record').aggregate(pipeline).each(
(err, adoc: any) => {
if (adoc) {
bulk.find({
domainId: adoc._id.domainId,
docType: document.TYPE_PROBLEM,
docId: adoc._id.pid,
}).updateOne({
$set: {
nSubmit: adoc.nSubmit,
nAccept: adoc.nAccept,
},
});
}
},
);
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();
}