core&ui: performance optimization

packages/hydrooj/package.json

@@ -1,6 +1,6 @@
 {
     "name": "hydrooj",
-    "version": "4.4.0",
+    "version": "4.4.1",
     "bin": "bin/hydrooj.js",
     "main": "src/plugin-api",
     "module": "src/plugin-api",

packages/hydrooj/src/handler/record.ts

@@ -171,7 +171,7 @@ class RecordDetailHandler extends ContestDetailBaseHandler {
                 canViewCode ||= tsdoc?.attend;
             }
             if (!tsdoc?.attend && !problem.canViewBy(pdoc, this.user)) throw new PermissionError(PERM.PERM_VIEW_PROBLEM_HIDDEN);
-        } else if (!problem.canViewBy(pdoc, this.user)) throw new PermissionError(PERM.PERM_VIEW_PROBLEM_HIDDEN);
+        } else if (pdoc && !problem.canViewBy(pdoc, this.user)) throw new PermissionError(PERM.PERM_VIEW_PROBLEM_HIDDEN);
         if (!canViewCode) {
             rdoc.code = '';
             rdoc.files = {};

packages/hydrooj/src/handler/training.ts

@@ -9,6 +9,7 @@ import problem from '../model/problem';
 import * as system from '../model/system';
 import * as training from '../model/training';
 import user from '../model/user';
+import { DomainModel } from '../plugin-api';
 import * as bus from '../service/bus';
 import { Handler, param, Types } from '../service/server';
 
@@ -97,7 +98,7 @@ class TrainingDetailHandler extends Handler {
         const tdoc = await training.get(domainId, tid);
         await bus.parallel('training/get', tdoc, this);
         let targetUser = this.user._id;
-        let enrollUsers = [];
+        let enrollUsers: number[] = [];
         let shouldCompare = false;
         const pids = training.getPids(tdoc.dag);
         if (this.user.hasPriv(PRIV.PRIV_USER_PROFILE)) {
@@ -109,21 +110,24 @@ class TrainingDetailHandler extends Handler {
             }
         }
         const canViewHidden = this.user.hasPerm(PERM.PERM_VIEW_PROBLEM_HIDDEN) || this.user._id;
-        const [udict, pdict] = await Promise.all([
-            user.getList(domainId, [tdoc.owner, ...enrollUsers]),
+        const [udoc, udocs, dudocs, pdict, psdict, selfPsdict] = await Promise.all([
+            user.getById(domainId, tdoc.owner),
+            user.getMulti({ _id: { $in: enrollUsers } }).project({ _id: 1, uname: 1, avatar: 1 }).toArray(),
+            DomainModel.getDomainUserMulti(domainId, enrollUsers).project({ uid: 1, displayName: 1 }).toArray(),
             problem.getList(domainId, pids, canViewHidden, true),
+            problem.getListStatus(domainId, targetUser, pids),
+            shouldCompare ? problem.getListStatus(domainId, this.user._id, pids) : {},
         ]);
-        const psdict = await problem.getListStatus(domainId, targetUser, pids);
-        const selfPsdict = shouldCompare ? await problem.getListStatus(domainId, this.user._id, pids) : {};
+        const dudict = {};
+        for (const dudoc of dudocs) dudict[dudoc.uid] = dudoc.displayName;
         const donePids = new Set<number>();
         const progPids = new Set<number>();
         for (const pid in psdict) {
             if (!+pid) continue;
             const psdoc = psdict[pid];
             if (psdoc.status) {
-                if (psdoc.status === builtin.STATUS.STATUS_ACCEPTED) {
-                    donePids.add(parseInt(pid, 10));
-                } else progPids.add(parseInt(pid, 10));
+                if (psdoc.status === builtin.STATUS.STATUS_ACCEPTED) donePids.add(+pid);
+                else progPids.add(+pid);
             }
         }
         const nsdict = {};
@@ -149,7 +153,7 @@ class TrainingDetailHandler extends Handler {
             done: doneNids.size === tdoc.dag.length,
         });
         this.response.body = {
-            tdoc, tsdoc, pids, pdict, psdict, ndict, nsdict, udict, enrollUsers, selfPsdict,
+            tdoc, tsdoc, pids, pdict, psdict, ndict, nsdict, udoc, udocs, dudict, selfPsdict,
         };
         if (pjax) {
             const html = await this.renderHTML('partials/training_detail.html', this.response.body);

packages/hydrooj/src/model/domain.ts

@@ -189,6 +189,10 @@ class DomainModel {
         return dudoc;
     }
 
+    static getDomainUserMulti(domainId: string, uids: number[]) {
+        return collUser.find({ domainId, uid: { $in: uids } });
+    }
+
     static setMultiUserInDomain(domainId: string, query: any, params: any) {
         deleteUserCache(domainId);
         return collUser.updateMany({ domainId, ...query }, { $set: params }, { upsert: true });

packages/hydrooj/src/service/storage.ts

@@ -10,7 +10,7 @@ import { createPresignedPost } from '@aws-sdk/s3-presigned-post';
 import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
 import {
     copyFile, createReadStream, createWriteStream, ensureDir,
-    remove, stat, writeFile,
+    existsSync, remove, stat, writeFile,
 } from 'fs-extra';
 import { lookup } from 'mime-types';
 import { Logger } from '../logger';
@@ -273,6 +273,7 @@ class LocalStorageService {
     async get(target: string, path?: string) {
         if (target.includes('..') || target.includes('//')) throw new Error('Invalid path');
         target = resolve(this.dir, target);
+        if (!existsSync(target)) throw new Error('File not found');
         if (path) await copyFile(target, path);
         return createReadStream(target);
     }