如何使用Mongo JS批量更新

debugcn 发表于 Dev

疯狂编码器

我试图在所有文档中基于另一个字段插入一个新字段。目前，我有近3.3亿份文件。一张一张地更新要花费很多时间。下面是我的示例代码，

var db = connect('127.0.0.1:27017/sampleDatabase');
print('Database connect');
 
collectionsList = [1 .... 100];
 
collectionsList.forEach(addNewTimeField);
 
function addNewTimeField(value, index, array){ 
    let collectionName = 'name_' + value;
    db[collectionName].find().forEach(function(doc) {
        db[collectionName].update({_id:doc._id}, {$set:{"ts": new Date(doc.timeStamp)}});   
    });
    db[collectionName].createIndex({ "ts" : 1 }, { expireAfterSeconds: 31622400 }, { "background": true} );
}

有没有一种方法可以使用批量更新同时更新多个记录？提前致谢。

土生的

您可以尝试使用MongoDB v4.2中的聚合管道进行更新，

使用updateMany（）
$ toDate将时间戳转换为日期，从MongoDB v4.0起

db[collectionName].updateMany(
  {}, 
  [{ 
    $set: { ts: { $toDate: "$timeStamp" } }
  }]
);

第二个选项是bulkWrite（），它将更新单个事务中的所有记录，

async function addNewTimeField(value, index, array){ 
  let collectionName = 'name_' + value;
  let allDocs = await db[collectionName].find({}, { timeStamp: 1 });
  let updateAllDocs = allDocs.map(({ _id, timeStamp }) => ({
    updateOne: {
      filter: { _id: ObjectId(_id) },
      update: { ts: new Date(timeStamp) }
    }
  }));
  db[collectionName].bulkWrite(updateAllDocs);
  db[collectionName].createIndex({ "ts" : 1 }, { expireAfterSeconds: 31622400 }, { "background": true} );
}