Prisma Client 扩展中异步计算字段的处理策略

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(user) {
          // 尝试直接拼接异步函数的返回值
          return `${user.name} (${user.age}y)` + getdata();
        },
      },
    },
  },
});

async function main() {
  const user = await prisma.user.findFirst();
  console.log(user?.nameAndAge);
}

main();

async compute(user) {
  return `${user.name} (${user.age}y)` + await getdata();
},

async function getdata(): Promise<string> {
  return "external data here!";
}

import { PrismaClient } from "@prisma/client";

const prisma = new PrismaClient().$extends({
  result: {
    user: {
      // 定义一个返回异步函数的字段
      getNameAndAge: {
        needs: { name: true, age: true },
        compute(user) {
          // compute 函数返回一个 async 函数
          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();

import { PrismaClient, Prisma } from "@prisma/client";

async function getdata(): Promise<string> {
  return "external data here!";
}

// 定义一个类型，用于扩展方法的参数
export type UserFindManyWithDataArgs = {
  where?: Prisma.UserWhereInput;
  select?: Prisma.UserSelect;
};

const prisma = new PrismaClient().$extends({
  model: {
    user: {
      /**
       * 自定义查找多个用户的方法，并在其中处理异步数据。
       * @param args 查询参数，包括 where 和 select。
       * @returns 包含异步计算字段的用户列表。
       */
      async findManyWithExternalData({ where, select }: UserFindManyWithDataArgs) {
        const users = await prisma.user.findMany({ where, select });

        // 对每个用户执行异步操作并添加计算字段
        // 注意：这种方式可能导致 N+1 查询问题，更优的策略是使用 Promise.all 并行处理
        const usersWithData = await Promise.all(users.map(async (user) => {
          const externalInfo = await getdata(); // 异步获取外部数据
          // 将计算字段直接添加到用户对象上
          return {
            ...user,
            nameAndAge: `${user.name} (${user.age}y) ${externalInfo}`
          };
        }));

        return usersWithData;
      },
    },
  },
});

async function main() {
  // 使用自定义的 model 扩展方法
  const users = await prisma.user.findManyWithExternalData({
    where: {
      age: { gt: 20 }
    },
    select: {
      id: true,
      name: true,
      age: true
    }
  });

  users.forEach(user => {
    console.log(user.nameAndAge); // 直接访问已计算好的字段
  });
}

main();