schron 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f49a1d20dee523a612a54957311f0b2e0d6c1261
-  data.tar.gz: 1b86c1df7aaa310f1729032f2a029c3135293a88
+  metadata.gz: 4aeceef48e41835c46613cdc24244e572de9be71
+  data.tar.gz: d12b5a7d4afe3becb745bf7f0c3d5f7ed85af577
 SHA512:
-  metadata.gz: e4b7793227d44a52f21568e892a3f1525c03bc8d0f037daf4115bbeda4900c3d593e5c301a03e951fd6a50946edab32d564682dff67bc3e34dcb2874830fc938
-  data.tar.gz: 6c4a329eb00714ee3b258b88e94ea313718b7e09f879b6db011681e2bc3b9d4654df9ee0d38911cea5f9087b7c1bd0e58ab28812cf788e47570fe7d8b528d5c7
+  metadata.gz: cd30c7665c334b7765286b0bbe340efd7d6a1a4165010bbfc13c56c0d72ffa29f316a55768c55269949adba84ebf8bdc408a129b01db0d08e5f6f26287153dd1
+  data.tar.gz: 88ce344e0850b608b49215f3e67d89c39297a9ad8ca036134d271b9d617892ad571cee88782b0775d5736e38d1401ae415e70bef5c1a8bbcef68df0ce13e6343

data/lib/schron/query.rb

@@ -87,16 +87,30 @@ module Schron
     end
     
     def batched_each(size, &block)
-      batches(size) { |group| group.each(&block) }
+      enum = Enumerator.new do |y|
+        batches(size) do |group|
+          group.each do |item|
+            y << item
+          end
+        end
+      end
+      block_given? ?
+        enum.each(&block) :
+        enum
     end
 
     def batches(size, &block)
-      current_page = 1
-      begin
-        results = page(current_page, per_page: size).all
-        block.call(results)
-        current_page = results.paging.next_page
-      end while current_page
+      enum = Enumerator.new do |y|
+        current_page = 1
+        begin
+          results = page(current_page, per_page: size).all
+          y << results
+          current_page = results.paging.next_page
+        end while current_page
+      end
+      block_given? ?
+        enum.each(&block) :
+        enum      
     end
     
     protected

data/schron.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "schron"
-  spec.version       = "0.0.5"
+  spec.version       = "0.0.6"
   spec.authors       = ["David Faber"]
   spec.email         = ["david@1bios.co"]
   spec.summary       = %q{Repository implementation for entity persistence}

data/spec/lib/schron/query_spec.rb

@@ -71,6 +71,13 @@ describe Schron::Query do
         expect(group.size).to eq(1)
       end
     end
+
+    it 'returns an enum if no block is passed' do
+      2.times { archive.insert(Schron::GenericEntity.new) }
+      enum = query.batches(1)
+      expect(enum).to be_kind_of(Enumerator)
+      expect(enum.map(&:size)).to eq([1, 1])
+    end
   end
   
   describe 'batched each' do
@@ -78,10 +85,17 @@ describe Schron::Query do
       expect(ds).to receive(:exec_query).at_least(2).and_call_original
       2.times { archive.insert(Schron::GenericEntity.new) }
       i = 0
-      query.batches(1) do |item|
+      query.batched_each(1) do |item|
         i += 1
       end
       expect(i).to eq(2)
     end
+
+    it 'returns an enum if no block is passed' do
+      2.times { archive.insert(Schron::GenericEntity.new) }
+      enum = query.batched_each(1)
+      expect(enum).to be_kind_of(Enumerator)
+      expect(enum.map(&:id).sort).to eq(archive.all.map(&:id).sort)
+    end
   end
-end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: schron
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - David Faber
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-08-27 00:00:00.000000000 Z
+date: 2014-10-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake