Prisma Client 扩展的类型提取与模块化实践

本文旨在解决prisma client扩展在进行模块化重构时遇到的类型定义难题。通过深入探讨typescript的类型推断机制和prisma `$extends`方法的内部类型结构，我们将演示如何精确地提取并定义分离的扩展模块类型，从而提升代码的可维护性和可读性，并提供清晰的实现步骤和示例代码。

引言：Prisma Client 扩展与模块化需求

Prisma Client 扩展提供了一种强大机制，允许开发者在不修改核心Client代码的情况下，为数据库操作添加自定义逻辑，例如在查询执行前后插入业务规则。随着项目复杂性增加，将所有扩展逻辑集中在一个大型对象中会导致代码难以维护和阅读。因此，将扩展逻辑拆分到独立文件中进行模块化管理成为必然需求。然而，这一过程中的主要挑战在于如何为这些分离的扩展模块精确地定义TypeScript类型。

挑战：分离扩展的类型定义

当我们将Prisma Client扩展（特别是针对特定模型查询的扩展）从主$extends调用中抽离出来时，TypeScript的类型推断能力往往难以直接为这些独立的对象提供准确的类型。例如，考虑以下将company模型查询扩展分离的场景：

// prismaClient.ts
import { PrismaClient } from '@prisma/client';
import { companyExtensions } from './extensions/companyExtensions'; // 尚未定义类型

const _prismaClient = new PrismaClient();

const prismaClient = _prismaClient.$extends({
  query: {
    company: companyExtensions, // 这里 companyExtensions 的类型需要明确
  },
});

// companyExtensions.ts
// 如何为 companyExtensions 定义精确的类型？
export const companyExtensions: unknown = { // 这里的 unknown 是待解决的问题
  update: async ({ args, query }) => {
    // 示例业务逻辑
    if (args.data?.status === 'DECLINED') {
      args.data.user = {
        update: {
          accountLocked: 'COMPANY_DECLINED',
        },
      };
    }
    return query(args);
  },
};

直接尝试使用Parameters[0]可能会得到一个过于宽泛或难以直接应用于嵌套对象的复杂类型。Prisma 官方提供的defineExtension函数主要面向通用、可分发的扩展，其设计目标并非为特定模型操作提供精细的args类型推断。因此，我们需要一种更精确的方法来提取这些内部扩展的类型。

解决方案：利用 Parameters 和 Extract 提取类型

解决此问题的关键在于利用TypeScript的内置工具类型Parameters和Extract，结合Prisma Client的内部类型结构。我们的目标是首先获取传递给_prismaClient.$extends方法的完整参数类型，然后从中精确地提取出我们所需的部分。

1. 获取完整的扩展参数类型

首先，我们需要获取_prismaClient.$extends方法接受的参数的类型。这里的_prismaClient指的是未经过任何扩展的原始Prisma Client实例。

// 定义一个基础的Prisma Client实例（仅用于类型推断）
import { PrismaClient } from '@prisma/client';
const _prismaClientForTypeInference = new PrismaClient();

// 获取 $extends 方法的参数类型
type FullExtensionArgs = Parameters[0];

FullExtensionArgs将是一个复杂的联合类型或交叉类型，它包含了Prisma Client扩展可能接受的所有结构。

2. 精确提取目标扩展类型

由于FullExtensionArgs可能非常通用，我们需要使用Extract工具类型来将其窄化到我们实际使用的扩展结构。Extract会从UnionType中提取所有可赋值给MemberType的成员。在这里，我们通常会使用一个包含name?: string属性的对象字面量来“锚定”或选择出代表客户端扩展配置的特定类型分支。

type ExtensionArgs = Extract;

经过这一步，ExtensionArgs现在代表了传递给$extends方法的整个配置对象的类型，其中包含了query、model、client等顶级属性及其对应的复杂类型。

成新网络商城购物系统

使用模板与程序分离的方式构建，依靠专门设计的数据库操作类实现数据库存取，具有专有错误处理模块，通过 Email 实时报告数据库错误，除具有满足购物需要的全部功能外，成新商城购物系统还对购物系统体系做了丰富的扩展，全新设计的搜索功能，自定义成新商城购物系统代码功能代码已经全面优化，杜绝SQL注入漏洞前台测试用户名：admin密码：admin888后台管理员名：admin密码：admin888

下载

3. 提取特定模型查询扩展的类型

现在，我们已经有了整个扩展配置的类型ExtensionArgs，接下来就可以从中访问我们需要的特定模型查询扩展的类型。在我们的例子中，companyExtensions是query.company下的一个对象。

type CompanyQueryExtension = ExtensionArgs['query']['company'];

CompanyQueryExtension现在精确地表示了company模型下所有查询扩展的类型，包括update方法及其内部的args和query参数的正确类型。

实践：模块化 Prisma Client 扩展

下面是结合上述类型提取方法，实现Prisma Client扩展模块化的完整示例。

1. src/extensions/companyExtensions.ts 文件

首先，在单独的文件中定义companyExtensions，并使用我们刚刚提取的CompanyQueryExtension类型。

// src/extensions/companyExtensions.ts
import { PrismaClient } from '@prisma/client';

// 注意：这个 _prismaClientForTypeInference 实例仅用于类型推断，不应在运行时实际使用。
// 它的目的是提供一个未扩展的 PrismaClient 类型上下文，以便 TypeScript 能够正确推断 $extends 的参数类型。
const _prismaClientForTypeInference = new PrismaClient();

// 步骤1: 获取 $extends 方法的完整参数类型
type FullExtensionArgs = Parameters[0];

// 步骤2: 精确提取目标扩展类型
type ExtensionArgs = Extract;

// 步骤3: 提取特定模型查询扩展的类型
export type CompanyQueryExtension = ExtensionArgs['query']['company'];

export const companyExtensions: CompanyQueryExtension = {
  update: async ({ args, query }) => {
    // 示例业务逻辑：如果公司状态被设置为 DECLINED，则锁定关联用户的账户
    if (args.data?.status === 'DECLINED') {
      args.data.user = {
        update: {
          accountLocked: 'COMPANY_DECLINED',
        },
      };
    }
    return query(args); // 继续执行原始的 update 操作
  },
  // 可以在这里添加其他 company 模型的查询扩展，例如 findMany, create 等
  // findMany: async ({ args, query }) => { /* ... */ return query(args); }
};

通过这种方式，companyExtensions对象现在拥有了正确的类型定义，并且其内部的update方法的args和query参数都将获得Prisma提供的精确类型提示。

2. src/prismaClient.ts 文件

然后，在主Prisma Client实例化文件中引入并使用这个分离的扩展模块。

// src/prismaClient.ts
import { PrismaClient } from '@prisma/client';
import { companyExtensions } from './extensions/companyExtensions'; // 导入分离的扩展

const _prismaClient = new PrismaClient(); // 实际的未扩展Prisma Client实例

export const prismaClient = _prismaClient.$extends({
  query: {
    company: companyExtensions, // 直接使用导入的扩展对象
  },
  // 可以添加其他模型或客户端级别的扩展
  // model: { /* ... */ },
  // client: { /* ... */ }
});

// 导出扩展后的客户端供应用程序使用
export type PrismaClientExtended = typeof prismaClient;

现在，prismaClient实例将包含companyExtensions中定义的逻辑，并且整个设置在TypeScript层面是完全