MongoDB中为数组元素添加与现有字段值相同的新字段

你遇到的问题是因为直接通过$set给数组字段的子属性赋值时，MongoDB会把整个数组的对应属性值作为整体赋值，而非遍历每个元素单独处理。下面提供两种可行的解决方案：

1. 聚合管道查询（临时生成结果） #

如果只需要在查询结果中添加字段，不修改原集合数据，可以用$map配合$mergeObjects遍历数组并为每个元素添加字段：

db.collection.aggregate([
  {
    $set: {
      AllApplicableSteps: {
        $map: {
          input: "$AllApplicableSteps",
          as: "step",
          in: {
            $mergeObjects: [
              "$$step",
              { StepTitle: "$$step.StepName" }
            ]
          }
        }
      }
    }
  }
])

原理说明 #

• $map遍历AllApplicableSteps数组的每个元素，用$$step指代当前元素
• $mergeObjects将原元素的所有字段和新添加的StepTitle字段合并，无需手动列举所有已有字段，更灵活

2. 批量更新原集合数据 #

如果需要永久修改集合中的文档（MongoDB 4.2及以上版本支持），可以使用更新管道配合updateMany：

db.collection.updateMany(
  {}, // 空查询条件匹配所有文档，如需指定范围可添加过滤规则
  [
    {
      $set: {
        AllApplicableSteps: {
          $map: {
            input: "$AllApplicableSteps",
            as: "step",
            in: {
              $mergeObjects: [
                "$$step",
                { StepTitle: "$$step.StepName" }
              ]
            }
          }
        }
      }
    }
  ]
)

注意事项 #

• 如果你的MongoDB版本低于4.2，无法使用更新管道，需要先查询文档，在客户端处理数组后再更新回去（不推荐大集合使用）
• 执行updateMany前建议先通过find验证查询条件，避免误修改数据

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