跳到主要内容

五十、MongoDB 自增 ID

MongoDB 的 _id 是系统自动生成的 12 字节唯一标识

为了实现 ObjectId 自动增长功能，我们需要另外取巧的方法来实现

接下来我们将学习如何在 counters 集合中实现自增字段 _id

使用 counters 集合​

假设我们有如下的 language 文档

我们希望 _id 字段是自增的，也就是从 1,2,3,4 到 n 的自动增长

{
  "_id":1,
  "name": "Python",
  "category": "server"

取巧的方法，就是创建 counters 集合，序列字段值可以实现自动长

> db.createCollection("counters")

然后向counters 集合中插入以下文档，使用 language_id 作为 key

{
  "_id":"language_id",
  "sequence_value": 0

sequence_value 字段是 language 中的 “_id” 通过自动增长后的一个值

最后，使用以下命令插入 counters 集合的序列文档中

>db.counters.insert({_id:"language_id",sequence_value:0})

准备工作到这里就完成了，接下来创建 JavaScript 函数

创建 Javascript 函数​

我们创建一个函数 getNextSequenceValue 作为序列名的输入

实现指定的序列自动增长 1 并返回最新序列值

>function getNextSequenceValue( sequenceName ) {
   var sequenceDocument = db.counters.findAndModify(
      {
         query:{_id: sequenceName },
         update: {$inc:{sequence_value:1}},
         new:true
      });
   return sequenceDocument.sequence_value;

getNextSequenceValue 函数也很简单，就是自增 sequence_value 而已