在 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 版本支持事務功能。