core: 优化任务分配逻辑

packages/hydrooj/package.json

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

packages/hydrooj/src/handler/judge.ts

@@ -1,5 +1,6 @@
 import { ObjectID } from 'mongodb';
 import { JudgeResultBody, Rdoc, TestCase } from '../interface';
+import { sleep } from '../utils';
 import * as record from '../model/record';
 import * as problem from '../model/problem';
 import * as builtin from '../model/builtin';
@@ -113,16 +114,32 @@ class JudgeHandler extends Handler {
 }
 
 class JudgeConnectionHandler extends ConnectionHandler {
-    processing: any;
+    processing: any = null;
+
+    async prepare() {
+        this.newTask();
+    }
+
+    async newTask() {
+        if (this.processing) return;
+        let t;
+        while (!t) {
+            // eslint-disable-next-line no-await-in-loop
+            await sleep(100);
+            // eslint-disable-next-line no-await-in-loop
+            t = await task.getFirst({ type: 'judge' });
+        }
+        this.send({ task: t });
+        this.processing = t;
+    }
 
     async message(msg) {
+        console.log(msg);
         if (msg.key === 'next') await next(msg);
         else if (msg.key === 'end') {
             await end({ judger: this.user._id, ...msg });
             this.processing = null;
-            const t = await task.getFirst({ type: 'judge' });
+            await this.newTask();
-            this.send({ task: t });
-            this.processing = t;
         }
     }
 

packages/hydrooj/src/utils.ts

@@ -193,3 +193,11 @@ export function buildProjection(fields: string[]): Record<string, 1> {
     for (const k of fields) o[k] = 1;
     return o;
 }
+
+export function sleep(timeout: number) {
+    return new Promise((resolve) => {
+        setTimeout(() => {
+            resolve();
+        }, timeout);
+    });
+}