MongoDB：如何对拼接数组排序后再执行索引提取？

假设你的文档包含两个待处理数组（如array1、array2）和指定保留索引的列表（如keepIndices），要解决提取元素被移至末尾的问题，核心是在聚合管道式更新中先完成数组拼接、排序，再按索引提取元素，具体实现如下：

示例文档结构 #

{
  "_id": ObjectId("60d21b4667d0d8992e610c85"),
  "array1": [5, 2, 8, 1],
  "array2": [3, 7, 4],
  "keepIndices": [0, 2, 4] // 需要保留的目标索引
}

完整更新查询 #

db.collection.updateMany(
  {}, // 替换为你的实际查询条件，空对象表示更新所有符合条件的文档
  [
    {
      $set: {
        // 拼接两个数组并按降序排序（升序将sortBy改为1即可）
        _tempSortedArray: {
          $sort: {
            input: { $concatArrays: ["$array1", "$array2"] },
            sortBy: -1
          }
        },
        // 遍历指定索引列表，提取排序后数组的对应元素
        sampledArray: {
          $map: {
            input: "$keepIndices",
            as: "idx",
            in: { $arrayElemAt: ["$_tempSortedArray", "$$idx"] }
          }
        }
      }
    },
    // 清理临时字段（可选，若不需要保留排序后的完整数组则添加）
    { $unset: "_tempSortedArray" }
  ]
)

关键说明 #

• 必须使用聚合管道作为更新逻辑（第二个参数为数组），普通更新操作符无法支持"排序→提取"的链式操作
• $sort专门用于数组排序：input传入拼接后的数组，sortBy指定排序方向（-1降序，1升序）
• $map+$arrayElemAt组合可精准按索引提取元素，避免元素位置偏移问题
• 若需直接替换原数组字段，将sampledArray改为目标字段名即可

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