From 6795730485e8243e9050416217de513db8e7bd70 Mon Sep 17 00:00:00 2001
From: imbytecat <imbytecat@gmail.com>
Date: Thu, 2 Apr 2026 07:48:16 +0800
Subject: [PATCH] =?UTF-8?q?refactor(db):=20=E6=9A=B4=E9=9C=B2=20closeDB()?=
 =?UTF-8?q?=20=E5=87=BD=E6=95=B0=E4=BB=A5=E6=94=AF=E6=8C=81=E8=BF=9E?=
 =?UTF-8?q?=E6=8E=A5=E6=B1=A0=E6=B8=85=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/server/db/index.ts | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/server/db/index.ts b/src/server/db/index.ts
index fc8e6be..21201f1 100644
--- a/src/server/db/index.ts
+++ b/src/server/db/index.ts
@@ -10,11 +10,16 @@ export const createDB = () =>
 
 export type DB = ReturnType<typeof createDB>
 
-export const getDB = (() => {
-  let db: DB | null = null
+let _db: DB | null = null
 
-  return (): DB => {
-    db ??= createDB()
-    return db
+export const getDB = (): DB => {
+  _db ??= createDB()
+  return _db
+}
+
+export const closeDB = async () => {
+  if (_db) {
+    await _db.$client.end()
+    _db = null
   }
-})()
+}