From 8e574b5ef6b484e26ca4594200d42ac4ecbb42fa Mon Sep 17 00:00:00 2001
From: zkldi <ktchidev@gmail.com>
Date: Fri, 5 Nov 2021 00:35:12 +0000
Subject: [PATCH] Refactor the set-indexes function to work on the current DB

---
 server/src/external/mongo/indexes.ts | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/server/src/external/mongo/indexes.ts b/server/src/external/mongo/indexes.ts
index 64838439f..3e8c47ca9 100644
--- a/server/src/external/mongo/indexes.ts
+++ b/server/src/external/mongo/indexes.ts
@@ -1,5 +1,5 @@
 /* eslint-disable no-await-in-loop */
-import monk from "monk";
+import monk, { IMonkManager } from "monk";
 import { IndexOptions } from "mongodb";
 import CreateLogCtx from "lib/logger/logger";
 import { ServerTypeInfo } from "lib/setup/config";
@@ -121,11 +121,7 @@ for (const game of ServerTypeInfo.supportedGames) {
 	}
 }
 
-export async function SetIndexes(mongoURL: string, reset: boolean) {
-	const db = monk(mongoURL);
-
-	logger.debug(`Starting indexing for ${mongoURL}...`);
-
+export async function SetIndexesWithDB(db: IMonkManager, reset: boolean) {
 	const collections = (await db.listCollections()).map((e) => e.name);
 
 	for (const collection in indexes) {
@@ -150,6 +146,12 @@ export async function SetIndexes(mongoURL: string, reset: boolean) {
 	}
 
 	logger.debug("Done.");
+}
 
-	await db.close();
+export async function SetIndexes(mongoUrl: string, reset: boolean) {
+	const monkDb = monk(mongoUrl);
+
+	await SetIndexesWithDB(monkDb, reset);
+
+	await monkDb.close();
 }