1. 程式人生 > >Mongodb 自動增長 自增id 實現

Mongodb 自動增長 自增id 實現

首先建立一個自動增長id集合 ids

>db.ids.save({name:"user", id:0});

可以檢視一下是否成功

> db.ids.find();
{ "_id" : ObjectId("4c637dbd900f00000000686c"), "name" : "user", "id" : 0 }

然後每次新增新使用者之前自增一下 ids集合 獲得id

然後每次在db.user集合裡新增新使用者之前 在db.ids集合中的name="user"文件的id值加1返回文件

>userid = db.ids  .findAndModify({update:{$inc:{'id':1}}, query:{"name":"user"}, new:true});
{ "_id" : ObjectId("4c637dbd900f00000000686c"), "name" : "user", "id" : 1 }

注:因為findAndModify是一個方法完成更新查詢兩個操作,所以具有原子性,多執行緒不會衝突。

然後儲存相應的資料

>db.user  .save({uid:userid.id, username:"dotcoo", password:"dotcoo", info:"http://www.dotcoo.com/  "});

// --------- update 2011-03-27 13:11 ------------------------

其實上邊兩行可以寫為一步

>db.user  .save({
    uid: db.ids  .findAndModify({
        update:{$inc:{'id':1}},
        query:{"name":"user"},
        new:true}).id,  //讓db.ids集合中的name="user"文件的id值加1並返回充當自增id
    username: "dotcoo",
    password:"dotcoo",
    info:"http://www.dotcoo.com/   "});

// --------- update 2011-03-27 13:11 ------------------------

檢視結果

> db.user.find();
{ "_id" : ObjectId("4c637f79900f00000000686d"), "uid" : 1, "username" : "admin", "password" : "admin" }

這是mongo的shell,如果用的是伺服器端程式java php python,可以自己對這些操作封裝一下,只用傳幾個引數就可以返回自增的id,還可以實現像Oracle的跨表的自增id。

自己寫了一段php的,拿出來給大家分享。

<?php
function mid($name, $db){
    $update = array('$inc'=>array("id"=>1));
    $query = array('name'=>$name);
    $command = array(
            'findandmodify'=>'ids', 'update'=>$update,
            'query'=>$query, 'new'=>true, 'upsert'=>true
    );
    $id = $db->command($command);
    return $id['value']['id'];
}
  
$conn = new Mongo();
$db = $conn->idtest;
$id = mid('user', $db);
$db->user->save(array('uid'=>$id, 'username'=>'kekeles', 'password'=>'kekeles', 'info'=>'http://www.dotcoo.com/  '));
$conn->close();
?>