面试题：MongoDB更新操作中的原子性与并发控制

保证更新原子性

单个文档更新：

MongoDB 对于单个文档的更新操作本身就是原子性的。例如，使用 updateOne 或 updateMany 方法对单个文档内的字段进行更新时，MongoDB 会确保整个更新操作要么完全成功，要么完全失败。
示例代码（Node.js 驱动）：

const { MongoClient } = require('mongodb');
const uri = "mongodb://localhost:27017";
const client = new MongoClient(uri);
async function updateDocument() {
    try {
        await client.connect();
        const database = client.db('testDB');
        const collection = database.collection('testCollection');
        const filter = { _id: 1 };
        const update = { $set: { field1: 'new value' } };
        const result = await collection.updateOne(filter, update);
        console.log(result);
    } catch (e) {
        console.error(e);
    } finally {
        await client.close();
    }
}
updateDocument();

更新嵌套文档：同样，单个文档内嵌套文档的更新也是原子性的。例如，更新一个文档中嵌套数组内的元素：

示例代码（Python 驱动）：

from pymongo import MongoClient
client = MongoClient('mongodb://localhost:27017/')
db = client['testDB']
collection = db['testCollection']
filter = {'_id': 1}
update = {'$set': {'nestedArray.0.field': 'new value'}}
result = collection.update_one(filter, update)
print(result)

多线程/多进程考虑：在多线程或多进程环境下，只要更新操作是针对单个文档，原子性依然能保证。但如果多个线程/进程同时更新不同文档，需要注意事务处理（MongoDB 4.0+ 支持多文档事务）。

处理并发冲突

乐观并发控制：

MongoDB 默认使用乐观并发控制。当多个操作尝试更新同一文档时，先到达的操作会成功，后续操作如果检测到文档版本已改变（通过文档的内部版本号，如 _ts 字段，在多文档事务中使用），则会失败并抛出异常。
例如，在 Node.js 中使用多线程库（如 worker_threads）模拟并发更新同一文档：

const { Worker } = require('worker_threads');
const { MongoClient } = require('mongodb');
const uri = "mongodb://localhost:27017";
const client = new MongoClient(uri);
async function updateInWorker() {
    try {
        await client.connect();
        const database = client.db('testDB');
        const collection = database.collection('testCollection');
        const filter = { _id: 1 };
        const update = { $inc: { counter: 1 } };
        const result = await collection.updateOne(filter, update);
        console.log(result);
    } catch (e) {
        console.error(e);
    } finally {
        await client.close();
    }
}
const worker1 = new Worker('./worker.js');
const worker2 = new Worker('./worker.js');

在 worker.js 中执行上述 updateInWorker 逻辑，第二个更新操作可能会因为文档版本改变而失败。

多文档事务：

从 MongoDB 4.0 开始支持多文档事务。通过事务，可以确保多个文档的更新操作要么全部成功，要么全部回滚，从而避免并发冲突导致的数据不一致。
示例代码（Java 驱动）：

MongoClient mongoClient = MongoClients.create("mongodb://localhost:27017");
MongoDatabase database = mongoClient.getDatabase("testDB");
try (ClientSession clientSession = mongoClient.startSession()) {
    clientSession.startTransaction();
    database.getCollection("collection1").updateOne(clientSession, Filters.eq("_id", 1), Updates.set("field1", "new value"));
    database.getCollection("collection2").updateOne(clientSession, Filters.eq("_id", 2), Updates.set("field2", "new value"));
    clientSession.commitTransaction();
} catch (Exception e) {
    e.printStackTrace();
}

在事务中，如果一个更新操作失败，整个事务会回滚，保证数据一致性。

重试机制：当出现并发冲突导致更新失败时，可以在应用层实现重试机制。捕获异常后，重新执行更新操作，直到成功为止。例如在 Python 中：

from pymongo import MongoClient
from pymongo.errors import WriteError
client = MongoClient('mongodb://localhost:27017/')
db = client['testDB']
collection = db['testCollection']
filter = {'_id': 1}
update = {'$set': {'field': 'new value'}}
max_retries = 3
for i in range(max_retries):
    try:
        result = collection.update_one(filter, update)
        break
    except WriteError as e:
        if i == max_retries - 1:
            raise e

星途面试题库

面试题：MongoDB更新操作中的原子性与并发控制

知识考点

面试题答案

保证更新原子性

处理并发冲突