MongoDB自动递增序列


MongoDB没有像SQL数据库那样的开箱即用的自动增量功能。默认情况下,它将12字节的ObjectId用于_id字段作为唯一标识文档的主键。但是,在某些情况下,我们可能希望_id字段具有除ObjectId之外的一些自动递增的值。

由于这不是MongoDB中的默认功能,因此我们将通过使用countersMongoDB文档建议的集合。

使用counters集合


考虑以下products文档,我们希望_id字段为自动递增的整数序列从1,2,3,4开始直到n。

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

为此,创建一个counters集合,它将跟踪所有序列字段的最后一个序列值。

>db.createCollection("counters")

现在,我们将以下文档插入计数器集合中,如下所示:productid作为其键:

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

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

使用以下代码将此序列文档插入计数器集合中:

>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函数


现在,我们将在创建新文档并将返回的序列值分配为文档的_id字段时使用getNextSequenceValue函数。

使用以下代码插入两个示例文档:

>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" }