用php在MongoDB中模拟Auto Increment

dly · 发表于 2011-3-8 17:48:28

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

melody0721 · 发表于 2011-5-9 11:59:55

不错，我喜欢

夕梨二世 · 发表于 2011-5-10 10:59:53

支持楼主，顶一下

fhiejkl · 发表于 2011-5-14 23:59:39

不错不错，我喜欢

kurt226 · 发表于 2011-5-19 21:59:58

好辛苦才找到啊

纤陌陌 · 发表于 2011-5-22 05:59:56

楼主真强大

qqoxygen · 发表于 2011-5-23 13:00:00

不错，我喜欢

丁加丁 · 发表于 2011-5-24 09:59:35

有意思~顶顶，继续顶顶。继续顶哦

lilac_yao · 发表于 2011-5-25 12:59:41

继续来索要

风中徜徉 · 发表于 2011-5-26 18:00:01

看帖必回