MongoDB 自动递增序列

  • 自动递增序列

    MongoDB没有像SQL数据库那样的开箱即用的自动增量功能。默认情况下,它使用_id字段的12字节ObjectId 作为主键来唯一标识文档。但是,在某些情况下,我们可能希望_id字段具有除ObjectId之外的一些自动递增的值。由于这不是MongoDB中的默认功能,因此我们将通过使用MongoDB文档建议的counters集合以编程方式实现此功能。
  • 使用 counters 集合

    考虑以下产品文档。我们希望_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跟踪序列的最后一个值。
    使用以下代码将此序列文档插入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函数

    现在,我们将在创建新文档并将返回的序列值分配为文档的_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" }