diff options
Diffstat (limited to 'lib/csv_builder.rb')
-rw-r--r-- | lib/csv_builder.rb | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/lib/csv_builder.rb b/lib/csv_builder.rb index 7df4e3bf85d..a9ef5a83ae8 100644 --- a/lib/csv_builder.rb +++ b/lib/csv_builder.rb @@ -14,6 +14,9 @@ # CsvBuilder.new(@posts, columns).render # class CsvBuilder + DEFAULT_ORDER_BY = 'id'.freeze + DEFAULT_BATCH_SIZE = 1000 + attr_reader :rows_written # @@ -68,6 +71,12 @@ class CsvBuilder } end + protected + + def each(&block) + @collection.find_each(&block) # rubocop: disable CodeReuse/ActiveRecord + end + private def headers @@ -91,7 +100,7 @@ class CsvBuilder def write_csv(csv, until_condition:) csv << headers - @collection.find_each do |object| + each do |object| csv << row(object) @rows_written += 1 |