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

文章目录

• • • 引言
    • 一、MongoDB 默认的ObjectId
    • 二、实现自动增长ID
    • • 1. 创建自增ID文档
      • 2. 获取并更新自增ID
    • 三、案例代码：实现自增ID
    • • 1. 创建自增ID文档
      • 2. 获取并更新自增ID
      • 3. 插入订单
    • 四、并发场景下的自增ID
    • • 1. 使用锁机制
      • 2. 使用事务
    • 五、结论

引言

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

一、MongoDB 默认的ObjectId

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

1. 时间戳（4字节）：记录ObjectId创建时的时间，单位为秒。
2. 机器标识符（3字节）：表示生成ObjectId的机器，前两字节是网络字节序的机器ID，后一字节是进程ID。
3. 计数器（2字节）：每次在同一台机器同一进程中生成新的ObjectId时，计数器会递增。
4. 随机数（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(); } 

五、结论

自增ID机制非常适合需要连续编号的场景，如订单号、发票号等。需要注意的是，在高并发环境下，要确保并发安全性，可以使用锁机制或事务来处理。

喜欢博主的同学，请给博主一丢丢打赏吧↓↓↓您的支持是我不断创作的最大动力哟！感谢您的支持哦😘😘😘

💝💝💝如有需要请大家订阅我的专栏【MongoDB系列】哟！我会定期更新相关系列的文章
💝💝💝关注！关注！！请关注！！！请大家关注下博主，您的支持是我不断创作的最大动力！！！

❤️❤️❤️觉得有用的话点个赞 👍🏻 呗。
❤️❤️❤️本人水平有限，如有纰漏，欢迎各位大佬评论批评指正！😄😄😄
💘💘💘如果觉得这篇文对你有帮助的话，也请给个点赞、收藏下吧，非常感谢!👍 👍 👍
🔥🔥🔥Stay Hungry Stay Foolish 道阻且长,行则将至,让我们一起加油吧！🌙🌙🌙