如何用Mongoose实现MongoDB的UPSERT：同唯一键更新异则插入？

你提到的“查询后插入”方案在批量场景下确实会遇到性能瓶颈，甚至可能因为竞态条件（比如多个请求同时处理同一个field_1时，都判断为不存在然后插入，触发唯一键冲突）导致数据异常。针对你的需求，MongoDB的原子化upsert操作是最优解——它能在单次数据库请求中完成“存在则更新，不存在则创建”的逻辑，完美适配批量场景。

第一步：给field_1创建唯一索引 #

首先要确保你的集合在field_1上创建唯一索引，这是保证数据唯一性的前提，也能大幅提升upsert操作的效率：

db.yourCollection.createIndex({ field_1: 1 }, { unique: true });

单文档操作：用updateOne实现核心逻辑 #

单条数据的处理代码非常简洁，核心是利用upsert: true参数自动判断创建或更新：

db.yourCollection.updateOne(
  { field_1: 'abc' }, // 匹配条件：定位field_1为目标值的文档
  {
    $setOnInsert: { field_2: 0, field_3: [] }, // 仅当创建新文档时，设置和你示例一致的初始值
    $inc: { field_2: 1 }, // 无论更新还是创建，都递增field_2
    $push: { field_3: '新元素' } // 无论更新还是创建，都向field_3追加元素
  },
  { upsert: true } // 关键参数：无匹配文档时自动创建新文档
)

这段代码的执行逻辑：

• 如果field_1: 'abc'的文档已存在：直接执行$inc和$push，原子更新字段值
• 如果不存在：先通过$setOnInsert初始化field_2和field_3，再执行递增和追加操作，最终生成符合你示例结构的新文档

批量场景优化：用bulkWrite打包操作 #

如果需要处理批量的“添加/更新”请求，用bulkWrite把多个upsert操作打包提交，能大幅减少网络往返次数，性能提升非常明显：

const batchOperations = [
  // 处理field_1='abc'的请求
  {
    updateOne: {
      filter: { field_1: 'abc' },
      update: {
        $setOnInsert: { field_2: 0, field_3: [] },
        $inc: { field_2: 1 },
        $push: { field_3: 'abc专属元素' }
      },
      upsert: true
    }
  },
  // 处理field_1='def'的请求
  {
    updateOne: {
      filter: { field_1: 'def' },
      update: {
        $setOnInsert: { field_2: 0, field_3: [] },
        $inc: { field_2: 1 },
        $push: { field_3: 'def专属元素' }
      },
      upsert: true
    }
  }
  // 可继续添加更多操作
];

// 执行批量操作
db.yourCollection.bulkWrite(batchOperations);

为什么这个方案比“查询后插入”更优？ #

• 原子性保障：整个判断和更新逻辑在数据库层面原子完成，不会出现竞态条件导致的数据冲突或丢失
• 性能高效：单操作仅需1次数据库请求，批量操作通过打包避免了多次查询+插入的网络开销
• 自动处理唯一键：配合field_1的唯一索引，upsert操作会自动避免重复创建文档，无需在客户端做额外判断

内容的提问来源于stack exchange，提问作者Flyn Sequeira