本文旨在解决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
解决方案：利用 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<typeof _prismaClientForTypeInference.$extends>[0];FullExtensionArgs将是一个复杂的联合类型或交叉类型，它包含了Prisma Client扩展可能接受的所有结构。
2. 精确提取目标扩展类型
由于FullExtensionArgs可能非常通用，我们需要使用Extract工具类型来将其窄化到我们实际使用的扩展结构。Extract
type ExtensionArgs = Extract<FullExtensionArgs, { name?: string }>;经过这一步，ExtensionArgs现在代表了传递给$extends方法的整个配置对象的类型，其中包含了query、model、client等顶级属性及其对应的复杂类型。
使用 Scrapfly SDK（scrapfly-sdk）在 TypeScript 或 JavaScript 中进行网页抓取，用于用户请求抓取 URL、绕过反爬虫防护、渲染页面等场景。
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<typeof _prismaClientForTypeInference.$extends>[0];
// 步骤2: 精确提取目标扩展类型
type ExtensionArgs = Extract<FullExtensionArgs, { name?: string }>;
// 步骤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层面是完全
