Prisma Client Extensions 中异步计算字段的实现策略

async function getdata(): Promise<string> {
  // 模拟异步操作
  return new Promise(resolve => setTimeout(() => resolve(" (外部数据!)"), 100));
}

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();

async function getdata(): Promise<string> {
  // 模拟异步操作
  return new Promise(resolve => setTimeout(() => resolve(" (外部数据!)"), 100));
}

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); // 输出: Joe (30y) (外部数据!)
  }
}

main();

async function getdata(): Promise<string> {
  // 模拟异步操作
  return new Promise(resolve => setTimeout(() => resolve(" (外部数据!)"), 100));
}

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

// 定义自定义查询方法的参数类型
export type UserFindManyWithData = {
  where?: Prisma.UserWhereInput;
  select?: Prisma.UserSelect;
};

const prisma = new PrismaClient().$extends({
  model: {
    user: {
      /**
       * 查找多个用户并为每个用户添加异步计算的 nameAndAge 字段。
       * @param {UserFindManyWithData} params - 包含 where 和 select 条件的对象。
       * @returns {Promise<Array<Prisma.UserGetPayload<{select: typeof params.select}> & { nameAndAge?: string }>>}
       */
      async findManyWithData({ where, select }: UserFindManyWithData) {
        // 首先执行原始查询获取用户数据
        const users = await prisma.user.findMany({ where, select });

        // 遍历用户列表，为每个用户添加异步计算的字段
        // 注意：这种在循环中 await 的方式可能不是最高效的，
        // 对于大量数据，考虑使用 Promise.all 来并行处理异步操作。
        for (const user of users) {
          // 确保 user 对象可以动态添加属性
          (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: 20 }
    },
    select: {
      id: true,
      name: true,
      age: true
    }
  });

  console.log(usersWithData);
  // 示例输出:
  // [
  //   { id: 1, name: 'Alice', age: 25, nameAndAge: 'Alice (25y) (外部数据!)' },
  //   { id: 2, name: 'Bob', age: 30, nameAndAge: 'Bob (30y) (外部数据!)' }
  // ]
}

main();