在 MongoDB 中,悲观锁不是通过直接设置来实现的,而是通过使用事务(transactions)来实现的。事务是一组操作,要么全部执行成功,要么全部回滚。
在 MongoDB 中,可以使用集合的 startSession()
方法来创建一个会话,并使用会话来执行事务。
下面是一个使用 MongoDB 事务和会话来实现悲观锁的示例:
const session = db.getMongo().startSession();
session.startTransaction();
try {
// 获取锁
db.collection.updateOne(
{ _id: ObjectId("documentId"), locked: false },
{ $set: { locked: true } }
);
// 在锁定的状态下执行操作
// ...
// 释放锁
db.collection.updateOne(
{ _id: ObjectId("documentId"), locked: true },
{ $set: { locked: false } }
);
session.commitTransaction();
} catch (error) {
session.abortTransaction();
throw error;
} finally {
session.endSession();
}
在上述示例中,我们创建了一个会话(session),并开始了一个事务(transaction)。在事务中,我们首先通过 updateOne()
方法获取到锁,然后在锁定的状态下执行操作,最后释放锁。如果在事务中出现了错误,我们会中止事务,并抛出错误。无论事务是否成功,最后都会关闭会话。
需要注意的是,MongoDB 的事务和会话功能是从 MongoDB 4.0 版本开始引入的,所以需要确保你的 MongoDB 版本支持事务功能。