Azure CosmosDB for MongoDB API中updateOne upsert竞态问题求解

你的问题本质是并发upsert操作的竞态条件：单个updateOne({upsert: true})是原子操作，但两个并发请求会各自执行「检查uuid=1234的文档是否存在」→「不存在则插入新文档」的流程，由于这两步不是全局原子的，就会出现两个请求都检测到文档不存在，最终插入两条重复uuid的文档。

数据库层面的解决办法 #

1. 唯一索引 + 冲突重试（推荐） #

给uuid字段创建唯一索引：

db.collection("messages").createIndex({uuid: 1}, {unique: true});

此时并发upsert会触发唯一键冲突异常，你只需要在应用中捕获这个异常，重试对应的update操作即可。重试时，第一个请求已经插入了文档，第二个请求的update会直接更新已存在的文档，最终得到包含foo和bar的单条文档。

这种方案成本低、性能影响小，是最常用的解决方式。

2. 使用多文档事务 #

Azure CosmosDB for MongoDB API（兼容MongoDB 4.0及以上版本）支持多文档事务，你可以把两个updateOne操作放到同一个事务中执行：

const session = await db.client.startSession();
session.startTransaction();
try {
  await db.collection("messages").updateOne({uuid: "1234"}, {$set: {foo: {}}}, {upsert: true, session});
  await db.collection("messages").updateOne({uuid: "1234"}, {$set: {bar: {}}}, {upsert: true, session});
  await session.commitTransaction();
} catch (e) {
  await session.abortTransaction();
  throw e;
} finally {
  session.endSession();
}

事务会保证两个操作的原子性：要么都成功执行（最终合并成单条文档），要么都回滚。但注意事务有性能开销，且要求操作的集合属于同一数据库。

应用层面的替代方案 #

如果不想依赖数据库特性，也可以在应用层做同步：

• 本地锁/分布式锁：对同一uuid的更新请求加锁，保证同一时间只有一个请求执行upsert操作。
• 请求排队：将更新请求放入队列，串行处理同一uuid的操作。

总结 #

不需要强制在应用层同步请求，数据库层面通过「唯一索引+冲突重试」就能解决问题，这也是最简洁高效的方案；如果业务需要强原子性，也可以选择事务方案。

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