From 5f8785362c6b8119017ba2cc539549ef2f9515e0 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Tue, 8 Oct 2024 18:56:08 +0800
Subject: [PATCH] core: db: add collectionMap support

---
 packages/hydrooj/src/service/db.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/hydrooj/src/service/db.ts b/packages/hydrooj/src/service/db.ts
index 9430b9ea6..6dee3f3fd 100644
--- a/packages/hydrooj/src/service/db.ts
+++ b/packages/hydrooj/src/service/db.ts
@@ -22,6 +22,7 @@ interface MongoConfig {
     url?: string,
     uri?: string,
     prefix?: string,
+    collectionMap?: Record<string, string>,
 }
 
 class MongoService {
@@ -54,8 +55,9 @@ class MongoService {
     }
 
     public collection<K extends keyof Collections>(c: K) {
-        if (this.opts.prefix) return this.db.collection<Collections[K]>(`${this.opts.prefix}.${c}`);
-        return this.db.collection<Collections[K]>(c);
+        let coll = this.opts.prefix ? `${this.opts.prefix}.${c}` : c;
+        if (this.opts.collectionMap?.[coll]) coll = this.opts.collectionMap[coll];
+        return this.db.collection<Collections[K]>(coll);
     }
 
     public async fixExpireAfter() {