core: problemStat: also calc psdoc

packages/hydrooj/package.json

@@ -1,6 +1,6 @@
 {
     "name": "hydrooj",
-    "version": "2.21.18",
+    "version": "2.21.19",
     "bin": "bin/hydrooj.js",
     "main": "dist/loader.js",
     "typings": "dist/loader.d.ts",

packages/hydrooj/src/handler/problem.ts

@@ -240,6 +240,7 @@ export class ProblemSubmitHandler extends ProblemDetailHandler {
         const [rdoc] = await Promise.all([
             record.get(domainId, rid),
             problem.inc(domainId, this.pdoc.docId, 'nSubmit', 1),
+            problem.incStatus(domainId, this.pdoc.docId, this.user._id, 'nSubmit', 1),
             domain.incUserInDomain(domainId, this.user._id, 'nSubmit'),
         ]);
         bus.boardcast('record/change', rdoc);

packages/hydrooj/src/model/problem.ts

@@ -223,6 +223,13 @@ export class ProblemModel {
         return !!res;
     }
 
+    static async incStatus(
+        domainId: string, pid: number, uid: number,
+        key: NumberKeys<ProblemStatusDoc>, count: number,
+    ) {
+        return await document.incStatus(domainId, document.TYPE_PROBLEM, pid, uid, key, count);
+    }
+
     static setStar(domainId: string, pid: number, uid: number, star: boolean) {
         return document.setStatus(domainId, document.TYPE_PROBLEM, pid, uid, { star });
     }

packages/hydrooj/src/script/problemStat.ts

@@ -48,9 +48,26 @@ export async function udoc() {
     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 },
+        { $match: { hidden: false, type: { $ne: 'run' }, effective: true } },
         {
             $group: {
                 _id: { domainId: '$domainId', pid: '$pid', uid: '$uid' },
@@ -93,7 +110,7 @@ export async function pdoc() {
 }
 
 export async function run() {
-    return await Promise.all([udoc(), pdoc()]);
+    return await Promise.all([udoc(), pdoc(), psdoc()]);
 }
 
 export const validate = {};