Prisma关系查询：深度解析如何通过外键获取关联字段详情

本文旨在详细指导如何在Prisma中通过外键关系高效地查询并获取关联实体的详细字段，而非仅仅获取外键ID。我们将通过一个实际的用户与朋友关系模型，演示如何利用Prisma的嵌套select语句，从多对多关系中的连接表进一步深入，获取关联用户的完整信息，从而优化数据查询与展示。

1. 理解Prisma中的关系模型

在数据库设计中，多对多关系通常通过一个中间表（或称连接表）来实现。在Prisma中，这种关系被清晰地定义在schema.prisma文件中。以下是一个用户和朋友关系的示例：

model User {
  id         Int           @id @default(autoincrement())
  name       String        // 用户姓名
  self       Friend[]      @relation("self")   // 用户作为发起方的朋友关系
  friend     Friend[]      @relation("friend")  // 用户作为被邀请方的朋友关系
}

model Friend {
  id         Int        @id @default(autoincrement())
  user       User       @relation("self", fields: [userId], references: [id]) // 关联到发起友谊的用户
  userId     Int        // 发起友谊用户的外键
  friend     User       @relation("friend", fields: [friendId], references: [id]) // 关联到被友谊的用户
  friendId   Int        // 被友谊用户的外键
  // @@unique([userId, friendID]) // 可选：确保友谊关系的唯一性
}

登录后复制

在这个模型中：

• User 模型代表用户。
• Friend 模型是连接表，它记录了两个User之间的友谊关系。
• Friend 模型通过 user 和 friend 两个关系字段，分别指向两个 User 实例。userId 和 friendId 是相应的外键。
• User 模型上的 self 和 friend 字段则表示该用户参与的所有 Friend 关系实例。

2. 初始查询与面临的问题

当我们尝试查询一个用户及其朋友时，一个常见的需求是不仅获取朋友的ID，还要获取朋友的姓名等详细信息。然而，如果只是简单地选择连接表的外键，结果往往不尽如人意。

考虑以下查询代码：

// sample.tsx (初始尝试)
const friends = await prisma.user.findUnique({
  where: {
    id: userId, // 当前用户的 ID
  },
  select: {
    name: true, // 选择当前用户的姓名
    friend: {   // 选择当前用户作为“朋友”关系中的被邀请方的所有Friend实例
      select: {
        userId: true, // 从Friend实例中选择发起友谊的用户的ID
      },
    },
  },
});

console.log(friends);

登录后复制

这段代码的输出可能如下：

{  
  "name": "Paul McCartney",
  "friend": [ { "userId": 1 }, { "userId": 3 }, { "userId": 4 } ]
}

登录后复制

这里的问题是，friend 数组中只包含了 Friend 模型的 userId 字段，这实际上是发起友谊的用户的ID，而不是我们期望的“朋友”的详细信息（即 friend 字段所指向的 User 实例的详细信息）。我们希望得到的是类似 { userId: 1, name: "John Lennon" } 这样的结构。

3. 解决方案：利用嵌套 select 获取关联实体详情

Prisma 允许在关系字段内部进行更深层次的 select 操作，从而选择关联实体的具体字段。要解决上述问题，我们需要在 friend 关系（从 User 到 Friend 模型）的 select 中，进一步选择 Friend 模型中的 friend 关系（从 Friend 到 User 模型），并指定我们需要的 User 字段。

标签： javascript java go ai