MongoDB更新嵌套数组foreach

Question

我收集了Users，每个用户都有一个数组Ancestors，以前的开发人员做错了DB体系结构，现在每个祖先都是string，但是必须是ObjectId 。 它仍然包含objectId（实际上是对象ID的十六进制，例如558470744a73274db0f0d65d ）。 如何将每个祖先转换为ObjectId？ 我这样写：

db.getCollection('Users').find({}).forEach(function(item){
  if (item.Ancestors instanceof Array){
      var tmp = new Array()
      item.Ancestors.forEach(function(ancestor){
          if (ancestor instanceof String){
               tmp.push(ObjectId(ancestor))
             }
          })
          item.Ancestors = tmp
          db.getCollection('Users').save(item) 
  }
})

但是看起来它无法正常工作，现在有些祖先是ObjectId，有些是null 。 而且祖先从一开始就可以为空。 所以我把所有的if

Answer 1

像猫鼬一样尝试

var mongoose = require('mongoose');

db.getCollection('Users').find({}).forEach(function(item){
  if (item.Ancestors instanceof Array){
      var tmp = new Array()
      item.Ancestors.forEach(function(ancestor){
          if (ancestor instanceof String){
               tmp.push(mongoose.Types.ObjectId(ancestor))
             }
          })
          item.Ancestors = tmp
          db.getCollection('Users').save(item) 
  }
})

Answer 2

这里的解决方案概念是使用游标遍历您的集合，并为游标中的每个文档收集有关Ancestors数组元素的索引位置的数据。

然后，您稍后将在循环中将此数据用作更新操作参数，以正确标识要更新的元素。

假设您的集合不是那么笨拙，可以使用游标的forEach()方法实现上述直觉，就像您尝试进行迭代并获取所有涉及的数组元素的索引数据一样。

下面展示了针对小型数据集的这种方法：

function isValidHexStr(id) {
    var checkForHexRegExp = new RegExp("^[0-9a-fA-F]{24}$");
    if(id == null) return false;
    if(typeof id == "string") {
        return id.length == 12 || (id.length == 24 && checkForHexRegExp.test(id));
    }
    return false;
};


db.users.find({"Ancestors.0": { "$exists": true, "$type": 2 }}).forEach(function(doc){ 
    var ancestors = doc.Ancestors,
        updateOperatorDocument = {}; 
    for (var idx = 0; idx < ancestors.length; idx++){ 
        if(isValidHexStr(ancestors[idx]))                   
            updateOperatorDocument["Ancestors."+ idx] = ObjectId(ancestors[idx]);           
    };  
    db.users.updateOne(
        { "_id": doc._id },
        { "$set": updateOperatorDocument }
    );      
});

---

现在，为了提高性能，尤其是在处理大型集合时，请利用Bulk() API Bulk()更新集合。 与上述操作相反，这非常有效，因为使用bulp API时，您将分批将操作发送到服务器（例如，批量大小为1000），这将为您提供更好的性能，因为您不会发送对服务器的每个请求，但每1000个请求中只有一个，因此使您的更新更高效，更快。

以下示例演示了如何使用MongoDB版本>= 2.6和< 3.2可用的Bulk() API。

function isValidHexStr(id) {
    var checkForHexRegExp = new RegExp("^[0-9a-fA-F]{24}$");
    if(id == null) return false;
    if(typeof id == "string") {
        return id.length == 12 || (id.length == 24 && checkForHexRegExp.test(id));
    }
    return false;
};

var bulkUpdateOps = db.users.initializeUnOrderedBulkOp(), 
    counter = 0;

db.users.find({"Ancestors.0": { "$exists": true, "$type": 2 }}).forEach(function(doc){ 
    var ancestors = doc.Ancestors,
        updateOperatorDocument = {}; 
    for (var idx = 0; idx < ancestors.length; idx++){ 
        if(isValidHexStr(ancestors[idx]))                   
            updateOperatorDocument["Ancestors."+ idx] = ObjectId(ancestors[idx]);           
    };
    bulkUpdateOps.find({ "_id": doc._id }).update({ "$set": updateOperatorDocument })

    counter++;  // increment counter for batch limit
    if (counter % 1000 == 0) { 
        // execute the bulk update operation in batches of 1000
        bulkUpdateOps.execute(); 
        // Re-initialize the bulk update operations object
        bulkUpdateOps = db.users.initializeUnOrderedBulkOp();
    } 
})

// Clean up remaining operation in the queue
if (counter % 1000 != 0) { bulkUpdateOps.execute(); }

---

下一个示例适用于新的MongoDB 3.2版，此版本已弃用 Bulk() API，并使用bulkWrite()提供了一组较新的api。

它使用与上面相同的游标，但是使用相同的forEach()游标方法通过批量操作创建数组，以将每个批量写入文档推入数组。 由于写命令最多只能接受1000个操作，因此您需要将操作分组以最多具有1000个操作，并在循环达到1000次迭代时重新初始化数组：

var cursor = db.users.find({"Ancestors.0": { "$exists": true, "$type": 2 }}),
    bulkUpdateOps = [];

cursor.forEach(function(doc){ 
    var ancestors = doc.Ancestors,
        updateOperatorDocument = {}; 
    for (var idx = 0; idx < ancestors.length; idx++){ 
        if(isValidHexStr(ancestors[idx]))                   
            updateOperatorDocument["Ancestors."+ idx] = ObjectId(ancestors[idx]);           
    };
    bulkUpdateOps.push({ 
        "updateOne": {
            "filter": { "_id": doc._id },
            "update": { "$set": updateOperatorDocument }
         }
    });

    if (bulkUpdateOps.length == 1000) {
        db.users.bulkWrite(bulkUpdateOps);
        bulkUpdateOps = [];
    }
});         

if (bulkUpdateOps.length > 0) { db.users.bulkWrite(bulkUpdateOps); }