MongoDB：数组元素的Upsert操作实现需求

针对MongoDB文档中数组元素的Upsert需求，以下是能覆盖三种场景的实现方法：

需求说明 #

• 场景1：数组中存在指定_id的元素时，仅更新该元素的a值，不改动其他字段（如b）；
• 场景2：数组存在但无指定_id的元素时，向数组追加包含指定_id和a值的新条目；
• 场景3：数组不存在时，创建数组并插入指定条目。

核心实现命令 #

替换代码中的变量为实际业务值即可使用：

// 替换为实际目标文档ID
const docId = 3;
// 替换为要操作的数组元素ID
const targetItemId = 1;
// 替换为要设置的a字段值
const newAValue = 90;

// 执行更新操作
db.yourCollection.updateOne(
  { "_id": docId },
  [
    {
      $set: {
        array: {
          $cond: [
            // 判断array字段是否为数组（是否存在）
            { $isArray: "$array" },
            // 数组存在时的逻辑
            {
              $let: {
                vars: {
                  // 过滤出数组中ID匹配的元素
                  matchedItems: { $filter: { input: "$array", cond: { $eq: ["$$this._id", targetItemId] } } }
                },
                in: {
                  $cond: [
                    // 判断是否匹配到目标元素
                    { $ne: [{ $size: "$$matchedItems" }, 0] },
                    // 匹配到：遍历数组，仅更新目标元素的a值
                    {
                      $map: {
                        input: "$array",
                        as: "item",
                        in: {
                          $mergeObjects: [
                            "$$item",
                            { $cond: [{ $eq: ["$$item._id", targetItemId] }, { a: newAValue }, {}] }
                          ]
                        }
                      }
                    },
                    // 未匹配到：追加新元素到数组
                    { $concatArrays: ["$array", [{ _id: targetItemId, a: newAValue }]] }
                  ]
                }
              }
            },
            // 数组不存在：创建数组并插入目标元素
            [{ _id: targetItemId, a: newAValue }]
          ]
        }
      }
    }
  ]
)

各场景验证结果 #

场景1：更新已存在的数组元素

设置targetItemId=1、newAValue=90执行命令后，原数组中_id=1元素的a值被更新为90，b值保留不变：

{
  "_id":3,
  "name":"alex",
  "array": [
    {
      "_id":1,
      "a":90,
      "b":60
    },
    {
      "_id":2,
      "a":40,
      "b":80
    }
  ]
}

场景2：添加新数组元素

设置targetItemId=4、newAValue=50执行命令后，新元素被追加到数组末尾：

{
  "_id":3,
  "name":"alex",
  "array": [
    {
      "_id":1,
      "a":30,
      "b":60
    },
    {
      "_id":2,
      "a":40,
      "b":80
    },
    {
      "_id":4,
      "a":50
    }
  ]
}

场景3：创建数组并插入元素

若原文档无array字段，设置targetItemId=1、newAValue=30执行命令后，会自动创建array字段并插入指定元素：

{
  "_id":3,
  "name":"alex",
  "array": [
    {
      "_id":1,
      "a":30
    }
  ]
}

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