MongoDB批量插入文档遇唯一索引冲突时如何全量更新？

问题场景 #

需要一次性处理5000条文档：无唯一索引冲突的新文档直接插入，存在冲突的文档则用新文档全量替换现有内容。尝试过insertMany({unordered: true})会跳过重复文档，updateMany({upsert: true})仅适合更新指定字段，无法实现全量替换。

解决方案：使用bulkWrite()批量执行replaceOne操作 #

MongoDB的bulkWrite()方法支持批量执行多种操作类型，结合replaceOne和upsert: true参数，可完美实现类似MySQL ON DUPLICATE KEY UPDATE的全量替换逻辑：

• replaceOne：用新文档完全替换匹配到的现有文档（若新文档未指定_id，则保留原文档的_id）
• upsert: true：当无匹配的唯一索引文档时，自动插入新文档

示例实现（Mongo Shell/Node.js通用逻辑） #

针对你的business集合（name为唯一索引），处理4条目标文档的代码如下：

db.business.bulkWrite([
  // 处理Apple：匹配name则全量替换，否则插入
  {
    replaceOne: {
      filter: { name: "Apple" },
      replacement: { name: "Apple", address: "...", employees: 55177, phone: 22222 },
      upsert: true
    }
  },
  // 处理Dell
  {
    replaceOne: {
      filter: { name: "Dell" },
      replacement: { name: "Dell", address: "...", employees: 77889, phone: 11223 },
      upsert: true
    }
  },
  // 处理Google
  {
    replaceOne: {
      filter: { name: "Google" },
      replacement: { name: "Google", address: "...", employees: 33333, phone: 44444 },
      upsert: true
    }
  },
  // 处理IBM
  {
    replaceOne: {
      filter: { name: "IBM" },
      replacement: { name: "IBM", address: "...", employees: 77777, phone: 88888 },
      upsert: true
    }
  }
])

批量处理5000条文档的优化 #

直接处理5000条可能触发MongoDB的批量操作限制，建议分批次执行（比如每1000条为一批），避免内存占用过高：

// 假设docs是包含5000条目标文档的数组
const batchSize = 1000;
for (let i = 0; i < docs.length; i += batchSize) {
  const batch = docs.slice(i, i + batchSize).map(doc => ({
    replaceOne: {
      filter: { name: doc.name }, // 以唯一索引字段为查询条件
      replacement: doc,
      upsert: true
    }
  }));
  db.business.bulkWrite(batch);
}

执行结果验证 #

执行后business集合的文档将和你预期的一致：

• Google、Apple被全量替换
• Dell、IBM被插入
• Microsoft、Meta保持不变

内容的提问来源于stack exchange，提问作者datasn.io