From cc9040038d7b99d7787a4c4f043427a8ddae3b07 Mon Sep 17 00:00:00 2001
From: wancat <wancat@wancat.cc>
Date: Sat, 14 Oct 2023 15:49:13 +0800
Subject: [PATCH] fix: change connect db error handling

---
 packages/models/index.js | 3 +--
 server/src/index.js      | 7 ++++++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/packages/models/index.js b/packages/models/index.js
index e84cac8..3b2bce9 100644
--- a/packages/models/index.js
+++ b/packages/models/index.js
@@ -16,7 +16,6 @@ export default async function connectDB(uri) {
     return db.connection.getClient()
   } catch (err) {
     console.error(err)
-    process.exit(1)
-    return null
+    throw new Error('Failed to connect to MongoDB')
   }
 }
diff --git a/server/src/index.js b/server/src/index.js
index 3bb154a..dbb23d0 100644
--- a/server/src/index.js
+++ b/server/src/index.js
@@ -17,7 +17,12 @@ import expiryRemovalScheduler from './containers/expiryChecker.js'
 import { createTestUser } from './utils/auth.js'
 
 async function main() {
-  await connectDB(config.db)
+  try {
+    await connectDB(config.db)
+  } catch (err) {
+    console.error(err)
+    process.exit(1)
+  }
 
   await createTestUser()