diff --git a/apps/server/src/api/middlewares/db.ts b/apps/server/src/api/middlewares/db.ts
index 6a904a6..eabc1e6 100644
--- a/apps/server/src/api/middlewares/db.ts
+++ b/apps/server/src/api/middlewares/db.ts
@@ -1,13 +1,11 @@
 import { os } from '@orpc/server'
 import { getDb } from '@/db'
 
-const IS_SERVERLESS = false // TODO: 这里需要优化
-
 export const db = os.middleware(async ({ context, next }) => {
   return next({
     context: {
       ...context,
-      db: getDb(IS_SERVERLESS),
+      db: getDb(),
     },
   })
 })
diff --git a/apps/server/src/db/index.ts b/apps/server/src/db/index.ts
index 4bbd860..78172b6 100644
--- a/apps/server/src/db/index.ts
+++ b/apps/server/src/db/index.ts
@@ -16,8 +16,8 @@ export type Db = ReturnType<typeof createDb>
 export const getDb = (() => {
   let db: Db | null = null
 
-  return (serverless: boolean = false): Db => {
-    if (serverless) {
+  return (singleton = true): Db => {
+    if (!singleton) {
       return createDb()
     }