如何利用MongoDB聚合为查询结果添加同参数其他记录存在标识？

Hey there! 刚好我之前处理过类似的需求，用MongoDB的聚合管道就能轻松实现。咱们直接上解决方案，再一步步拆解逻辑~

简洁实现方案 #

这种方法先筛选出目标记录，再关联统计同model的总记录数，代码直观易懂：

db.Car.aggregate([
  // 1. 先筛选出year为"2020"的记录
  { $match: { year: "2020" } },
  // 2. 关联整个集合中所有同model的记录
  {
    $lookup: {
      from: "Car",
      localField: "model",
      foreignField: "model",
      as: "modelRecords"
    }
  },
  // 3. 添加hasAnotherRecord字段：判断同model的记录总数是否大于1
  {
    $addFields: {
      hasAnotherRecord: { $gt: [{ $size: "$modelRecords" }, 1] }
    }
  },
  // 4. 整理输出格式，去掉不需要的临时字段
  {
    $project: {
      _id: 0,
      year: 1,
      model: 1,
      hasAnotherRecord: 1
    }
  }
])

各阶段逻辑解释 #

• $match：精准筛选出year为"2020"的记录，也就是你要处理的目标数据集。
• $lookup：把当前记录的model作为关联条件，从整个Car集合中拉取所有同model的记录，存到modelRecords数组里。
• $addFields：用$size获取modelRecords的长度（也就是同model的总记录数），如果长度大于1，说明存在其他同model的记录，hasAnotherRecord设为true，否则为false。
• $project：调整输出字段，只保留你需要的year、model和hasAnotherRecord，同时去掉默认的_id字段。

高效优化方案（适合大数据集） #

如果你的集合数据量很大，先统计所有model的总次数再关联会更高效，避免重复查询：

db.Car.aggregate([
  // 1. 先统计整个集合中每个model的总记录数
  {
    $group: {
      _id: "$model",
      totalCount: { $sum: 1 }
    }
  },
  // 2. 关联year为"2020"的同model记录
  {
    $lookup: {
      from: "Car",
      let: { currentModel: "$_id" },
      pipeline: [
        { $match: { $expr: { $and: [{ $eq: ["$year", "2020"] }, { $eq: ["$model", "$$currentModel"] }] } } },
        { $project: { year: 1, model: 1, _id: 0 } }
      ],
      as: "filteredRecords"
    }
  },
  // 3. 展开关联后的记录数组
  { $unwind: "$filteredRecords" },
  // 4. 添加hasAnotherRecord字段
  {
    $addFields: {
      hasAnotherRecord: { $gt: ["$totalCount", 1] }
    }
  },
  // 5. 整理输出格式
  {
    $project: {
      _id: 0,
      year: "$filteredRecords.year",
      model: "$filteredRecords.model",
      hasAnotherRecord: 1
    }
  }
])

最终结果 #

运行任意一段代码，都会得到你预期的输出：

[
  { "year": "2020", "model": "Opel", "hasAnotherRecord": false },
  { "year": "2020", "model": "Renault", "hasAnotherRecord": true }
]

内容的提问来源于stack exchange，提问作者IntoTheDeep