MongoDB 中的 $lookup 和 $pipeline 实践-JavaScript中文网-JavaScript教程资源分享门户

前言

在 MongoDB 中，$lookup 和 $pipeline 是两个非常重要的操作符。$lookup 可以将多个集合进行关联查询，而 $pipeline 则可以对查询结果进行多次处理和过滤。本文将介绍如何使用 $lookup 和 $pipeline 进行 MongoDB 数据库的查询和处理，并提供相关的示例代码和指导意义。

$lookup 操作符

$lookup 操作符是 MongoDB 中用于关联查询的操作符，它可以将多个集合进行关联查询，实现类似 SQL 中的 JOIN 操作。$lookup 的语法如下：

{
  $lookup: {
    from: <collection to join>,
    localField: <field from the input documents>,
    foreignField: <field from the documents of the "from" collection>,
    as: <output array field>
  }
}

其中，from 表示要进行关联查询的集合名称，localField 表示当前集合中用于关联的字段，foreignField 表示要关联的集合中用于关联的字段，as 表示输出结果的字段名。

下面是一个 $lookup 的示例：

假设我们有两个集合，一个是学生集合，另一个是课程集合，它们的结构如下：

// 学生集合
{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4c"),
  "name": "张三",
  "age": 18,
  "courses": [
    ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
    ObjectId("5f9c0c7d4f4c4f4e4e4c4f4e")
  ]
}

// 课程集合
{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
  "name": "数学",
  "teacher": "李四"
},
{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4e"),
  "name": "英语",
  "teacher": "王五"
}

现在我们需要查询每个学生所选的课程信息，可以使用 $lookup 来实现：

db.students.aggregate([
  {
    $lookup: {
      from: "courses",
      localField: "courses",
      foreignField: "_id",
      as: "course_info"
    }
  }
])

执行上述查询后，会返回每个学生所选的课程信息，结果如下：

{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4c"),
  "name": "张三",
  "age": 18,
  "courses": [
    ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
    ObjectId("5f9c0c7d4f4c4f4e4e4c4f4e")
  ],
  "course_info": [
    {
      "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
      "name": "数学",
      "teacher": "李四"
    },
    {
      "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4e"),
      "name": "英语",
      "teacher": "王五"
    }
  ]
}

$pipeline 操作符

$pipeline 操作符是 MongoDB 中用于对查询结果进行多次处理和过滤的操作符，它可以实现类似 SQL 中的 GROUP BY 和 HAVING 操作。$pipeline 的语法如下：

{
  $pipeline: [
    {
      <stage1>
    },
    {
      <stage2>
    },
    ...
  ]
}

$pipeline 中包含了多个 stage，每个 stage 表示一个操作，可以是 $match、$group、$sort 等操作。下面是一个 $pipeline 的示例：

假设我们有一个订单集合，它的结构如下：

{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4f"),
  "user_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4c"),
  "order_time": ISODate("2020-10-30T10:00:00Z"),
  "amount": 100
},
{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f50"),
  "user_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4c"),
  "order_time": ISODate("2020-10-30T11:00:00Z"),
  "amount": 200
},
{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f51"),
  "user_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
  "order_time": ISODate("2020-10-30T12:00:00Z"),
  "amount": 300
},
{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f52"),
  "user_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
  "order_time": ISODate("2020-10-30T13:00:00Z"),
  "amount": 400
}

现在我们需要查询每个用户的总消费金额，可以使用 $pipeline 来实现：

db.orders.aggregate([
  {
    $group: {
      _id: "$user_id",
      total_amount: { $sum: "$amount" }
    }
  },
  {
    $lookup: {
      from: "users",
      localField: "_id",
      foreignField: "_id",
      as: "user_info"
    }
  },
  {
    $match: {
      total_amount: { $gt: 500 }
    }
  }
])

上述查询中，第一个 stage 使用 $group 操作对订单进行分组，计算每个用户的总消费金额。第二个 stage 使用 $lookup 操作将用户信息关联进来。第三个 stage 使用 $match 操作过滤出总消费金额大于 500 的用户。

执行上述查询后，会返回每个用户的总消费金额和用户信息，结果如下：

{
  "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
  "total_amount": 700,
  "user_info": [
    {
      "_id": ObjectId("5f9c0c7d4f4c4f4e4e4c4f4d"),
      "name": "李四",
      "age": 20
    }
  ]
}

总结

$lookup 和 $pipeline 是 MongoDB 中非常重要的操作符，它们可以实现复杂的查询和处理操作。在实际应用中，我们需要根据具体的需求来选择使用哪种操作符，并进行合理的优化，以提高查询效率和性能。同时，我们还需要注意避免过度使用操作符，以免导致查询复杂度过高，影响系统的性能和稳定性。

来源：JavaScript中文网，转载请注明来源本文地址：https://www.javascriptcn.com/post/658f33a5eb4cecbf2d4e4fd4