更改updateStatus逻辑，导入problem.status

hydro/handler/judge.js

@@ -19,14 +19,12 @@ async function _postJudge(rdoc) {
             contest.updateStatus(rdoc.tid, rdoc.uid, rdoc._id, rdoc.pid, accept, rdoc.score),
         );
     }
-    if (!rdoc.rejudged) {
-        if (await problem.updateStatus(rdoc.pid, rdoc.uid, rdoc._id, rdoc.status)) {
-            if (accept) {
-                tasks.push(
-                    problem.inc(rdoc.pid, 'nAccept', 1),
-                    user.inc(rdoc.uid, 'nAccept', 1),
-                );
-            }
+    if (await problem.updateStatus(rdoc.pid, rdoc.uid, rdoc._id, rdoc.status)) {
+        if (accept && !rdoc.rejudged) {
+            tasks.push(
+                problem.inc(rdoc.pid, 'nAccept', 1),
+                user.inc(rdoc.uid, 'nAccept', 1),
+            );
         }
     }
     await Promise.all(tasks);

hydro/interface.js

@@ -19,20 +19,22 @@ exports.udoc = {
     loginat: new Date(),
     loginip: '0.0.0.0',
 };
+
+exports.pdoc = {
+    _id: new ObjectID(),
+    pid: 'P1000',
+    owner: 0,
+    title: 'No Title',
+    content: 'No Content',
+    nSubmit: 0,
+    nAccept: 0,
+    tag: [],
+    category: [],
+    data: null,
+    hidden: false,
+};
+
 /*
-export interface Pdoc {
-    _id: ObjectID
-    pid: string
-    owner: number
-    title: string
-    content: string
-    nSubmit: number
-    nAccept: number
-    tag: string[]
-    category: string[],
-    data: ObjectID | null
-    hidden: boolean
-}
 export interface TestCase {
     time: number,
     memory: number,

hydro/model/problem.js

@@ -1,8 +1,9 @@
 const { ObjectID } = require('bson');
-const gridfs = require('../service/gridfs');
+const { STATUS_ACCEPTED } = require('./builtin').STATUS;
 const { ProblemNotFoundError } = require('../error');
 const validator = require('../lib/validator');
 const db = require('../service/db.js');
+const gridfs = require('../service/gridfs');
 
 const coll = db.collection('problem');
 const collStatus = db.collection('problem.status');
@@ -141,12 +142,16 @@ async function getListStatus(uid, pids) {
     return r;
 }
 async function updateStatus(pid, uid, rid, status) {
-    const res = await collStatus.findOneAndUpdate(
+    const cur = collStatus.findOne({ pid, uid });
+    if (cur && status !== STATUS_ACCEPTED) {
+        if (cur.status !== STATUS_ACCEPTED) return false;
+    }
+    await collStatus.updateOne(
         { pid, uid },
         { $set: { rid, status } },
         { upsert: true },
     );
-    return res.value;
+    return true;
 }
 async function setTestdata(_id, readStream) {
     const pdoc = await getById(_id);