mongomapper_plugins 0.0.4 → 0.0.5

data/README.rdoc

@@ -27,3 +27,19 @@ This plugin will update the internal value of a model when using modifiers like
    counter => Counter.create(:count => 1)
    counter.increment(:count, 1)
    counter.count #=> 2
+
+== Auto increment ids
+
+This plugin will allow the auto generation of numeric ids
+Uses the "Insert if Not Present" technique (see Mongo documentation[http://www.mongodb.org/display/DOCS/Atomic+Operations#AtomicOperations-%22InsertifNotPresent%22])
+
+====Usage
+
+   class Model
+     include MongoMapper::Document
+     auto_increment_id
+   end
+
+   #Assuming the first document in the collection
+   Model.create.id #=> 1
+   Model.create.id #=> 2 etc

data/lib/mongo_mapper/plugins/auto_increment_id.rb

@@ -0,0 +1,36 @@
+module MongoMapper
+  module Plugins
+    module AutoIncrementId
+      module ClassMethods
+        def auto_increment_id
+          key :_id, Integer
+
+          after_create :generate_document_id
+        end
+      end
+
+      module InstanceMethods
+        private
+          def generate_document_id(options={})
+            while true
+              oldest_number = self.class.fields(:_id).sort([[:_id, :descending]]).first.try(:id)
+              self.id = oldest_number.to_i + 1
+              begin
+                break if collection.insert({:_id => id}, :safe => true)
+              rescue Mongo::OperationFailure => e
+                #Ignored, trying to get the next key
+              end
+            end
+          end
+      end
+
+      module Addition
+        def self.included(model)
+          model.plugin AutoIncrementId
+        end
+      end
+    end
+  end
+end
+
+MongoMapper::Document.append_inclusions MongoMapper::Plugins::AutoIncrementId::Addition

data/lib/mongomapper_plugins/railtie.rb

@@ -1,2 +1,3 @@
 require 'mongo_mapper/plugins/versioned_update'
 require 'mongo_mapper/plugins/updating_modifiers'
+require 'mongo_mapper/plugins/auto_increment_id'

metadata

@@ -2,7 +2,7 @@
 name: mongomapper_plugins
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors: 
 - Andrew Timberlake
@@ -35,6 +35,7 @@ extra_rdoc_files:
 - README.rdoc
 files: 
 - LICENSE
+- lib/mongo_mapper/plugins/auto_increment_id.rb
 - lib/mongo_mapper/plugins/updating_modifiers.rb
 - lib/mongo_mapper/plugins/versioned_update.rb
 - lib/mongomapper_plugins/railtie.rb