Mongo中使用$sample与updateOne是否安全？竞态条件及解决办法咨询

原方案的竞态问题 #

你提到的方案（比如先通过$sample获取随机文档再更新）确实存在竞态条件：高并发下，两个请求可能同时拿到同一个随机文档，后执行的更新会覆盖前一次的操作，导致数据异常。

正确的原子实现方式 #

解决核心是让查询和更新在同一个原子操作中完成，利用MongoDB的原子性保证同一时间只有一个请求能修改目标文档。具体步骤如下：

1. 给文档添加状态标记 #

先给集合里的文档加一个用于标记是否被选中的字段，比如isLocked，默认值为false（表示未被占用）。

2. 原子化的findOneAndUpdate调用 #

使用findOneAndUpdate，在查询条件中同时指定「未被锁定」和「随机筛选」，并在更新步骤中标记该文档为已锁定。这样MongoDB会自动保证这个操作的原子性——第一个匹配到文档的请求会完成更新，后续请求无法再选中该文档。

示例代码（Node.js MongoDB驱动）：

const selectedDoc = await db.collection('your_collection').findOneAndUpdate(
  {
    isLocked: false,
    // 用$rand()实现随机筛选，$lt后的数值控制随机命中概率，比如0.05表示从5%的未锁定文档里随机选
    $expr: { $lt: [ { $rand: {} }, 0.05 ] }
  },
  {
    $set: {
      isLocked: true,
      assignedAt: new Date() // 可选：记录锁定时间
    }
  },
  {
    returnDocument: 'after', // 返回更新后的文档
    upsert: false // 没有匹配到就不创建新文档
  }
);

if (selectedDoc.value) {
  // 成功获取并锁定了随机文档
  console.log('选中的随机文档:', selectedDoc.value);
} else {
  // 没有可用的未锁定文档
  console.log('无可用文档');
}

3. 可选：自动解锁机制 #

如果需要文档在一段时间后自动解锁，可以给assignedAt字段添加TTL索引，让MongoDB自动重置锁定标记：

// 创建TTL索引，锁定1小时后自动将isLocked改回false
await db.collection('your_collection').createIndex(
  { assignedAt: 1 },
  { expireAfterSeconds: 3600, partialFilterExpression: { isLocked: true } }
);

是否需要悲观锁？ #

不需要。MongoDB的findOneAndUpdate本身就是原子操作，底层通过乐观锁机制保证数据一致性，性能远高于悲观锁。只有在极端复杂的业务场景（比如跨多个文档的事务操作）下，才需要考虑悲观锁，你的场景用原子操作完全足够。

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