jchris-couchrest 0.8.1 → 0.8.2

data/lib/pager.rb

@@ -16,6 +16,13 @@ module Enumerable
       grouped
     end
   end unless [].respond_to?(:group_by)
+  
+  def group_by_fast
+    inject({}) do |grouped, element|
+      (grouped[yield(element)] ||= []) << element
+      grouped
+    end
+  end
 end
 
 class CouchRest
@@ -25,6 +32,21 @@ class CouchRest
       @db = db
     end
     
+    def all_docs(count=100, &block)
+      startkey = nil
+      oldend = nil
+      
+      while docrows = request_all_docs(count+1, startkey)        
+        startkey = docrows.last['key']
+        docrows.pop if docrows.length > count
+        if oldend == startkey
+          break
+        end
+        yield(docrows)
+        oldend = startkey
+      end
+    end
+    
     def key_reduce(view, count, firstkey = nil, lastkey = nil, &block)
       # start with no keys
       startkey = firstkey
@@ -74,6 +96,17 @@ class CouchRest
       end
     end
 
+    private
+    
+    def request_all_docs count, startkey = nil
+      opts = {}
+      opts[:count] = count if count
+      opts[:startkey] = startkey if startkey      
+      results = @db.documents(opts)
+      rows = results['rows']
+      rows unless rows.length == 0
+    end
+
     def request_view view, count = nil, startkey = nil, endkey = nil
       opts = {}
       opts[:count] = count if count

data/spec/database_spec.rb

@@ -4,14 +4,12 @@ describe CouchRest::Database do
   before(:each) do
     @cr = CouchRest.new(COUCHHOST)
     begin
-      @db = @cr.create_db(TESTDB)
+      @db = @cr.database(TESTDB)
+      @db.delete!
     rescue RestClient::Request::RequestFailed
     end
-  end
-  
-  after(:each) do
     begin
-      @db.delete!
+      @db = @cr.create_db(TESTDB)
     rescue RestClient::Request::RequestFailed
     end
   end

data/spec/pager_spec.rb

@@ -25,6 +25,39 @@ describe CouchRest::Pager do
     @pager.db.should == @db
   end
   
+  describe "paging all docs" do
+    before(:all) do
+      @docs = []
+      100.times do |i|
+        @docs << ({:number => (i % 10)})
+      end
+      @db.bulk_save(@docs)
+    end
+    it "should yield total_docs / count times" do
+      n = 0
+      @pager.all_docs(10) do |doc|
+        n += 1
+      end
+      n.should == 10
+    end
+    it "should yield each docrow group without duplicate docs" do
+      docids = {}
+      @pager.all_docs(10) do |docrows|
+        docrows.each do |row|
+          docids[row['id']].should be_nil
+          docids[row['id']] = true
+        end
+      end      
+      docids.keys.length.should == 100
+    end
+    it "should yield each docrow group" do
+      @pager.all_docs(10) do |docrows|
+        doc = @db.get(docrows[0]['id'])
+        doc['number'].class.should == Fixnum
+      end      
+    end
+  end
+  
   describe "Pager with a view and docs" do
     before(:all) do
       @docs = []

data/spec/spec_helper.rb

@@ -1,4 +1,4 @@
 require File.dirname(__FILE__) + '/../lib/couchrest'
 
-COUCHHOST = "http://localhost:5985"
+COUCHHOST = "http://localhost:5984"
 TESTDB = 'couchrest-test'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jchris-couchrest
 version: !ruby/object:Gem::Version 
-  version: 0.8.1
+  version: 0.8.2
 platform: ruby
 authors: 
 - J. Chris Anderson