MongoDB：如何为每个嵌套字段添加含单个值的数组字段？

问题背景 #

现有MongoDB集合文档结构如下：

{
  _id: ObjectId("…"),
  items: [
   {value: "abc", other: "other1"},
   {value: "def", other: "other2"}
  ],
  other: "other"
}

需要为items数组中的每个子文档新增values字段（值为对应value的单元素数组），转换后结构如下：

{
  _id: ObjectId("…"),
  items: [
   {values: ["abc"], value: "abc", other: "other1"},
   {values: ["def"], value: "def", other: "other2"}
  ],
  other: "other"
}

原有的find().forEach()方法处理150万条文档时效率极低，核心原因是该方法会将所有文档加载到客户端内存，逐条修改后再保存，产生大量IO和网络开销。

高效解决方案 #

方案一：使用聚合管道批量更新（MongoDB 4.2+ 推荐）

利用MongoDB 4.2+支持的聚合管道更新，直接在服务器端完成所有文档的修改，无需将数据拉到客户端，是效率最高的方案。

代码实现（灵活保留所有子字段）：

db.collection.updateMany(
  {}, // 匹配集合中所有文档
  [
    {
      $set: {
        items: {
          $map: {
            input: "$items",
            as: "item",
            in: {
              // 合并原字段与新增的values字段
              $mergeObjects: [
                "$$item",
                { values: ["$$item.value"] }
              ]
            }
          }
        }
      }
    }
  ]
)

说明：

• $map遍历items数组的每个子文档
• $mergeObjects自动保留子文档的所有原有字段，同时新增values字段，无需手动列举每个字段
• 整个操作由MongoDB服务器端批量执行，网络和内存开销极小

方案二：批量写入操作（适配MongoDB 4.2以下版本）

如果你的MongoDB版本低于4.2，无法使用聚合管道更新，可以用批量写入减少网络往返次数，比逐条save效率提升明显。

代码实现：

const bulk = db.collection.initializeUnorderedBulkOp();
let batchCount = 0;

// 只查询items字段，减少数据传输量
db.collection.find({}, { items: 1 }).forEach(doc => {
  // 生成更新后的items数组
  const updatedItems = doc.items.map(item => ({
    ...item,
    values: [item.value]
  }));
  
  bulk.find({ _id: doc._id }).updateOne({
    $set: { items: updatedItems }
  });

  batchCount++;
  // 每1000条提交一次批量操作，避免内存溢出
  if (batchCount % 1000 === 0) {
    bulk.execute();
    bulk = db.collection.initializeUnorderedBulkOp();
  }
});

// 提交剩余未处理的文档
if (batchCount % 1000 !== 0) {
  bulk.execute();
}

说明：

• initializeUnorderedBulkOp()创建无序批量操作，服务器并行执行
• 每1000条提交一次，避免客户端内存占用过高
• 查询时只获取items字段，减少数据传输量

注意事项 #

1. 操作前务必备份数据：批量更新不可逆，建议先备份集合或在测试环境验证后再执行
2. 根据集合大小调整批次：方案二的批次大小（1000）可根据服务器性能和内存情况调整
3. 优先使用方案一：聚合管道更新的性能远高于批量写入，是大数量级文档更新的首选

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