MongoDB中匹配文档并移动嵌套数组元素的实现方案咨询

需求说明 #

需要更新MongoDB文档中匹配指定字段的嵌套元素：将childrens数组里name为"Robert"的元素，更新其degree为"Master"，同时将该元素移动到students数组，并从原childrens数组中移除。

原始数据集 #

[
  {
    "name": "Guillaume",
    "childrens": [
      {
        "name": "Robert",
        "degree": "License"
      }
    ],
    "students": [
      {
        "name": "Hélène",
        "degree": "License"
      }
    ]
  },
  {
    "name": "Mathilde",
    "childrens": [
      {
        "name": "Lucie",
        "degree": "License"
      }
    ],
    "students": [
      {
        "name": "Michel",
        "degree": "License"
      }
    ]
  }
]

预期结果 #

[
  {
    "name": "Guillaume",
    "childrens": [],
    "students": [
      {
        "name": "Hélène",
        "degree": "License"
      },
      {
        "name": "Robert",
        "degree": "Master"
      }
    ]
  },
  {
    "name": "Mathilde",
    "childrens": [
      {
        "name": "Lucie",
        "degree": "License"
      }
    ],
    "students": [
      {
        "name": "Michel",
        "degree": "License"
      }
    ]
  }
]

问题分析 #

你之前的尝试存在两个核心错误：

1. 语法错误：将$push嵌套在$set内部，MongoDB要求$set、$push等更新操作符必须是同级顶层字段。
2. 逻辑限制：数组过滤器仅用于定位更新路径，无法直接提取元素值作为$push的内容。

要实现这种复杂的数组操作，需要使用MongoDB 4.2+支持的聚合管道式更新，它允许在更新时访问文档的现有字段值，完成多步骤的数组修改。

正确实现方案 #

方案一：分步式更新 #

db.person.updateMany(
  // 匹配包含目标元素的文档
  {"childrens.name": "Robert"},
  [
    // 第一步：更新childrens中Robert的degree为Master
    {
      $set: {
        childrens: {
          $map: {
            input: "$childrens",
            as: "child",
            in: {
              $cond: [
                {$eq: ["$$child.name", "Robert"]},
                {$mergeObjects: ["$$child", {degree: "Master"}]},
                "$$child"
              ]
            }
          }
        }
      }
    },
    // 第二步：将更新后的Robert元素追加到students数组
    {
      $set: {
        students: {
          $concatArrays: [
            "$students",
            [
              {$arrayElemAt: [
                {$filter: {input: "$childrens", cond: {$eq: ["$$this.name", "Robert"]}}},
                0
              ]}
            ]
          ]
        }
      }
    },
    // 第三步：从childrens数组中移除Robert元素
    {
      $set: {
        childrens: {
          $filter: {input: "$childrens", cond: {$ne: ["$$this.name", "Robert"]}}
        }
      }
    }
  ]
)

方案二：简洁式更新（使用临时字段） #

db.person.updateMany(
  {"childrens.name": "Robert"},
  [
    // 提取并更新目标元素到临时字段
    {
      $set: {
        targetChild: {
          $mergeObjects: [
            {$arrayElemAt: [
              {$filter: {input: "$childrens", cond: {$eq: ["$$this.name", "Robert"]}}},
              0
            ]},
            {degree: "Master"}
          ]
        }
      }
    },
    // 更新students数组并移除childrens中的目标元素
    {
      $set: {
        students: {$concatArrays: ["$students", ["$targetChild"]]},
        childrens: {$filter: {input: "$childrens", cond: {$ne: ["$$this.name", "Robert"]}}}
      }
    },
    // 删除临时字段（可选）
    {$unset: "targetChild"}
  ]
)

代码说明 #

• $map：遍历数组并修改指定元素，用于更新Robert的degree字段。
• $filter：筛选数组元素，用于提取目标元素或移除目标元素。
• $concatArrays：合并两个数组，将更新后的目标元素追加到students数组。
• $mergeObjects：合并对象，保留原元素的所有属性，仅更新degree字段。

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