50、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 而已

使用 Javascript 函数

既然创建好了函数，是时候使用它了

我们可以像下面一样调用 getNextSequenceValue 函数创建一个新的文档

并设置文档 _id 自动为返回的序列值

>db.language.insert({
   "_id":getNextSequenceValue("language_id"),
   "name":"Kotlin",
   "category":"APP"})

>db.language.insert({
   "_id":getNextSequenceValue("language_id"),
   "name":"JavaScript",
   "category":"script"})

为了验证函数是否有效，我们可以使用以下命令读取文档

>db.language.find()

如果一切顺利，你就能看到如下的输出了 以上命令将返回以下结果，我们发现 _id 字段是自增长的：

> db.language.find()
{ "_id" : 1, "name" : "Kotlin", "category" : "APP" }
{ "_id" : 3, "name" : "JavaScript", "category" : "script" }