MongoDB教程（二十三）：关于MongoDB自增机制

用户11147438

发布于 2025-05-29 12:42:26

3070

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

引言

在MongoDB中，自动生成或自动增长的ID是许多应用场景中的常见需求，特别是在需要连续编号的情况下。尽管MongoDB默认使用ObjectId作为文档的主键，但在某些情况下，开发者可能需要实现自定义的自增ID机制。本文将深入探讨如何在MongoDB中实现自动增长的ID，并通过具体的案例代码展示这一过程的每一个细节。

一、MongoDB 默认的ObjectId

MongoDB 默认使用ObjectId作为文档的_id字段。ObjectId是一个12字节的BSON类型，由以下四个部分组成：

时间戳（4字节）：记录ObjectId创建时的时间，单位为秒。
机器标识符（3字节）：表示生成ObjectId的机器，前两字节是网络字节序的机器ID，后一字节是进程ID。
计数器（2字节）：每次在同一台机器同一进程中生成新的ObjectId时，计数器会递增。
随机数（3字节）：增加随机性，降低冲突概率。

二、实现自动增长ID

在某些场景下，如订单编号、流水号等，需要使用连续的数字作为ID。MongoDB 不直接支持自增ID，但可以通过创建一个文档来模拟实现。

1. 创建自增ID文档

首先，需要在数据库中创建一个用于存储自增ID的文档。

db.auto_incr_ids.insert({ _id: "orders", sequence_value: 0 });

这里，orders 是自增ID的名称，sequence_value 是当前的ID值。

2. 获取并更新自增ID

每当需要一个新的自增ID时，可以通过原子操作获取并更新该文档。

db.auto_incr_ids.update(
    { _id: "orders" },
    { $inc: { sequence_value: 1 } },
    { upsert: true, new: true }
);

这里，$inc 操作符用于增加 sequence_value 的值，upsert 选项表示如果文档不存在，则创建新文档。

三、案例代码：实现自增ID

假设我们正在开发一个电子商务平台，需要为每个订单生成唯一的自增ID。

1. 创建自增ID文档

db.auto_incr_ids.insert({ _id: "order_numbers", sequence_value: 0 });

2. 获取并更新自增ID

function getNextSequence(name) {
    let result = db.auto_incr_ids.findOneAndUpdate(
        { _id: name },
        { $inc: { sequence_value: 1 } },
        { upsert: true, new: true }
    );
    return result.sequence_value;
}

let orderId = getNextSequence("order_numbers");
console.log(orderId);

这里，getNextSequence 函数用于获取下一个自增ID。

3. 插入订单

db.orders.insertOne({
    order_number: getNextSequence("order_numbers"),
    customer_name: "John Doe",
    items: [
        { product: "T-shirt", quantity: 2 },
        { product: "Jeans", quantity: 1 }
    ],
    total_amount: 99.99
});

四、并发场景下的自增ID

在高并发场景下，直接使用 findOneAndUpdate 可能会遇到竞态条件。为了确保线程安全，可以使用锁机制或事务来处理。

1. 使用锁机制

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

try {
    let result = db.auto_incr_ids.findOneAndUpdate(
        { _id: "order_numbers" },
        { $inc: { sequence_value: 1 } },
        { upsert: true, new: true, session }
    );

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

2. 使用事务

在MongoDB 4.0及以上版本，可以使用事务来确保操作的原子性。

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

try {
    let result = db.auto_incr_ids.findOneAndUpdate(
        { _id: "order_numbers" },
        { $inc: { sequence_value: 1 } },
        { upsert: true, new: true, session }
    );

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