在T3 Stack项目中优化Prisma Client的类型推断

在基于T3 Stack构建的项目中，Prisma ORM是常用的数据库访问层解决方案。开发者经常会遇到一个典型问题：当使用Prisma Client的$extends方法扩展模型时，TypeScript无法正确推断出自定义方法的类型。

问题背景

在标准的T3 Stack项目设置中，Prisma Client通常被初始化为一个单例模式。常见的实现方式如下：

import { PrismaClient } from '@prisma/client'

const prisma = globalThis.prisma || new PrismaClient()

if (process.env.NODE_ENV !== 'production') {
  globalThis.prisma = prisma
}

export default prisma

这种实现方式虽然简单，但在使用$extends方法扩展Prisma Client时会出现类型推断问题。例如：

const extendedPrisma = prisma.$extends({
  model: {
    user: {
      async findByEmail(email: string) {
        return prisma.user.findUnique({ where: { email } })
      }
    }
  }
})

当尝试使用extendedPrisma.user.findByEmail()时，TypeScript会报错，因为它无法识别这个新添加的方法。

解决方案

Prisma官方文档推荐了一种更完善的类型安全解决方案：

import { PrismaClient } from '@prisma/client'

const prismaClientSingleton = () => {
  return new PrismaClient()
}

declare global {
  var prisma: undefined | ReturnType<typeof prismaClientSingleton>
}

export const prisma = globalThis.prisma ?? prismaClientSingleton()

if (process.env.NODE_ENV !== 'production') globalThis.prisma = prisma

这个解决方案的关键改进在于：

1. 将Prisma Client的创建封装在一个函数中
2. 使用ReturnType获取该函数的返回类型
3. 在全局类型声明中使用这个返回类型

实现原理

这种解决方案之所以有效，是因为：

1. 函数封装：通过将Prisma Client的创建封装在函数中，我们可以利用TypeScript的类型推断能力
2. ReturnType工具类型：ReturnType能够动态获取函数的返回类型，包括通过$extends添加的任何扩展方法
3. 全局类型声明：在全局作用域中声明类型，确保在整个应用中类型一致性

实际应用

在实际项目中应用这个解决方案后：

1. 开发者可以安全地使用$extends方法添加自定义查询
2. TypeScript能够正确识别所有扩展方法
3. 保持了单例模式的性能优势
4. 在开发和生产环境中都能正常工作

最佳实践

对于T3 Stack项目，建议：

1. 在src/server/db.ts中实现这个改进后的Prisma Client初始化
2. 对于复杂的扩展逻辑，可以考虑单独创建扩展文件
3. 在团队项目中，确保所有成员都使用相同的Prisma Client实例

这种改进不仅解决了类型推断问题，还为项目提供了更好的类型安全性和开发体验，是T3 Stack项目中Prisma集成的推荐实践。