Node.js中为MongoDB对象数组批量添加新字段的方法

一、MongoDB 4.2+ 高效批量更新方案（推荐） #

利用MongoDB 4.2引入的聚合管道更新功能，可一次性完成所有文档的数组元素批量修改，无需循环遍历单个文档，效率最高。

执行以下命令：

db.users.updateMany(
  {}, // 空对象匹配所有用户文档，如需过滤可添加条件，比如 { "plates": { $exists: true } }
  [
    {
      $set: {
        plates: {
          $map: {
            input: "$plates",
            as: "plate",
            in: {
              $mergeObjects: [
                "$$plate",
                { plateStyle: "default" } // 替换为你需要的默认值
              ]
            }
          }
        }
      }
    }
  ]
)

关键逻辑说明：

• $map：遍历plates数组的每一个元素
• $mergeObjects：将原plate对象与新增的plateStyle字段合并，保留原对象所有字段的同时添加新字段
• updateMany：批量匹配并更新符合条件的文档，避免多次IO操作

二、低版本MongoDB兼容方案（不推荐） #

如果你的MongoDB版本低于4.2，无法使用聚合管道更新，只能通过遍历单个文档实现：

db.users.find({ "plates": { $exists: true } }).forEach(user => {
  // 遍历当前用户的plates数组，为每个元素添加字段
  const updatedPlates = user.plates.map(plate => ({
    ...plate,
    plateStyle: "default"
  }));
  // 更新当前用户文档
  db.users.updateOne(
    { _id: user._id },
    { $set: { plates: updatedPlates } }
  );
});

注意：此方案需要对每个用户文档发起一次更新请求，数据量大时性能较差，仅作为兼容低版本的备选方案。

前置验证建议 #

执行更新前，建议先通过聚合查询预览修改结果，确认逻辑正确：

db.users.aggregate([
  {
    $project: {
      plates: {
        $map: {
          input: "$plates",
          as: "plate",
          in: { $mergeObjects: ["$$plate", { plateStyle: "default" }] }
        }
      }
    }
  }
])

进阶：动态设置plateStyle值 #

如果需要根据plate的其他字段动态赋值，比如根据type字段设置不同的plateStyle，可以修改聚合管道：

db.users.updateMany(
  {},
  [
    {
      $set: {
        plates: {
          $map: {
            input: "$plates",
            as: "plate",
            in: {
              $mergeObjects: [
                "$$plate",
                { 
                  plateStyle: {
                    $switch: {
                      branches: [
                        { case: { $eq: ["$$plate.type", "car"] }, then: "automotive" },
                        { case: { $eq: ["$$plate.type", "bike"] }, then: "cycle" }
                      ],
                      default: "default"
                    }
                  }
                }
              ]
            }
          }
        }
      }
    }
  ]
)

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