`result`:向查询结果添加自定义字段和方法
Prisma Client 扩展从 4.16.0 及更高版本开始正式可用。它们在 4.7.0 版本中作为预览版推出。如果您使用的是早于 4.16.0 的版本,请确保启用了 clientExtensions
预览功能标志。
您可以使用 result
Prisma Client 扩展 组件类型向查询结果添加自定义字段和方法。
使用 $extends
客户端级别方法 创建一个扩展客户端。扩展客户端是标准 Prisma Client 的一个变体,它被一个或多个扩展所包装。
要向查询结果添加自定义 字段 或 方法,请使用以下结构。在此示例中,我们将自定义字段 myComputedField
添加到 user
模型查询的结果中。
const prisma = new PrismaClient().$extends({
name?: 'name',
result?: {
user: { // in this case, we extend the `user` model
myComputedField: { // the name of the new computed field
needs: { ... },
compute() { ... }
},
},
},
});
参数如下:
name
:(可选)指定扩展的名称,该名称将显示在错误日志中。result
:为查询结果定义新的字段和方法。needs
:一个对象,描述结果字段的依赖关系。compute
:定义在访问虚拟字段时如何计算它的方法。
向查询结果添加自定义字段
您可以使用 result
扩展组件向查询结果添加字段。这些字段是在运行时计算的,并且是类型安全的。
在以下示例中,我们向 user
模型添加了一个名为 fullName
的新虚拟字段。
const prisma = new PrismaClient().$extends({
result: {
user: {
fullName: {
// the dependencies
needs: { firstName: true, lastName: true },
compute(user) {
// the computation logic
return `${user.firstName} ${user.lastName}`
},
},
},
},
})
const user = await prisma.user.findFirst()
// return the user's full name, such as "John Doe"
console.log(user.fullName)
在上面的示例中,compute
的输入 user
会根据 needs
中定义的对象自动类型化。firstName
和 lastName
的类型为 string
,因为它们是在 needs
中指定的。如果它们没有在 needs
中指定,则无法访问它们。
在另一个计算字段中重用计算字段
以下示例以类型安全的方式计算用户的标题和全名。titleFullName
是一个计算字段,它重用了 fullName
计算字段。
const prisma = new PrismaClient()
.$extends({
result: {
user: {
fullName: {
needs: { firstName: true, lastName: true },
compute(user) {
return `${user.firstName} ${user.lastName}`
},
},
},
},
})
.$extends({
result: {
user: {
titleFullName: {
needs: { title: true, fullName: true },
compute(user) {
return `${user.title} (${user.fullName})`
},
},
},
},
})
字段的注意事项
-
出于性能原因,Prisma Client 在访问时计算结果,而不是在检索时计算结果。
-
您只能创建基于标量字段的计算字段。
-
您只能将计算字段与
select
一起使用,并且不能对其进行聚合。例如const user = await prisma.user.findFirst({
select: { email: true },
})
console.log(user.fullName) // undefined
向结果对象添加自定义方法
您可以使用 result
组件向查询结果添加方法。以下示例向结果对象添加了一个名为 save
的新方法。
const prisma = new PrismaClient().$extends({
result: {
user: {
save: {
needs: { id: true },
compute(user) {
return () =>
prisma.user.update({ where: { id: user.id }, data: user })
},
},
},
},
})
const user = await prisma.user.findUniqueOrThrow({ where: { id: someId } })
user.email = '[email protected]'
await user.save()
将 omit
查询选项与 result
扩展组件一起使用
您可以将 omit
(预览版)选项 与 自定义字段 和自定义字段所需的字段一起使用。
从查询结果中省略自定义字段所需的字段
如果您省略了作为自定义字段依赖项的字段,即使它不会包含在查询结果中,它仍然会从数据库中读取。
以下示例省略了 password
字段,它是自定义字段 sanitizedPassword
的依赖项
const xprisma = prisma.$extends({
result: {
user: {
sanitizedPassword: {
needs: { password: true },
compute(user) {
return sanitize(user.password)
},
},
},
},
})
const user = await xprisma.user.findFirstOrThrow({
omit: {
password: true,
},
})
在这种情况下,虽然 password
被从结果中省略,但它仍然会从数据库中查询,因为它 是 sanitizedPassword
自定义字段的依赖项。
从查询结果中省略自定义字段及其依赖项
为了确保省略的字段根本不会从数据库中查询,您必须同时省略自定义字段及其依赖项。
以下示例同时省略了自定义字段 sanitizedPassword
和依赖的 password
字段
const xprisma = prisma.$extends({
result: {
user: {
sanitizedPassword: {
needs: { password: true },
compute(user) {
return sanitize(user.password)
},
},
},
},
})
const user = await xprisma.user.findFirstOrThrow({
omit: {
sanitizedPassword: true,
password: true,
},
})
在这种情况下,同时省略 password
和 sanitizedPassword
会将它们都从结果中排除,并防止 password
字段从数据库中读取。