MongoDB 自动递增序列

MongoDB 没有像 SQL 一样有自动增长的功能, MongoDB 的 _id 是系统自动生成的12字节唯一标识。但是,在某些情况下,我们可能希望 _id 字段具有除 ObjectId 之外的一些自动递增值。

由于这不是 MongoDB 中的默认功能,我们将按照 MongoDB 文档的建议,通过使用计数器集合以编程方式实现此功能。

使用计数器集合

考虑以下 products 文档。我们希望 _id 字段实现 从 1,2,3,4 到 n 的自动增长功能。

{
  "_id":1,
  "product_name": "Apple iPhone",
  "category": "mobiles"
}

为此,创建 counters 集合,序列字段值可以实现自动长:

> db.createCollection("counters")

现在,我们将在 counters 集合中插入以下文档,并将productid作为其键

> db.counters.insert({
    "_id":"productid",
    "sequence_value": 0
})
WriteResult({ "nInserted" : 1 })
>

字段 sequence_value 跟踪序列的最后一个值。

使用以下代码在 counters 集合中插入此序列文档

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

创建 Javascript 函数

现在,我们创建函数 getNextSequenceValue 来作为序列名的输入, 指定的序列会自动增长 1 并返回最新序列值。在本文的实例中序列名为 productid 。

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

使用 JavaScript 函数

接下来我们将使用 getNextSequenceValue 函数创建一个新的文档, 并设置文档 _id 自动为返回的序列值:

> db.products.insert({
   "_id":getNextSequenceValue("productid"),
   "product_name":"Apple iPhone",
   "category":"mobiles"
})
> db.products.insert({
   "_id":getNextSequenceValue("productid"),
   "product_name":"Samsung S3",
   "category":"mobiles"
})

如您所见,我们使用了 getNextSequenceValue 函数来设置 _id 字段的值。

为了验证功能,让我们使用 find 命令获取文档

> db.products.find()

上述查询返回以下具有自动递增 _id 字段的文档

{ "_id" : 1, "product_name" : "Apple iPhone", "category" : "mobiles"}
{ "_id" : 2, "product_name" : "Samsung S3", "category" : "mobiles" }

查看笔记

扫码一下
查看教程更方便