fb2b4d8a8a
chore: fix api test conventions chore: rubocop -A spec/ chore: more rubocop -A fix failing test rubocop fixes removes helper methods that are in my opinion dead code more rubocop fixes rubocop -a --auto-gen-config
37 lines
898 B
Ruby
37 lines
898 B
Ruby
# @see https://gist.github.com/virtualstaticvoid/8705533
|
|
module FindEachWithOrder
|
|
extend ActiveSupport::Concern
|
|
|
|
class_methods do
|
|
def find_each_with_order(options = {}, &block)
|
|
find_in_batches_with_order(options) do |records|
|
|
records.each(&block)
|
|
end
|
|
end
|
|
|
|
# NOTE: any limit() on the query is overridden with the batch size
|
|
def find_in_batches_with_order(options = {})
|
|
options.assert_valid_keys(:batch_size)
|
|
|
|
relation = self
|
|
|
|
start = 0
|
|
batch_size = options.delete(:batch_size) || 1000
|
|
|
|
relation = relation.limit(batch_size)
|
|
records = relation.offset(start).to_a
|
|
|
|
while records.any?
|
|
records_size = records.size
|
|
|
|
yield records
|
|
|
|
break if records_size < batch_size
|
|
|
|
# get the next batch
|
|
start += batch_size
|
|
records = relation.offset(start).to_a
|
|
end
|
|
end
|
|
end
|
|
end
|