
Prisma Client Extensions 引入的 result 扩展允许开发者为模型添加派生字段。然而,其 compute 函数被设计为同步执行,以便在访问时以最小开销计算字段。这意味着,如果在 compute 函数内部直接调用异步函数并尝试使用 await,例如从外部服务获取数据,结果将不会是期望的最终值,而是处于 pending 状态的 Promise 对象,或者 Promise 对象本身被字符串化。
示例问题代码:
async function getdata(): Promise<string> {
return "external data here!";
}
import { PrismaClient } from "@prisma/client";
const prisma = new PrismaClient().$extends({
result: {
user: {
nameAndAge: {
needs: { name: true, age: true },
// 尝试在 compute 中直接使用 await
async compute(user) {
return `${user.name} (${user.age}y) ${await getdata()}`;
},
},
},
},
});
async function main() {
const user = await prisma.user.findFirst();
console.log(user?.nameAndAge); // 输出: Promise { <pending> }
}
main();这种行为是 Prisma 设计使然,旨在保持 result 扩展的高性能和低开销。对于需要异步操作的场景,Prisma 提供了其他更合适的扩展点。
一种解决 compute 函数同步限制的方法是,让 compute 函数本身返回一个异步函数。这样,异步操作的执行就被延迟到实际需要该计算字段值的时候。
实现方式:
将 compute 函数设计为返回一个 async 函数。当访问这个扩展字段时,你需要显式地 await 这个返回的异步函数来获取最终结果。
示例代码:
async function getdata(): Promise<string> {
return "external data here!";
}
import { PrismaClient } from "@prisma/client";
const prisma = new PrismaClient().$extends({
result: {
user: {
// 字段名可以更具描述性,例如 getNameAndAge
getNameAndAge: {
needs: { name: true, age: true },
compute(user) {
// compute 函数返回一个异步函数
return async () => (`${user.name} (${user.age}y) ${await getdata()}`);
},
},
},
},
});
async function main() {
const users = await prisma.user.findMany(); // 获取用户列表
if (users.length > 0) {
// 访问扩展字段时,需要 await 调用返回的异步函数
const joeNameAndAge = await users[0].getNameAndAge();
console.log(joeNameAndAge); // 输出: Sonia Lomo (25y) external data here!
}
}
main();注意事项:
对于更复杂的数据聚合、需要提前执行异步操作或希望在查询层面控制数据处理的场景,model 扩展提供了更强大的能力。通过 model 扩展,你可以在现有模型上定义全新的异步方法,这些方法可以完全控制查询逻辑和数据转换。
实现方式:
在 model 扩展中定义一个异步方法,该方法可以执行 Prisma 客户端的查询,并在返回结果之前进行异步的数据处理和字段添加。
示例代码:
async function getdata(): Promise<string> {
return "external data here!";
}
import { PrismaClient, Prisma } from "@prisma/client";
// 定义自定义查询方法的参数类型
export type UserFindManyWithData = {
where?: Prisma.UserWhereInput;
select?: Prisma.UserSelect;
};
const prisma = new PrismaClient().$extends({
model: {
user: {
async findManyWithData({ where, select }: UserFindManyWithData) {
// 执行原始查询
const users = await prisma.user.findMany({ where, select });
// 遍历结果,进行异步数据处理并添加新字段
// 注意:这种逐个 await 的方式在处理大量数据时可能影响性能
// 更好的做法是使用 Promise.all() 并行处理异步操作
for (const user of users) {
// 假设 user 类型可以动态添加属性
// 在实际应用中,你可能需要一个更具体的类型定义来包含 nameAndAge
(user as any).nameAndAge = `${user.name} (${user.age}y) ${await getdata()}`;
}
return users;
},
},
},
});
async function main() {
const usersWithData = await prisma.user.findManyWithData({
where: { age: { gte: 18 } },
select: { id: true, name: true, age: true },
});
console.log(usersWithData);
// 输出:
// [
// { id: '...', name: 'Sonia Lomo', age: 25, nameAndAge: 'Sonia Lomo (25y) external data here!' },
// ...
// ]
}
main();性能考量:
在上述 model 扩展的示例中,for...of 循环内部的 await 会导致异步操作串行执行。如果 getdata() 是一个耗时操作,并且需要处理大量用户数据,这可能会严重影响性能。为了优化性能,建议使用 Promise.all() 并行处理所有用户的异步数据获取:
// ... (其他代码相同)
const prisma = new PrismaClient().$extends({
model: {
user: {
async findManyWithData({ where, select }: UserFindManyWithData) {
const users = await prisma.user.findMany({ where, select });
// 使用 Promise.all 并行处理所有用户的异步数据
const processedUsers = await Promise.all(
users.map(async (user) => {
const externalData = await getdata();
return {
...user,
nameAndAge: `${user.name} (${user.age}y) ${externalData}`,
};
})
);
return processedUsers;
},
},
},
});
// ... (main 函数相同)何时选择 model 扩展:
Prisma Client Extensions 的 result 扩展适用于同步计算的派生字段。当需要集成异步操作时,开发者应避免在 compute 函数中直接使用 await。本文提供了两种有效的替代方案:
根据你的具体需求和性能考量,选择最适合的 Prisma Client Extensions 方案,可以有效地扩展 Prisma 客户端的功能,同时保持代码的清晰性和效率。
以上就是Prisma Client Extensions 中异步计算字段的解决方案的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号