MongoDB原子操作：按数组项Key增改文档数组内字典键值对

原始MongoDB文档 #

{
  "_id": ObjectId("57315ba4846dd82425ca2408"),
  "myarray": [
    {
      "key": 1,
      "mydictionary": {
        "PointA": {
          userId: ObjectId("570ca5e48dbe673802c2d035"),
          point: 5
        },
        "PointB": {
          userId: ObjectId("613ca5e48dbe673802c2d522"),
          point: 2
        }
      }
    },
    {
      "key": 2,
      "mydictionary": {
        "PointA": {
          userId: ObjectId("570ca5e48dbe673802c2d035"),
          point: 5
        },
        "PointB": {
          userId: ObjectId("613ca5e48dbe673802c2d522"),
          point: 2
        }
      }
    }
  ]
}

需求说明 #

• 通过原子操作完成更新
• 针对myarray数组中key匹配的特定项
• 对该项的mydictionary字段执行：
  • 更新已有键（如PointA的point值）
  • 新增新键值对（如添加PointC）

现有尝试的问题 #

你之前的代码尝试将字典转为数组处理，但存在两个核心问题：

1. 引用了不存在的mysubarray字段，与原始文档结构不匹配
2. 逻辑未针对myarray的key筛选目标项，无法定位到需要更新的数组元素

正确实现方案 #

方案1：匹配单个数组项（简洁版） #

当只需要更新第一个匹配key的数组项时，可利用$位置操作符定位目标项，结合$mergeObjects实现字典的合并更新（自动覆盖旧键、新增新键）：

db.collection.updateOne(
  {
    _id: ObjectId("57315ba4846dd82425ca2408"),
    "myarray.key": 1 // 匹配key为1的数组项
  },
  {
    $set: {
      "myarray.$.mydictionary": {
        $mergeObjects: [
          "$myarray.$.mydictionary", // 原字典内容
          {
            // 需要更新/新增的键值对
            "PointA": { userId: ObjectId("570ca5e48dbe673802c2d035"), point: 6 },
            "PointC": { userId: ObjectId("713ca5e48dbe673802c2d666"), point: 8 }
          }
        ]
      }
    }
  }
)

方案2：匹配多个数组项（通用版） #

如果需要更新所有匹配key的数组项（比如同时更新key=1和key=2的项），使用$map遍历数组并逐个判断处理：

db.collection.updateOne(
  {
    _id: ObjectId("57315ba4846dd82425ca2408")
  },
  [
    {
      $set: {
        "myarray": {
          $map: {
            input: "$myarray",
            as: "item",
            in: {
              $cond: [
                { $in: ["$$item.key", [1, 2]] }, // 匹配key为1或2的项
                {
                  $mergeObjects: [
                    "$$item", // 保留原数组项的其他字段
                    {
                      mydictionary: {
                        $mergeObjects: [
                          "$$item.mydictionary", // 原字典内容
                          {
                            "PointA": { userId: ObjectId("570ca5e48dbe673802c2d035"), point: 6 },
                            "PointC": { userId: ObjectId("713ca5e48dbe673802c2d666"), point: 8 }
                          }
                        ]
                      }
                    }
                  ]
                },
                "$$item" // 非目标项保持不变
              ]
            }
          }
        }
      }
    }
  ]
)

方案说明 #

• $mergeObjects是核心操作：它会将多个对象合并，已存在的键会被新值覆盖，不存在的键会被新增
• 两种方案均为原子操作，MongoDB会将整个更新作为单个事务执行，避免中间状态
• 可根据实际需求调整匹配条件（如$eq匹配单个key，$in匹配多个key）和需要更新/新增的键值对

内容的提问来源于stack exchange，提问作者Michael Trullas Garcia