在MongoDB中模拟Auto Increment的php代码
程序员文章站
2022-04-09 10:10:22
代码大致如下所示: 复制代码 代码如下:
代码大致如下所示:
<?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)));
?>
复制代码 代码如下:
<?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”的名字以下划线开头,这样列表的时候会排在前面,更容易分辨些。
推荐阅读