更新时间:2023-01-22 17:28:32
An alternative is to use bulk.find().upsert().replaceOne()
instead:
MongoClient.connect(mongoURL, function(err, db) {
if(err) console.err(err)
let col = db.collection('user_ids')
let batch = col.initializeUnorderedBulkOp()
ids.forEach(function(id) {
batch.find({ userid: id }).upsert().replaceOne({
userid: id,
used: false,
group: argv.groupID
});
});
batch.execute(function(err, result) {
if(err) {
console.error(new Error(err))
db.close()
}
// Do some work
db.close()
});
});
使用上述方法,如果文档与查询{ userid: id }
相匹配,它将被新文档替换,否则将被创建,因此不会引发重复的键错误.
With the above, if a document matches the query { userid: id }
it will be replaced with the new document, otherwise it will be created hence there are No duplicate key errors thrown.
对于3.2或更高版本的MongoDB服务器,请使用 bulkWrite
为:
For MongoDB server versions 3.2+, use bulkWrite
as:
MongoClient.connect(mongoURL, function(err, db) {
if(err) console.err(err)
let col = db.collection('user_ids')
let ops = []
let counter = 0
ids.forEach(function(id) {
ops.push({
"replaceOne": {
"filter": { "userid": id },
"replacement": {
userid: id,
used: false,
group: argv.groupID
},
"upsert": true
}
})
counter++
if (counter % 500 === 0) {
col.bulkWrite(ops, function(err, r) {
// do something with result
db.close()
})
ops = []
}
})
if (counter % 500 !== 0) {
col.bulkWrite(ops, function(err, r) {
// do something with result
db.close()
}
}
})