MongoDB $lookup聚合后关联数组被ObjectId重排，如何保留原顺序？

MongoDB 的 $lookup 阶段本身不保证返回的关联结果数组与原文档中 localField 数组的元素顺序一致。它会扫描被关联集合，把所有匹配到的文档按集合的自然扫描顺序输出，由于 ObjectId 自带按生成时间升序排列的特性，你看到的结果就是匹配到的文档按 ObjectId 大小排序的结果，和你原文档里存的引用顺序没有关联。

通过提前记录原数组的引用顺序、关联完成后按原顺序重排的逻辑即可解决，该写法兼容所有支持聚合功能的 MongoDB 版本，修改后的可直接运行的查询如下：

db.getCollection("users").aggregate([
  // 匹配目标用户文档
  {
    "$match": {
      "_id": "5a708a38e6a4078bd49f01d5"
    }
  },
  // 临时留存原locations数组的引用，避免被lookup结果覆盖
  {
    "$addFields": {
      "_tempLocationRefs": "$locations"
    }
  },
  // 执行关联查询，结果存在独立临时字段中
  {
    "$lookup": {
      "from": "user-locations",
      "localField": "_tempLocationRefs",
      "as": "_tempLocationDocs",
      "foreignField": "_id"
    }
  },
  // 按原引用数组的顺序，逐个匹配对应关联文档重排最终结果
  {
    "$addFields": {
      "locations": {
        "$map": {
          "input": "$_tempLocationRefs",
          "as": "refId",
          "in": {
            "$arrayElemAt": [
              "$_tempLocationDocs",
              {
                "$indexOfArray": [
                  "$_tempLocationDocs._id",
                  "$$refId"
                ]
              }
            ]
          }
        }
      }
    }
  },
  // 清理所有流程中产生的临时字段
  {
    "$project": {
      "_tempLocationRefs": 0,
      "_tempLocationDocs": 0
    }
  }
])

核心逻辑说明： 先把原文档里存的ObjectId引用数组完整保存，等$lookup拿到所有匹配的位置文档后，遍历原引用数组的每一个元素，从匹配结果中找到对应文档，再按原引用的顺序拼接成新的locations数组，完全还原你需要的原始存储顺序，不会被ObjectId的默认自然排序打乱。

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