如何在MongoDB中更新文档未知嵌套位置的指定子文档

原始集合数据 #

[
  {
    "_id": "3b4f1d16-2706-4a92-bf0c-f4949b43953b",
    "someProperty": {
      "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
      "label": "Apple"
    },
    "someArray": [
      {
        "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
        "label": "Apple"
      }
    ]
  },
  {
    "_id": "234234324........",
    "propLevel1": {
      "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
      "label": "Apple"
    },
    "propLevel1_2": {
      "propOnLevel2": {
        "deeper": {
          "andDepper": {
            "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
            "label": "Apple"
          }
        }
      },
      "array1OnLevel2": [
        {
          "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
          "label": "Apple"
        }
      ]
    }
  }
]

更新要求 #

将所有_id为2253baac-68b5-4bce-9f78-0829de3b597c的子文档（无论嵌套层级、是否在数组中）的label字段从Apple改为Car，最终结果如下：

[
  {
    "_id": "3b4f1d16-2706-4a92-bf0c-f4949b43953b",
    "someProperty": {
      "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
      "label": "Car"
    },
    "someArray": [
      {
        "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
        "label": "Car"
      }
    ]
  },
  {
    "_id": "234234324........",
    "propLevel1": {
      "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
      "label": "Car"
    },
    "propLevel1_2": {
      "propOnLevel2": {
        "deeper": {
          "andDepper": {
            "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
            "label": "Car"
          }
        }
      },
      "array1OnLevel2": [
        {
          "_id": "2253baac-68b5-4bce-9f78-0829de3b597c",
          "label": "Car"
        }
      ]
    }
  }
]

解决方案 #

方法1：MongoDB 4.2+聚合管道更新（推荐） #

利用$function递归遍历文档所有层级，处理嵌套对象和数组中的目标子文档：

db.collection.updateMany(
  {}, // 匹配集合中所有文档
  [
    {
      $set: {
        root: {
          $function: {
            body: function(obj, targetId) {
              // 处理数组类型
              if (Array.isArray(obj)) {
                return obj.map(item => {
                  if (typeof item === 'object' && item !== null) {
                    return this.body(item, targetId);
                  }
                  return item;
                });
              }
              // 处理对象类型
              if (typeof obj === 'object' && obj !== null) {
                const updated = {...obj};
                // 匹配目标子文档，更新label字段
                if (updated._id === targetId) {
                  updated.label = 'Car';
                }
                // 递归处理对象的每个属性
                for (const key in updated) {
                  updated[key] = this.body(updated[key], targetId);
                }
                return updated;
              }
              return obj;
            },
            args: ["$$ROOT", "2253baac-68b5-4bce-9f78-0829de3b597c"],
            lang: "js"
          }
        }
      }
    },
    { $replaceRoot: { newRoot: "$root" } },
    { $unset: "root" } // 清理临时生成的root字段
  ]
)

方法2：脚本遍历更新（适配低版本MongoDB） #

如果MongoDB版本低于4.2，无法使用$function，可以用Node.js脚本递归修改文档后批量更新：

const { MongoClient } = require('mongodb');

async function batchUpdateDocuments() {
  // 替换为你的MongoDB连接字符串
  const uri = 'mongodb://localhost:27017';
  const client = new MongoClient(uri);
  // 替换为你的数据库名和集合名
  const dbName = 'your-database-name';
  const collectionName = 'your-collection-name';
  const targetId = '2253baac-68b5-4bce-9f78-0829de3b597c';

  try {
    await client.connect();
    const collection = client.db(dbName).collection(collectionName);
    const cursor = collection.find({});

    while (await cursor.hasNext()) {
      const doc = await cursor.next();
      const updatedDoc = deepUpdate(doc, targetId);
      
      await collection.updateOne(
        { _id: doc._id },
        { $set: updatedDoc }
      );
    }
    console.log('所有文档更新完成');
  } finally {
    await client.close();
  }
}

// 递归更新函数
function deepUpdate(obj, targetId) {
  if (Array.isArray(obj)) {
    return obj.map(item => deepUpdate(item, targetId));
  }
  if (typeof obj === 'object' && obj !== null) {
    const updated = { ...obj };
    if (updated._id === targetId) {
      updated.label = 'Car';
    }
    for (const key in updated) {
      updated[key] = deepUpdate(updated[key], targetId);
    }
    return updated;
  }
  return obj;
}

batchUpdateDocuments().catch(console.error);

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