refactor: 重构数据库实例管理，支持按需创建与上下文注入

- 移除默认导出的 db 实例，改为按需创建数据库实例
- 使用数据库中间件提供上下文中的数据库实例，统一通过上下文获取数据库连接并执行操作。
- 添加数据库中间件以在请求上下文中注入数据库实例，支持服务器端全局缓存和无服务器环境下的每次新建实例。
- 导出中间件模块中的数据库相关功能。

src/db/index.ts

@@ -11,5 +11,3 @@ export function createDb() {
     schema,
   })
 }
-
-export const db = createDb()

src/orpc/handlers/todo.ts

@@ -6,8 +6,8 @@ import {
   createUpdateSchema,
 } from 'drizzle-zod'
 import { z } from 'zod'
-import { db } from '@/db'
 import { todoTable } from '@/db/schema'
+import { dbProvider } from '../middlewares'
 
 const selectSchema = createSelectSchema(todoTable)
 
@@ -26,8 +26,9 @@ const updateSchema = createUpdateSchema(todoTable).omit({
 export const list = os
   .input(z.void())
   .output(z.array(selectSchema))
-  .handler(async () => {
+  .use(dbProvider)
-    const todos = await db.query.todoTable.findMany({
+  .handler(async ({ context }) => {
+    const todos = await context.db.query.todoTable.findMany({
       orderBy: (todos, { desc }) => [desc(todos.createdAt)],
     })
     return todos
@@ -36,10 +37,11 @@ export const list = os
 export const create = os
   .input(insertSchema)
   .output(selectSchema)
-  .handler(async ({ input }) => {
+  .use(dbProvider)
-    const [newTodo] = await db
+  .handler(async ({ context, input }) => {
+    const [newTodo] = await context.db
       .insert(todoTable)
-      .values({ title: input.title })
+      .values(input)
       .returning()
 
     if (!newTodo) {
@@ -57,8 +59,9 @@ export const update = os
     }),
   )
   .output(selectSchema)
-  .handler(async ({ input }) => {
+  .use(dbProvider)
-    const [updatedTodo] = await db
+  .handler(async ({ context, input }) => {
+    const [updatedTodo] = await context.db
       .update(todoTable)
       .set(input.data)
       .where(eq(todoTable.id, input.id))
@@ -78,6 +81,7 @@ export const remove = os
     }),
   )
   .output(z.void())
-  .handler(async ({ input }) => {
+  .use(dbProvider)
-    await db.delete(todoTable).where(eq(todoTable.id, input.id))
+  .handler(async ({ context, input }) => {
+    await context.db.delete(todoTable).where(eq(todoTable.id, input.id))
   })

src/orpc/middlewares/db.ts

@@ -0,0 +1,29 @@
+import { os } from '@orpc/server'
+import { createDb } from '@/db'
+
+const IS_SERVERLESS = false // TODO: 这里需要优化
+
+let globalDb: ReturnType<typeof createDb> | null = null
+
+function getDb() {
+  if (IS_SERVERLESS) {
+    return createDb()
+  }
+
+  if (!globalDb) {
+    globalDb = createDb()
+  }
+
+  return globalDb
+}
+
+export const dbProvider = os.middleware(async ({ context, next }) => {
+  const db = getDb()
+
+  return next({
+    context: {
+      ...context,
+      db,
+    },
+  })
+})

src/orpc/middlewares/index.ts

@@ -0,0 +1 @@
+export * from './db'