MongoDB查询需求：基于指定字段添加排名新字段

嘿，刚接触MongoDB查询是吧？给数组里的节点按count字段添加排名这个需求，我给你两种可行的方案，看你的MongoDB版本选就行～

方法1：使用MongoDB 5.0+的$setWindowFields（推荐） #

如果你的MongoDB版本是5.0或以上，用窗口函数的方法最简洁直观，步骤清晰：

db.yourCollectionName.aggregate([
  // 先把nodes数组拆成单个文档，方便单独计算每个节点的排名
  { $unwind: "$nodes" },
  // 基于主文档的_id分区，确保只在当前文档的节点集合内排名
  {
    $setWindowFields: {
      partitionBy: "$_id",
      sortBy: { "nodes.count": -1 }, // 按count降序排序，数值越大排名越靠前
      output: {
        "nodes.rank": {
          $rank: {} // 相同count的节点会获得并列排名，比如两个count=10的节点排名都是2
          // 如果想要相同count的节点排名连续（比如2和3），可以换成$denseRank: {}
        }
      }
    }
  },
  // 把拆分后的节点重新合并回数组
  {
    $group: {
      _id: "$_id",
      nodes: { $push: "$nodes" }
    }
  }
])

步骤解释： #

• $unwind：把nodes数组拆分成独立的子文档，这样每个节点可以单独参与排名计算
• $setWindowFields：指定按主文档的_id分区（保证排名只在当前文档的节点范围内生效），按count降序排序后，用$rank给每个节点添加排名字段
• $group：将拆分后的节点重新聚合为数组，得到最终带排名的nodes集合

方法2：兼容旧版本的$map + $filter实现 #

如果你的MongoDB版本低于5.0，没法用窗口函数，可以用数组操作符组合来计算排名：

db.yourCollectionName.aggregate([
  {
    $addFields: {
      nodes: {
        $map: {
          input: "$nodes",
          as: "node",
          in: {
            // 保留原节点的所有字段，同时添加rank字段
            $mergeObjects: ["$$node", {
              rank: {
                // 排名逻辑：统计数组中count大于当前节点的数量，再加1就是当前排名
                $add: [
                  {
                    $size: {
                      $filter: {
                        input: "$nodes",
                        cond: { $gt: ["$$this.count", "$$node.count"] }
                      }
                    }
                  },
                  1
                ]
              }
            }]
          }
        }
      }
    }
  }
])

逻辑说明： #

• $addFields：更新主文档的nodes字段
• $map：遍历原nodes数组的每一个节点
• $mergeObjects：将原节点的所有字段和新计算的rank字段合并，保留原有数据
• 排名计算：用$filter筛选出数组中count大于当前节点的元素，用$size统计数量，加1后就是当前节点的排名（比如有3个节点的count比它大，那它的排名就是第4位）

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