MongoDB 自動(dòng)遞增序列

MongoDB沒(méi)有像SQL數(shù)據(jù)庫(kù)那樣的開(kāi)箱即用的自動(dòng)增量功能。默認(rèn)情況下，它將_id字段的12字節(jié)ObjectId作為主鍵來(lái)唯一標(biāo)識(shí)文檔。但是，在某些情況下，我們可能希望_id字段具有除ObjectId之外的一些自動(dòng)遞增的值。

由于這不是MongoDB中的默認(rèn)功能，因此我們將通過(guò)使用countersMongoDB文檔建議的集合以編程方式實(shí)現(xiàn)此功能。

使用計(jì)數(shù)器集合

請(qǐng)看以下products文檔。我們希望_id字段是一個(gè)從1,2,3,4到n的自遞增整數(shù)序列。

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

為此，創(chuàng)建一個(gè)counters集合，該集合將跟蹤所有序列字段的最后一個(gè)序列值。

>db.createCollection("counters")

現(xiàn)在，我們將以下文檔productid作為其鍵插入counters集合中-

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

使用以下代碼將此序列文檔插入counters集合中-

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

創(chuàng)建Javascript函數(shù)

現(xiàn)在，我們將創(chuàng)建一個(gè)函數(shù)getNextSequenceValue，該函數(shù)將序列名稱作為輸入，將序列號(hào)加1并返回更新的序列號(hào)。在我們的實(shí)例中，序列名稱為productid。

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

使用Javascript函數(shù)

現(xiàn)在，我們將在創(chuàng)建新文檔并將返回的序列值分配為文檔的_id字段時(shí)使用getNextSequenceValue函數(shù)。

使用以下代碼插入兩個(gè)樣本文檔-

>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函數(shù)來(lái)設(shè)置_id字段的值。

為了驗(yàn)證功能，讓我們使用find命令獲取文檔-

>db.products.find()

上面的查詢返回了具有自動(dòng)遞增的_id字段的以下文檔-

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