如何在Mongoose的updateMany中为嵌套数组对象生成唯一ObjectId？

现有MongoDB文档结构：

{
   _id: something,
   property: value,
   animals: [
     {
       dogs: [
         { name: 'Dog1' },
         { name: 'Dog2' }
       ],
       cats: [
         { name: 'Cat1' },
         { name: 'Cat2' }
       ]
     }
   ]
}

需要给dogs和cats数组中的每个对象添加唯一_id，期望结果：

{
   _id: something,
   property: value,
   animals: [
     {
       dogs: [
         { name: 'Dog1', _id: uniqueID },
         { name: 'Dog2', _id: uniqueID }
       ],
       cats: [
         { name: 'Cat1', _id: uniqueID },
         { name: 'Cat2', _id: uniqueID }
       ]
     }
   ]
}

尝试了以下代码，但所有生成的ID完全相同：

db.collection('collectionName').updateMany(
  { animals: { $ne: [] } },
  {
    $set: {
      'animals.$[].cats.$[]._id': mongoose.Types.ObjectId(),
      'animals.$[].dogs.$[]._id': mongoose.Types.ObjectId()
    }
  }
);

问题根源 #

你的代码中，mongoose.Types.ObjectId()在构建更新语句时仅执行一次，生成单个ID值，所以所有数组元素都会被设置为这个相同的ID。

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

通过聚合管道的$map遍历嵌套数组，结合$function在服务器端为每个元素生成独立的ObjectId：

db.collection('collectionName').updateMany(
  { animals: { $ne: [] } },
  [
    {
      $set: {
        animals: {
          $map: {
            input: "$animals",
            as: "animal",
            in: {
              dogs: {
                $map: {
                  input: "$$animal.dogs",
                  as: "dog",
                  in: {
                    name: "$$dog.name",
                    _id: { 
                      $function: {
                        body: "() => ObjectId()",
                        args: [],
                        lang: "js"
                      }
                    }
                  }
                }
              },
              cats: {
                $map: {
                  input: "$$animal.cats",
                  as: "cat",
                  in: {
                    name: "$$cat.name",
                    _id: { 
                      $function: {
                        body: "() => ObjectId()",
                        args: [],
                        lang: "js"
                      }
                    }
                  }
                }
              }
            }
          }
        }
      }
    }
  ]
);

$function会在处理每个数组元素时调用一次，确保每个_id都是唯一的。

方案2：客户端查询后逐个更新 #

如果MongoDB版本不支持聚合管道更新，可以先拉取文档，在客户端生成唯一ID后再保存：

// 批量查询目标文档
const docs = await db.collection('collectionName').find({ animals: { $ne: [] } }).toArray();

// 遍历每个文档处理嵌套数组
for (const doc of docs) {
  doc.animals.forEach(animal => {
    animal.dogs.forEach(dog => dog._id = new mongoose.Types.ObjectId());
    animal.cats.forEach(cat => cat._id = new mongoose.Types.ObjectId());
  });

  // 更新单个文档
  await db.collection('collectionName').updateOne(
    { _id: doc._id },
    { $set: { animals: doc.animals } }
  );
}

注意：如果文档数量庞大，建议分批处理；同时要考虑并发更新可能带来的冲突。

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