Add batch_size(batch_size) to __find_in_batches (Mongoid)

elastic · Jul 11, 2022 · ef8985e · ef8985e
1 parent b66fe71
commit ef8985e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/elasticsearch-model/lib/elasticsearch/model/adapters/mongoid.rb b/elasticsearch-model/lib/elasticsearch/model/adapters/mongoid.rb
@@ -89,7 +89,7 @@ def __find_in_batches(options={}, &block)
             scope = scope.send(named_scope) if named_scope
             scope = query.is_a?(Proc) ? scope.class_exec(&query) : scope.where(query) if query
 
-            scope.no_timeout.each_slice(batch_size) do |items|
+            scope.no_timeout.batch_size(batch_size).each_slice(batch_size) do |items|
               yield (preprocess ? self.__send__(preprocess, items) : items)
             end
           end