mongomodel 0.2.16 → 0.2.17

data/lib/mongomodel/concerns/associations/has_many_by_foreign_key.rb

@@ -111,7 +111,7 @@ module MongoModel
       
       class Proxy < Base::Proxy
         # Pass these methods to the scope rather than the Array target
-        OVERRIDE_METHODS = [ :find, :first, :last, :count ]
+        OVERRIDE_METHODS = [ :find, :first, :last, :count, :paginate ]
         
         delegate :ensure_class, :to => :association
         

data/lib/mongomodel/document/scopes.rb

@@ -13,7 +13,7 @@ module MongoModel
       module ClassMethods
         delegate :find, :first, :last, :all, :exists?, :count, :to => :scoped
         delegate :update, :update_all, :delete, :delete_all, :destroy, :destroy_all, :to => :scoped
-        delegate :select, :order, :where, :limit, :offset, :from, :paginate, :to => :scoped
+        delegate :select, :order, :where, :limit, :offset, :from, :paginate, :in_batches, :to => :scoped
         
         def unscoped
           @unscoped ||= MongoModel::Scope.new(self)

data/lib/mongomodel/support/scope.rb

@@ -10,8 +10,9 @@ module MongoModel
     autoload :FinderMethods,  'mongomodel/support/scope/finder_methods'
     autoload :DynamicFinders, 'mongomodel/support/scope/dynamic_finders'
     autoload :Pagination,     'mongomodel/support/scope/pagination'
+    autoload :Batches,        'mongomodel/support/scope/batches'
     
-    include Pagination, DynamicFinders, FinderMethods, QueryMethods, SpawnMethods
+    include Batches, Pagination, DynamicFinders, FinderMethods, QueryMethods, SpawnMethods
     
     delegate :inspect, :as_json, :to => :to_a
     

data/lib/mongomodel/support/scope/batches.rb

@@ -0,0 +1,15 @@
+module MongoModel
+  class Scope
+    module Batches
+      def in_batches(batch_size=1000)
+        offset = 0
+        
+        begin
+          documents = offset(offset).limit(batch_size).all
+          yield documents if block_given? && !documents.empty?
+          offset += batch_size
+        end until documents.size < batch_size
+      end
+    end
+  end
+end

data/lib/mongomodel/version.rb

@@ -1,3 +1,3 @@
 module MongoModel
-  VERSION = "0.2.16"
+  VERSION = "0.2.17"
 end

data/spec/mongomodel/support/scope_spec.rb

@@ -775,6 +775,20 @@ module MongoModel
           paginator.total_entries.should == 57
         end
       end
+      
+      describe "#in_batches" do
+        it "should yield documents in groups of given size" do
+          model.should_find(finder_options.merge(:offset => 0, :limit => 3), posts.first(3))
+          model.should_find(finder_options.merge(:offset => 3, :limit => 3), posts.last(2))
+          
+          expected_size = 3
+          
+          subject.in_batches(3) do |docs|
+            docs.size.should == expected_size
+            expected_size = 2
+          end
+        end
+      end
     end
     
     

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: mongomodel
 version: !ruby/object:Gem::Version 
-  hash: 55
+  hash: 53
   prerelease: 
   segments: 
   - 0
   - 2
-  - 16
-  version: 0.2.16
+  - 17
+  version: 0.2.17
 platform: ruby
 authors: 
 - Sam Pohlenz
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-04 00:00:00 +10:30
+date: 2011-01-07 00:00:00 +10:30
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -188,6 +188,7 @@ files:
 - lib/mongomodel/support/mongo_order.rb
 - lib/mongomodel/support/reference.rb
 - lib/mongomodel/support/scope.rb
+- lib/mongomodel/support/scope/batches.rb
 - lib/mongomodel/support/scope/dynamic_finders.rb
 - lib/mongomodel/support/scope/finder_methods.rb
 - lib/mongomodel/support/scope/pagination.rb