diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-05-20 17:34:42 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-05-20 17:34:42 +0300 |
commit | 9f46488805e86b1bc341ea1620b866016c2ce5ed (patch) | |
tree | f9748c7e287041e37d6da49e0a29c9511dc34768 /lib/csv_builder.rb | |
parent | dfc92d081ea0332d69c8aca2f0e745cb48ae5e6d (diff) |
Add latest changes from gitlab-org/gitlab@13-0-stable-ee
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 |