在MongoDB中模拟Auto Increment

日期： 2011-04-19 作者：火丁笔记 来源：TechTarget中国 英文

　　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”的名字以下划线开头，这样列表的时候会排在前面，更容易分辨些。

我们一直都在努力坚持原创.......请不要一声不吭，就悄悄拿走。

我原创，你原创，我们的内容世界才会更加精彩！

【所有原创内容版权均属TechTarget，欢迎大家转发分享。但未经授权，严禁任何媒体（平面媒体、网络媒体、自媒体等）以及微信公众号复制、转载、摘编或以其他方式进行使用。】

微信公众号

TechTarget微信公众号二维码

TechTarget

官方微博

TechTarget中国官方微博二维码

TechTarget中国

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

敬请读者发表评论，本站保留删除与本文无关和不雅评论的权力。

作者

火丁笔记
火丁笔记

相关推荐