MongoDB聚合管道中更新数组指定索引元素的正确方法咨询

问题原因 #

你最初尝试的$set: { 'values.0': 12 }在聚合管道中失效，是因为聚合管道对数组字段路径的处理逻辑和普通更新操作不同。这里的'values.0'会被解析为给数组每个元素添加一个键为"0"的字段，而不是定位到数组的第一个元素进行替换，所以才会出现所有数组元素变成{'0':12}的奇怪结果。

正确实现方法 #

方法1：使用$map结合索引判断（推荐，无元素数量限制） #

通过$zip将数组元素与对应的索引配对，再用$map遍历，判断索引是否为目标索引（这里是0），是则替换为新值，否则保留原元素。这种方法不受数组长度限制，且能正确处理含重复元素的数组：

db.aggregate([
  { $documents: [{ values: [-354, 52, 234, 1] }] },
  {
    $set: {
      values: {
        $map: {
          input: { 
            $zip: { 
              inputs: ["$values", { $range: [0, { $size: "$values" }] }] 
            }
          },
          as: "pair",
          in: {
            $cond: {
              if: { $eq: [{ $arrayElemAt: ["$$pair", 1] }, 0] },
              then: 12,
              else: { $arrayElemAt: ["$$pair", 0] }
            }
          }
        }
      }
    }
  }
]);

如果能确保数组中没有重复元素，也可以简化用$indexOfArray获取元素索引：

db.aggregate([
  { $documents: [{ values: [-354, 52, 234, 1] }] },
  {
    $set: {
      values: {
        $map: {
          input: "$values",
          as: "item",
          in: {
            $cond: {
              if: { $eq: [{ $indexOfArray: ["$values", "$$item"] }, 0] },
              then: 12,
              else: "$$item"
            }
          }
        }
      }
    }
  }
]);

方法2：改进$concatArrays + $slice方案（更简洁） #

你原来的方法可以优化，$slice如果省略第三个参数，会自动返回从起始索引到数组末尾的所有元素，无需手动写最大数值：

db.aggregate([
  { $documents: [{ values: [-354, 52, 234, 1] }] },
  {
    $set: {
      values: {
        $concatArrays: [[12], { $slice: ["$values", 1] }]
      }
    }
  }
]);

这个方案更简洁，且不存在元素数量限制问题。

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