MongoDB：如何确保updateOne()完成后再执行find()？

MongoDB原生处理方案 #

MongoDB没有直接提供“阻塞所有查询直到更新完成”的集合级锁API——它基于文档级并发锁设计，优先保障高并发性能。不过可以通过以下机制避免读写不一致：

乐观锁机制 #

给sessions集合的每个文档新增version字段，利用版本号校验实现一致性：

• 查询活跃会话时，同时获取文档的version值
• 执行关闭会话的updateOne操作时，带上版本号作为过滤条件：

  filter := bson.M{"_id": sessionID, "version": currentVersion}
  update := bson.M{"$set": bson.M{"status": "closed", "version": currentVersion + 1}}
  result, err := collection.UpdateOne(ctx, filter, update)
  

  若result.MatchedCount == 0，说明查询到的文档已被修改，需重新查询最新的活跃会话。

事务（多文档操作场景） #

如果匹配逻辑涉及多文档读写，可使用MongoDB事务包裹操作，确保读操作拿到一致的数据视图。事务仅支持副本集或分片集群环境：

session, err := client.StartSession()
if err != nil {
    // 错误处理
}
defer session.EndSession(ctx)

err = mongo.WithSession(ctx, session, func(sc mongo.SessionContext) error {
    if err := session.StartTransaction(); err != nil {
        return err
    }

    // 查询活跃会话
    var sessionDoc Session
    err := collection.FindOne(sc, bson.M{"status": "active"}).Decode(&sessionDoc)
    if err != nil {
        _ = session.AbortTransaction(sc)
        return err
    }

    // 执行加入会话的更新操作
    _, err = collection.UpdateOne(sc, bson.M{"_id": sessionDoc.ID}, bson.M{"$push": bson.M{"users": userID}})
    if err != nil {
        _ = session.AbortTransaction(sc)
        return err
    }

    return session.CommitTransaction(sc)
})

Go应用层同步锁实现 #

单实例部署时，可通过sync.RWMutex在代码层实现读写互斥，确保写操作（关闭会话）执行期间，所有读操作（查询活跃会话）被阻塞：

1. 定义全局读写锁 #

import "sync"

var sessionRWMutex sync.RWMutex

2. 查询活跃会话加读锁 #

读锁允许多个读操作并行，但会阻塞写操作：

func findActiveSession(ctx context.Context, collection *mongo.Collection) (*Session, error) {
    sessionRWMutex.RLock()
    defer sessionRWMutex.RUnlock()

    var sessionDoc Session
    err := collection.FindOne(ctx, bson.M{"status": "active"}).Decode(&sessionDoc)
    return &sessionDoc, err
}

3. 关闭会话加写锁 #

写锁会阻塞所有读、写操作，直到锁释放：

func closeSession(ctx context.Context, collection *mongo.Collection, sessionID primitive.ObjectID) error {
    sessionRWMutex.Lock()
    defer sessionRWMutex.Unlock()

    _, err := collection.UpdateOne(ctx, bson.M{"_id": sessionID}, bson.M{"$set": bson.M{"status": "closed"}})
    return err
}

注意：该方案仅适用于单实例应用，多实例部署需改用分布式锁（如基于MongoDB实现的分布式锁）保障跨实例的读写互斥。

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