MongoDB 自动增长

MongoDB 没有像 SQL 一样有自动增长的功能, MongoDB 的 _id 是系统自动生成的12字节唯一标识。

但在某些情况下,我们可能需要实现 ObjectId 自动增长功能。

由于 MongoDB 没有实现这个功能,我们可以通过编程的方式来实现,以下我们将在 counters 集合中实现_id字段自动增长。


使用 counters 集合

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

  1. {
  2. "_id":1,
  3. "product_name": "Apple iPhone",
  4. "category": "mobiles"
  5. }

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

  1. >db.createCollection("counters")

现在我们向 counters 集合中插入以下文档,使用 productid 作为 key:

  1. {
  2. "_id":"productid",
  3. "sequence_value": 0
  4. }

sequence_value 字段是序列通过自动增长后的一个值。

使用以下命令插入 counters 集合的序列文档中:

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

创建 Javascript 函数

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

  1. >function getNextSequenceValue(sequenceName){
  2. var sequenceDocument = db.counters.findAndModify(
  3. {
  4. query:{_id: sequenceName },
  5. update: {$inc:{sequence_value:1}},
  6. new:true
  7. });
  8. return sequenceDocument.sequence_value;
  9. }

使用 Javascript 函数

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

  1. >db.products.insert({
  2. "_id":getNextSequenceValue("productid"),
  3. "product_name":"Apple iPhone",
  4. "category":"mobiles"})
  5.  
  6. >db.products.insert({
  7. "_id":getNextSequenceValue("productid"),
  8. "product_name":"Samsung S3",
  9. "category":"mobiles"})

就如你所看到的,我们使用 getNextSequenceValue 函数来设置 _id 字段。

为了验证函数是否有效,我们可以使用以下命令读取文档:

  1. >db.prodcuts.find()

以上命令将返回以下结果,我们发现 _id 字段是自增长的:

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