MongoDB - 自动递增序列

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

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

使用Counter Collection

请考虑以下产品文件.我们希望_id字段是自动递增的整数序列，从1,2,3,4开始到n.

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

为此，创建一个计数器集合，它将跟踪所有序列字段的最后一个序列值.

>db.createCollection("counters")

现在，我们将在使用 productid 作为其关键号计数器集合;

{
  "_id":"productid",
  "sequence_value": 0
}

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

我们现在将在创建新文档时使用函数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" }