MongoDB聚合查询：如何为$push的分组文档添加索引值

嘿，这个问题我之前处理过！首先得明确一个关键点：MongoDB的集合本身是无序的，所以不存在「原始文档在集合中的索引位置」这种天然属性——除非你显式给文档添加了顺序相关的字段，或者在聚合阶段生成一个逻辑序号。

针对你的需求，我们可以分两步实现：

1. 先用$setWindowFields（MongoDB 5.0及以上版本支持）给每个文档生成一个逻辑序号（用来模拟你要的“索引”）
2. 再执行你原本的$group分组操作

完整的聚合查询代码 #

var query = [
  // 第一步：给每个文档生成逻辑序号docIndex
  {
    $setWindowFields: {
      sortBy: { _id: 1 }, // 这里可以按你需要的规则排序，比如按info.date排序
      output: {
        docIndex: {
          $documentNumber: {} // 生成从1开始的递增序号
        }
      }
    }
  },
  // 第二步：分组时使用生成的docIndex作为x的值
  {
    $group: {
      _id: '$name',
      Marks: {
        $push: { 
          x: '$docIndex', 
          y: '$info.marks' 
        }
      }
    }
  }
]

重要细节说明 #

• sortBy参数一定要设置：因为集合无序，你必须指定排序规则来确定序号的生成逻辑，比如按_id、info.date或者其他业务相关字段排序，这样生成的序号才有实际意义。
• 如果你的MongoDB版本低于5.0，没法用窗口函数的话，可以考虑两种替代方案：
  • 插入文档时就维护一个自增的序号字段；
  • 在客户端先拉取所有文档，在代码层面给文档加序号后再分组，但这种方式数据量大时效率不高，不推荐。

执行结果示例 #

用你提供的测试数据跑这个查询，会得到类似这样的结果：

[
  {
    "_id": "John",
    "Marks": [
      { "x": 1, "y": "12" },
      { "x": 2, "y": "15" }
    ]
  },
  {
    "_id": "Ben",
    "Marks": [
      { "x": 3, "y": "18" }
    ]
  }
]

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