使用Prisma Client Extensions集成外部数据与异步计算字段

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

const prisma = new PrismaClient().$extends({
  result: {
    user: {
      fullName: {
        needs: { firstName: true, lastName: true },
        compute(user) {
          return `${user.firstName} ${user.lastName}`;
        },
      },
    },
  },
});

// 假设我们有一个User模型，包含firstName和lastName字段
// 调用 prisma.user.findMany() 后，每个user对象都会自动包含一个fullName字段
async function main() {
  const users = await prisma.user.findMany({
    select: { firstName: true, lastName: true, fullName: true }, // 注意这里可以查询fullName
  });
  console.log(users);
}

main();

// 模拟一个异步获取外部数据的函数
async function getExternalUserData(userId: string): Promise<string> {
  return new Promise(resolve => {
    setTimeout(() => {
      // 模拟网络延迟和数据获取
      resolve(` [外部数据: ID-${userId} 认证通过]`);
    }, 500);
  });
}

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

// 假设schema.prisma中有一个User模型
// model User {
//   id        String @id @default(uuid())
//   email     String @unique
//   firstName String
//   lastName  String
// }

// 模拟一个异步获取外部数据的函数（同上）
async function getExternalUserData(userId: string): Promise<string> {
  return new Promise(resolve => {
    setTimeout(() => {
      resolve(` [外部数据: ID-${userId} 认证通过]`);
    }, 500);
  });
}

const prisma = new PrismaClient().$extends({
  result: {
    user: {
      // 可以在一个扩展中定义多个计算字段
      fullName: {
        needs: { firstName: true, lastName: true },
        compute(user) {
          return `${user.firstName} ${user.lastName}`;
        },
      },
      // 定义一个异步计算字段，用于获取外部数据
      externalInfo: {
        needs: { id: true }, // 假设外部数据需要用户的ID
        async compute(user) {
          // 在这里可以执行任何异步操作，例如调用外部API
          const externalData = await getExternalUserData(user.id);
          return `(${user.fullName}) ${externalData}`;
        },
      },
    },
  },
});

async function main() {
  // 创建一些测试用户（如果数据库中没有）
  await prisma.user.upsert({
    where: { email: 'alice@example.com' },
    update: {},
    create: {
      email: 'alice@example.com',
      firstName: 'Alice',
      lastName: 'Smith',
    },
  });
  await prisma.user.upsert({
    where: { email: 'bob@example.com' },
    update: {},
    create: {
      email: 'bob@example.com',
      firstName: 'Bob',
      lastName: 'Johnson',
    },
  });

  console.log('--- 查询所有用户并获取外部信息 ---');
  const users = await prisma.user.findMany({
    select: {
      id: true,
      email: true,
      fullName: true, // 同步计算字段
      externalInfo: true, // 异步计算字段
    },
  });

  for (const user of users) {
    console.log(`- 用户ID: ${user.id}, 邮箱: ${user.email}, 全名: ${user.fullName}, 外部信息: ${user.externalInfo}`);
  }
}

main()
  .then(async () => {
    await prisma.$disconnect();
  })
  .catch(async (e) => {
    console.error(e);
    await prisma.$disconnect();
    process.exit(1);
  });