Prisma Client Extensions中处理异步计算字段的策略与实践

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)`;
        },
      },
    },
  },
});

async function main() {
  const user = await prisma.user.findFirst();
  console.log(user?.nameAndAge); // 例如: Sonia Lomo (25y)
}

main();

async function getExternalData(): Promise<string> {
  return " [外部数据]";
}

const prisma = new PrismaClient().$extends({
  result: {
    user: {
      nameAndAge: {
        needs: { name: true, age: true },
        compute(user) {
          // 直接调用异步函数
          return `${user.name} (${user.age}y)` + getExternalData();
        },
      },
    },
  },
});

async function main() {
  const user = await prisma.user.findFirst();
  console.log(user?.nameAndAge); // 输出: Sonia Lomo (25y)[object Promise]
}

main();

// 尝试将 compute 声明为 async
async compute(user) {
  return `${user.name} (${user.age}y)` + await getExternalData();
},
// ...
// console.log(user?.nameAndAge); // 输出: Promise { <pending> }

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

async function getExternalData(): Promise<string> {
  // 模拟异步操作，例如网络请求或文件读取
  return new Promise(resolve => setTimeout(() => resolve(" [外部数据]"), 50));
}

const prisma = new PrismaClient().$extends({
  result: {
    user: {
      // 字段名可以改为动词形式，暗示它是一个可调用的方法
      getNameAndAgeWithExternalData: {
        needs: { name: true, age: true },
        compute(user) {
          // compute 函数同步返回一个异步函数
          return async () => (`${user.name} (${user.age}y)${await getExternalData()}`);
        },
      },
    },
  },
});

async function main() {
  const users = await prisma.user.findMany();
  if (users.length > 0) {
    // 当需要获取计算字段的值时，调用并 await 返回的异步函数
    const firstUserNameAndAge = await users[0].getNameAndAgeWithExternalData();
    console.log(firstUserNameAndAge); // 例如: Sonia Lomo (25y) [外部数据]
  }
}

main();

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

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

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

const prisma = new PrismaClient().$extends({
  model: {
    user: {
      /**
       * 查找用户并为每个用户添加异步计算的 nameAndAge 字段。
       * 注意：此方法会修改返回的用户对象，添加一个非Prisma模型定义的字段。
       */
      async findManyWithComputedData({ where, select }: UserFindManyWithComputedDataArgs) {
        // 1. 执行标准的 Prisma 查询
        const users = await prisma.user.findMany({ where, select });

        // 2. 遍历查询结果，为每个用户添加异步计算的字段
        // 注意：在循环中 await 可能会导致性能问题，尤其是在处理大量数据时。
        // 更好的策略是使用 Promise.all 并行处理异步操作。
        for (const user of users) {
          // 假设 user 对象是可扩展的，或者我们将其转换为一个可扩展的类型
          (user as any).nameAndAgeWithExternalData = `${user.name} (${user.age}y)${await getExternalData()}`;
        }

        return users;
      },

      /**
       * 优化版本：使用 Promise.all 并行处理异步数据
       */
      async findManyWithComputedDataOptimized({ where, select }: UserFindManyWithComputedDataArgs) {
        const users = await prisma.user.findMany({ where, select });

        const computedDataPromises = users.map(async user => {
          const external = await getExternalData();
          return {
            ...user,
            nameAndAgeWithExternalData: `${user.name} (${user.age}y)${external}`
          };
        });

        return Promise.all(computedDataPromises);
      }
    },
  },
});

async function main() {
  // 使用自定义的 model 扩展方法
  const usersWithData = await prisma.user.findManyWithComputedData({});
  if (usersWithData.length > 0) {
    console.log(usersWithData[0].nameAndAgeWithExternalData); // 例如: Sonia Lomo (25y) [外部数据]
  }

  const usersWithDataOptimized = await prisma.user.findManyWithComputedDataOptimized({});
  if (usersWithDataOptimized.length > 0) {
    console.log(usersWithDataOptimized[0].nameAndAgeWithExternalData); // 例如: Sonia Lomo (25y) [外部数据]
  }
}

main();