MongoDB教程（十五）：MongoDB原子操作

用户11147438

发布于 2025-05-29 11:49:30

2870

💝💝💝首先，欢迎各位来到我的博客，很高兴能够在这里和您见面！希望您在这里不仅可以有所收获，同时也能感受到一份轻松欢乐的氛围，祝你生活愉快！

引言

在多线程或分布式环境中，保证数据一致性是一项重大挑战。原子操作是解决这一问题的关键技术之一，它确保一系列操作要么全部成功，要么全部失败，从而维持数据的完整性和一致性。MongoDB 提供了一系列内置的原子操作，使开发者能够轻松地在数据库层面实现事务性和数据完整性。本文将深入探讨 MongoDB 中的原子操作，包括更新、插入和删除操作的原子性，并通过具体案例代码展示如何在实际应用中运用这些原子操作。

一、MongoDB 原子操作概述

MongoDB 中的原子操作通常指的是在单个文档或单个写操作级别上，确保操作的不可分割性和隔离性。在4.0版本之后，MongoDB 还引入了多文档事务，允许在多个文档或集合上执行跨文档的原子操作。

二、单文档原子操作

1. 更新操作

MongoDB 的更新操作天然具备原子性。例如，updateOne() 和 updateMany() 方法保证了更新操作要么完全应用，要么完全不应用。

db.users.updateOne(
    { _id: ObjectId("5f9c9d...") },
    { $inc: { balance: 100 } }
);

在这个例子中，如果更新操作因为某种原因（比如磁盘满）失败，那么文档的balance字段将保持不变，确保了数据的一致性。

2. 插入操作

insertOne() 和 insertMany() 方法同样具有原子性，如果插入失败，文档将不会被部分插入。

db.users.insertOne(
    { _id: "123", name: "John Doe", age: 30 }
);

3. 删除操作

deleteOne() 和 deleteMany() 方法同样保证了操作的原子性。

db.users.deleteOne(
    { _id: ObjectId("5f9c9d...") }
);

三、多文档原子操作

在 MongoDB 4.0 及以上版本中，可以使用事务来实现跨文档的原子操作。

1. 启用事务

事务在副本集或分片集群中是可用的，需要在配置服务器中启用事务。

2. 使用事务

事务操作需要在 withTransaction() 函数中定义，该函数接收一个回调函数作为参数，在这个回调函数中可以执行一系列的写操作。

const session = db.getMongo().startSession();
session.startTransaction();

try {
    db.users.updateOne(
        { _id: ObjectId("5f9c9d...") },
        { $inc: { balance: 100 } },
        { session }
    );
    
    db.transactions.insertOne(
        { _id: "tx123", description: "Deposit", amount: 100 },
        { session }
    );

    session.commitTransaction();
} catch (error) {
    session.abortTransaction();
} finally {
    session.endSession();
}

在这个例子中，如果更新操作或插入操作中的任何一个失败，事务将被回滚，所有操作都不会被应用。

四、案例：库存管理系统

假设我们正在构建一个库存管理系统，需要确保在减少商品库存的同时，记录一笔交易日志。这是一个典型的需要多文档原子操作的场景。

1. 创建集合

首先，创建 inventory 和 transactions 集合。

db.createCollection("inventory");
db.createCollection("transactions");

2. 执行事务

使用事务来减少库存并记录交易。

const session = db.getMongo().startSession();
session.startTransaction();

try {
    db.inventory.updateOne(
        { _id: "item123" },
        { $inc: { quantity: -1 } },
        { session }
    );
    
    db.transactions.insertOne(
        { _id: "tx456", item: "item123", quantity: -1 },
        { session }
    );

    session.commitTransaction();
} catch (error) {
    session.abortTransaction();
} finally {
    session.endSession();
}