在MongoDB中模拟Auto Increment的php代码
author:一佰互联 2019-04-30   click:146
代码大致如下所示:
复制代码 代码如下:
<?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