MongoDB批量新增或替换文档（基于唯一字段myKey）

嘿，这个需求正好戳中MongoDB bulkWrite()方法的强项！它完美支持批量的“Upsert”操作——也就是你要的：存在myKey相同的文档就替换，不存在就新增。下面一步步给你讲怎么实现：

第一步：先给myKey加唯一索引（必做！） #

敲黑板：一定要先给myKey字段创建唯一索引，这样数据库层面能强制保证myKey的唯一性，避免后续操作出现重复数据的坑。

在Mongo Shell里执行这条命令（换成你的集合名）：

db.your_collection_name.createIndex({ myKey: 1 }, { unique: true })

第二步：编写批量Upsert代码 #

方式1：Mongo Shell直接执行 #

把你的待写入文档数组转换成bulkWrite需要的操作数组，然后执行即可：

// 你的待写入文档数组
const docsToUpsert = [
  {"a":0, "b":0, "myKey": "aaaaaa"},
  {"a":0, "b":0, "myKey": "hhhhhh"},
  {"a":0, "b":0, "myKey": "oooooo"},
  {"a":0, "b":0, "myKey": "mmmmmm"},
  {"a":0, "b":0, "myKey": "eeeeee"}
];

// 转换成批量操作数组
const bulkOperations = docsToUpsert.map(doc => ({
  updateOne: {
    filter: { myKey: doc.myKey }, // 匹配规则：找myKey相同的文档
    update: { $set: doc }, // 用当前文档的字段更新匹配到的文档
    upsert: true // 核心开关：没匹配到就插入新文档，匹配到就更新
  }
}));

// 执行批量操作
db.your_collection_name.bulkWrite(bulkOperations);

方式2：Node.js驱动中执行 #

如果是在Node.js项目里用官方驱动，写法类似，只是多了连接数据库的步骤：

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

async function runBulkUpsert() {
  const client = new MongoClient('你的MongoDB连接字符串');
  
  try {
    await client.connect();
    const db = client.db('你的数据库名');
    const collection = db.collection('你的集合名');

    const docsToUpsert = [
      {"a":0, "b":0, "myKey": "aaaaaa"},
      {"a":0, "b":0, "myKey": "hhhhhh"},
      {"a":0, "b":0, "myKey": "oooooo"},
      {"a":0, "b":0, "myKey": "mmmmmm"},
      {"a":0, "b":0, "myKey": "eeeeee"}
    ];

    const bulkOperations = docsToUpsert.map(doc => ({
      updateOne: {
        filter: { myKey: doc.myKey },
        update: { $set: doc },
        upsert: true
      }
    }));

    const result = await collection.bulkWrite(bulkOperations);
    console.log(`搞定！新增了${result.insertedCount}条文档，更新了${result.modifiedCount}条现有文档`);
  } finally {
    await client.close();
  }
}

runBulkUpsert();

可选：完全替换文档 vs 仅更新字段 #

上面用的$set是仅更新文档中存在的字段——如果匹配到的旧文档有额外字段，这些字段会被保留。如果你想要完全替换匹配到的文档（除了_id），可以把updateOne换成replaceOne：

replaceOne: {
  filter: { myKey: doc.myKey },
  replacement: doc, // 用当前文档完全替换旧文档
  upsert: true
}

内容的提问来源于stack exchange，提问作者Gandalf The Gray