Add unlimited method to fetch all records from ES

lib/chewy/query.rb

@@ -894,8 +894,8 @@ def types! *params
     #   scope = UsersIndex.aggs(max_age: { max: { field: 'age' } }).search_type(:count)
     #   max_age = scope.aggs['max_age']['value']
     #
-    def search_type val
-      chain { options.merge!(search_type: val) }
+    def search_type value
+      chain { criteria.update_search_options search_type: value }
     end
 
     # Merges two queries.
@@ -961,6 +961,16 @@ def exists?
       search_type(:count).total > 0
     end
 
+    # Sets limit to be equal to total documents count
+    #
+    #  PlacesIndex.query(...).filter(...).unlimited
+    #
+
+    def unlimited
+      size = search_type(:count).total
+      chain { criteria.update_request_options size: Integer(size), from: 0 }
+    end
+
     # Returns request total time elapsed as reported by elasticsearch
     #
     #   UsersIndex.query(...).filter(...).took
@@ -1004,7 +1014,7 @@ def _request
       @_request ||= begin
         request = criteria.request_body
         request.merge!(index: _indexes.map(&:index_name), type: _types.map(&:type_name))
-        request.merge!(search_type: options[:search_type]) if options[:search_type]
+        request.merge!(search_type: criteria.search_options[:search_type]) if criteria.search_options[:search_type]
         request
       end
     end

lib/chewy/query/criteria.rb

@@ -5,7 +5,7 @@ class Query
     class Criteria
       include Compose
       ARRAY_STORAGES = [:queries, :filters, :post_filters, :sort, :fields, :types, :scores]
-      HASH_STORAGES = [:options, :request_options, :facets, :aggregations, :suggest, :script_fields]
+      HASH_STORAGES = [:options, :request_options, :facets, :aggregations, :suggest, :script_fields, :search_options]
       STORAGES = ARRAY_STORAGES + HASH_STORAGES
 
       def initialize options = {}
@@ -48,6 +48,10 @@ def update_request_options(modifier)
         request_options.merge!(modifier)
       end
 
+      def update_search_options(modifier)
+        search_options.merge!(modifier)
+      end
+
       def update_facets(modifier)
         facets.merge!(modifier)
       end

lib/chewy/search.rb

@@ -10,7 +10,7 @@ module Search
         :boost_factor, :weight, :random_score, :field_value_factor, :decay, :aggregations,
         :suggest, :none, :strategy, :query, :filter, :post_filter, :boost_mode,
         :score_mode, :order, :reorder, :only, :types, :delete_all, :find, :total,
-        :total_count, :total_entries, to: :all
+        :total_count, :total_entries, :unlimited, to: :all
     end
 
     module ClassMethods

spec/chewy/query_spec.rb

@@ -454,7 +454,7 @@
   end
 
   describe '#exists?' do
-    let(:data) { 10.times.map { |i| {id: i.next.to_s, name: "Name#{i.next}", age: 10 * i.next}.stringify_keys! } }
+    let(:data) { Array.new(10) { |i| {id: i.next.to_s, name: "Name#{i.next}", age: 10 * i.next}.stringify_keys! } }
 
     before { ProductsIndex::Product.import!(data.map { |h| double(h) }) }
 
@@ -464,6 +464,18 @@
     specify { expect(subject.filter(range: {age: {lt: 0}}).exists?).to eq false }
   end
 
+  describe '#unlimited' do
+    let(:data_length) { 10 }
+    let(:data) { Array.new(data_length) { |i| {id: i.next.to_s, name: "Name#{i.next}", age: 10 * i.next}.stringify_keys! } }
+
+    before { ProductsIndex::Product.import!(data.map { |h| double(h) }) }
+
+    specify { expect(subject.unlimited.count).to eq data_length }
+    specify { expect(subject.offset(5).unlimited.count).to eq data_length }
+    specify { expect(subject.limit(1).unlimited.count).to eq data_length }
+    specify { expect(subject.unlimited.limit(1).count).to eq 1 }
+  end
+
   describe '#none' do
     specify { expect(subject.none).to be_a described_class }
     specify { expect(subject.none).not_to eq(subject) }
@@ -577,7 +589,7 @@
   end
 
   describe '#search_type' do
-    specify { expect(subject.search_type(:count).options).to include(search_type: :count) }
+    specify { expect(subject.search_type(:count).criteria.search_options).to include(search_type: :count) }
   end
 
   describe '#aggregations' do