Pluck in batches

pluck_in_batches.rb

module MorePluckable
  extend ActiveSupport::Concern

  class_methods do
    def pluck_in_batches(*column_names, batch_size: 1000)
      column_names = column_names.map(&:to_sym)
      column_names.delete(:id)
      column_names.unshift(:id)

      data = []
      start = 0

      loop do
        new_data = reorder(:id).where('id > ?', start).limit(batch_size).pluck(*column_names)
        if block_given?
          yield new_data
        else
          data.concat(new_data)
        end

        break if new_data.count < batch_size
        start = new_data.last.first
      end

      data unless block_given?
    end
  end
end

ApplicationRecord.send(:include, MorePluckable)