Add batch_size(batch_size) to __find_in_batches (Mongoid)

sylvain-8422 · sylvain-8422 · commit ef8985e97575 · 2022-07-11T10:03:20.000-04:00
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
