From a7d6265f97a12d36007a3f910534d71a6e5bbe28 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Wed, 29 May 2024 21:59:53 +0800
Subject: [PATCH] core: fix record stat update

---
 packages/hydrooj/src/model/record.ts   |  2 +-
 packages/hydrooj/src/model/schedule.ts | 26 --------------------------
 packages/hydrooj/src/service/api.ts    | 10 ++++++++++
 3 files changed, 11 insertions(+), 27 deletions(-)

diff --git a/packages/hydrooj/src/model/record.ts b/packages/hydrooj/src/model/record.ts
index 472fcf318..093cf0664 100644
--- a/packages/hydrooj/src/model/record.ts
+++ b/packages/hydrooj/src/model/record.ts
@@ -264,7 +264,7 @@ export function apply(ctx: Context) {
     ]));
     ctx.on('domain/delete', (domainId) => RecordModel.coll.deleteMany({ domainId }));
     ctx.on('record/judge', async (rdoc, updated) => {
-        if (rdoc.status === STATUS.STATUS_ACCEPTED && updated && !rdoc.contest) {
+        if (rdoc.status === STATUS.STATUS_ACCEPTED && updated) {
             await RecordModel.collStat.updateOne({
                 _id: rdoc._id,
             }, {
diff --git a/packages/hydrooj/src/model/schedule.ts b/packages/hydrooj/src/model/schedule.ts
index 3d14f32bc..4c78f7ea6 100644
--- a/packages/hydrooj/src/model/schedule.ts
+++ b/packages/hydrooj/src/model/schedule.ts
@@ -1,13 +1,10 @@
 import moment from 'moment-timezone';
 import { Filter, ObjectId } from 'mongodb';
-import { Time } from '@hydrooj/utils';
-import { STATUS } from '@hydrooj/utils/lib/status';
 import { Context } from '../context';
 import { Schedule } from '../interface';
 import { Logger } from '../logger';
 import db from '../service/db';
 import type { WorkerService } from '../service/worker';
-import * as DocumentModel from './document';
 import RecordModel from './record';
 
 const logger = new Logger('model/schedule');
@@ -67,29 +64,6 @@ export async function apply(ctx: Context) {
         ScheduleModel.Worker = c.worker;
         c.worker.addHandler('task.daily', async () => {
             await RecordModel.coll.deleteMany({ contest: { $in: [RecordModel.RECORD_PRETEST, RecordModel.RECORD_GENERATE] } });
-            const tdocs = await db.collection('document').find({
-                docType: DocumentModel.TYPE_CONTEST,
-                endAt: { $gt: new Date(Date.now() - 2 * Time.day) },
-            }).project({ docId: 1, beginAt: 1 }).toArray();
-            const first = Math.min(...tdocs.map((i) => i.beginAt.getTime()));
-            const bulk = RecordModel.collStat.initializeUnorderedBulkOp();
-            const cursor = RecordModel.coll.find({
-                status: STATUS.STATUS_ACCEPTED,
-                contest: { $in: tdocs.map((i) => i.docId) },
-                _id: { $gt: Time.getObjectID(new Date(first)) },
-            });
-            for await (const rdoc of cursor) {
-                bulk.find({ _id: rdoc._id }).upsert().updateOne({
-                    domainId: rdoc.domainId,
-                    pid: rdoc.pid,
-                    uid: rdoc.uid,
-                    time: rdoc.time,
-                    memory: rdoc.memory,
-                    length: rdoc.code?.length || 0,
-                    lang: rdoc.lang,
-                });
-            }
-            await bulk.execute();
             await global.Hydro.script.rp?.run({}, new Logger('task/rp').debug);
             await global.Hydro.script.problemStat?.run({}, new Logger('task/problem').debug);
             if (global.Hydro.model.system.get('server.checkUpdate') && !(new Date().getDay() % 3)) {
diff --git a/packages/hydrooj/src/service/api.ts b/packages/hydrooj/src/service/api.ts
index 8f66a322b..7806620d9 100644
--- a/packages/hydrooj/src/service/api.ts
+++ b/packages/hydrooj/src/service/api.ts
@@ -200,6 +200,16 @@ class ApiService extends Service {
         unions[typeName] = unionTypes.join(' | ');
         this.rebuild();
     }
+
+    query(query: string, args: Record<string, any>) {
+        return graphql({
+            schema,
+            source: query,
+            rootValue: root,
+            contextValue: this,
+            variableValues: args,
+        });
+    }
 }
 
 export const sideEffect = true;