MongoDB：在 $lookup 管道中添加 localField

本文介绍了如何在 MongoDB 的 $lookup 聚合管道中，将 localField 的相关信息添加到 lookup 的结果中。通常，$lookup 用于连接不同集合的数据，但有时需要将连接字段的附加信息也包含在结果中。通过结合 $map 和 $mergeObjects 操作符，我们可以巧妙地实现这一目标，从而在查询结果中保留所需的 localField 信息。

使用 $lookup 和 $map 添加 localField

在 MongoDB 中，$lookup 操作符用于执行集合间的连接操作。当我们需要在 lookup 的结果中包含 localField 的信息时，可以在 $lookup 的 pipeline 中使用 $map 和 $mergeObjects 操作符。

假设我们有两个集合：Users 和 Orders。Users 集合包含用户信息，Orders 集合包含订单信息，并通过 user_id 字段关联。我们希望在 lookup Orders 集合时，同时获取 Users 集合中的 added_at 字段。

以下是一个示例聚合管道：

db.Users.aggregate([
  {
    $lookup: {
      from: "Orders",
      localField: "inbox.participant",
      foreignField: "_id",
      as: "participants",
      pipeline: [
        { $project: { order_id: 1, order_date: 1, _id: 1 } }
      ]
    }
  },
  {
    $set: {
      participants: {
        $map: {
          input: "$participants",
          in: {
            $mergeObjects: [
              "$$this",
              { added_at: { $arrayElemAt: [
                    "$inbox.added_at",
                    { $indexOfArray: ["$inbox.participant", "$$this._id"] }
              ] } }
            ]
          }
        }
      }
    }
  }
])

代码解释：

Otter.ai

一个自动的会议记录和笔记工具，会议内容生成和实时转录

下载

1. $lookup 阶段： 连接 Users 和 Orders 集合，将匹配的订单信息存储在 participants 字段中。pipeline 用于指定 lookup 结果中需要包含的字段。
2. $set 阶段： 使用 $map 操作符遍历 participants 数组，对于每个订单，使用 $mergeObjects 将 added_at 字段添加到订单对象中。$arrayElemAt 和 $indexOfArray 用于查找与当前订单对应的 added_at 值。

注意事项：

• $indexOfArray 用于查找 localField 数组中与 foreignField 匹配的索引。
• $arrayElemAt 用于根据索引从 added_at 数组中获取对应的值。
• 确保 localField 和 foreignField 的数据类型匹配。

总结：

通过结合 $lookup、$map 和 $mergeObjects 操作符，我们可以灵活地将 localField 的信息添加到 lookup 的结果中，从而满足更复杂的数据查询需求。这种方法可以有效地避免多次查询数据库，提高查询效率。