MongoDB数组更新：仅更新存在元素，新增元素自动追加

场景说明 #

现有文档：

{
  _id: 11,
  marksObtained: [{physics:10}, {chemistry:12}, {maths: 11}, {biology:9}, {social:9}],
  name:"Someone"
}

待更新数据：[{chemistry:13},{zoology:12}]，要求：

• 更新数组中已存在的chemistry值为13
• 新增zoology:12到数组末尾
• 避免客户端循环对比逻辑，支持动态键

解决方案（MongoDB 4.2+） #

利用MongoDB聚合管道式更新，结合$map、$concatArrays、$cond等操作符，在数据库端完成逻辑处理：

// 定义动态更新数据
const updateItems = { chemistry:13, zoology:12 };

db.collection.updateOne(
  { _id: 11 }, // 查询条件
  [
    {
      $set: {
        marksObtained: {
          $concatArrays: [
            // 第一部分：更新原数组中已存在的项
            {
              $map: {
                input: "$marksObtained",
                as: "oldItem",
                in: {
                  $cond: [
                    // 判断当前元素的键是否在待更新列表中
                    { $in: [ { $arrayElemAt: [ { $objectToArray: "$$oldItem" }, 0 ] }.k, Object.keys(updateItems) ] },
                    // 匹配成功：替换为新值
                    { $arrayToObject: [ [ { 
                      k: { $arrayElemAt: [ { $objectToArray: "$$oldItem" }, 0 ] }.k,
                      v: updateItems[ { $arrayElemAt: [ { $objectToArray: "$$oldItem" }, 0 ] }.k ]
                    } ] ] },
                    // 匹配失败：保留原元素
                    "$$oldItem"
                  ]
                }
              }
            },
            // 第二部分：过滤出原数组中没有的新项，追加到末尾
            {
              $filter: {
                input: Object.entries(updateItems).map(([k, v]) => ({ [k]: v })),
                as: "newItem",
                cond: {
                  $not: [
                    { $in: [ { $arrayElemAt: [ { $objectToArray: "$$newItem" }, 0 ] }.k, 
                            { $map: { input: "$marksObtained", as: "item", in: { $arrayElemAt: [ { $objectToArray: "$$item" }, 0 ] }.k } } ] }
                  ]
                }
              }
            }
          ]
        }
      }
    }
  ]
)

逻辑解析 #

1. 更新已有项：通过$map遍历原数组，将每个元素转为键值对数组（$objectToArray），提取键名后判断是否在待更新键列表中，是则替换值，否则保留原元素。
2. 追加新项：将待更新数据转为数组格式，用$filter过滤掉原数组已存在的键，剩下的新元素通过$concatArrays拼接到更新后的原数组末尾。
3. 动态适配：只需修改updateItems对象即可处理任意动态键，无需修改核心更新逻辑。

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