From 3c1625a479d686035f45d0e2d237e0914fb0e492 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Sat, 11 May 2024 00:48:52 +0800
Subject: [PATCH] core: fix

---
 packages/hydrooj/src/model/record.ts | 6 ++++--
 packages/hydrooj/src/upgrade.ts      | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/packages/hydrooj/src/model/record.ts b/packages/hydrooj/src/model/record.ts
index dfb70ff29..f5bcc5e5b 100644
--- a/packages/hydrooj/src/model/record.ts
+++ b/packages/hydrooj/src/model/record.ts
@@ -153,6 +153,8 @@ export default class RecordModel {
             data.input = args.input || '';
             isContest = false;
             data.contest = RecordModel.RECORD_PRETEST;
+        } else if (args.type === 'generate') {
+            data.contest = RecordModel.RECORD_GENERATE;
         }
         const res = await RecordModel.coll.insertOne(data);
         bus.broadcast('record/change', data);
@@ -265,7 +267,7 @@ export function apply(ctx: Context) {
                 uid: rdoc.uid,
                 time: rdoc.time,
                 memory: rdoc.memory,
-                code: rdoc.code?.length || 0,
+                length: rdoc.code?.length || 0,
                 lang: rdoc.lang,
             });
         }
@@ -284,7 +286,7 @@ export function apply(ctx: Context) {
             { key: { domainId: 1, pid: 1, uid: 1, _id: -1 }, name: 'basic' },
             { key: { domainId: 1, pid: 1, uid: 1, time: 1 }, name: 'time' },
             { key: { domainId: 1, pid: 1, uid: 1, memory: 1 }, name: 'memory' },
-            { key: { domainId: 1, pid: 1, uid: 1, code: 1 }, name: 'code' },
+            { key: { domainId: 1, pid: 1, uid: 1, length: 1 }, name: 'length' },
         );
     });
 }
diff --git a/packages/hydrooj/src/upgrade.ts b/packages/hydrooj/src/upgrade.ts
index bbc358004..37b03b2a6 100644
--- a/packages/hydrooj/src/upgrade.ts
+++ b/packages/hydrooj/src/upgrade.ts
@@ -660,7 +660,7 @@ const scripts: UpgradeScript[] = [
                 uid: doc.uid,
                 time: doc.time,
                 memory: doc.memory,
-                code: doc.code?.length || 0,
+                length: doc.code?.length || 0,
                 lang: doc.lang,
             });
             if (bulk.length > 500) {