在MongoDB中模拟Auto Increment

MySQL用户多半都有Auto Increment情结,不过MongoDB缺省并没有实现,所以需要模拟一下,编程语言以PHP为例,代码大致如下所示:

<?php

function generate_auto_increment_id($namespace, array $option = array())
{
    $option += array(
        'init' => 1,
        'step' => 1,
    );

    $instance = new Mongo();

    $instance = $instance->selectCollection('_seq', 'seq');

    $seq = $instance->db->command(array(
        'findAndModify' => 'seq',
        'query'         => array('_id' => $namespace),
        'update'        => array('$inc' => array('id' => $option['step'])),
        'new'           => true,
    ));

    if (isset($seq['value']['id'])) {
        return $seq['value']['id'];
    }

    $instance->insert(array(
        '_id' => $namespace,
        'id'  => $option['init'],
    ));

    return $option['init'];
}

var_dump(generate_auto_increment_id('foo'));
var_dump(generate_auto_increment_id('bar', array('init' => 123)));

?>

其具体实现方式主要是利用MongoDB中findAndModify命令,只要每次往MongoDB里insert对象前生成ID赋值给_id就OK了,因为它的实现满足原子性,所以不存在并发问题。

另外说明一点,findAndModify本身提供了一个upsert参数,为true的话可以自动insert,但那样就不能自定义初始值了,所以文中示例没有使用upsert。

BTW,数据库“_seq”的名字以下划线开头,这样列表的时候会排在前面,容易分辨。

参考:Auto Increment with MongoDB

在MongoDB中模拟Auto Increment》上有9个想法

  1. Pingback引用通告: Tweets that mention 在MongoDB中模拟auto_increment | 火丁笔记 -- Topsy.com

  2. Pingback引用通告: 網站製作學習誌 » [Web] 連結分享

  3. Pingback引用通告: MongoDB 实现字段自动增长id « LAMP100

  4. Pingback引用通告: » MongoDB 实现字段自动增长id junjun的技术博客

  5. Pingback引用通告: MongoDB实现mysql主键autoincrement at 天空的颜色

发表评论

电子邮件地址不会被公开。 必填项已用*标注