MySQL使用者多半都有Auto Increment情結,不過MongoDB預設並沒有實現,所以需要類比一下,程式設計語言以PHP為例,代碼大致如下所示:
<?phpfunction 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
mongoDB 資料檔案超過2G後,寫入就超級慢了。解決方案:shard 分區