Merge pull request #12 from junwchina/master

Works on replicate set mongodb well
huacnlee · Jun 4, 2014 · eb5b71c · eb5b71c
2 parents 5d24129 + 04e292b
commit eb5b71c
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/lib/mongoid_auto_increment_id.rb b/lib/mongoid_auto_increment_id.rb
@@ -2,11 +2,18 @@ module Mongoid
   class Identity
     # Generate auto increment id
     def self.generate_id(document)
-      o = Mongoid::Sessions.default.command({:findAndModify => "mongoid.auto_increment_ids",
-                                             :query  => { :_id => document.collection_name },
-                                             :update => { "$inc" => { :c => 1 } },
-                                             :upsert => true,
-                                             :new    => true })
+      database_name = Mongoid::Sessions.default.send(:current_database).name
+
+      o = nil
+      Mongoid::Sessions.default.cluster.with_primary do |node| 
+         o = node.command(database_name, 
+                      {"findAndModify" => "mongoid.auto_increment_ids",  
+                       :query  => { :_id => document.collection_name }, 
+                       :update => { "$inc" => { :c => 1 }}, 
+                       :upsert => true, 
+                       :new => true }, {})
+      end
+
       o["value"]["c"].to_i
     end
   end