diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-11-19 11:27:35 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-11-19 11:27:35 +0300 |
commit | 7e9c479f7de77702622631cff2628a9c8dcbc627 (patch) | |
tree | c8f718a08e110ad7e1894510980d2155a6549197 /lib/csv_builders | |
parent | e852b0ae16db4052c1c567d9efa4facc81146e88 (diff) |
Add latest changes from gitlab-org/gitlab@13-6-stable-eev13.6.0-rc42
Diffstat (limited to 'lib/csv_builders')
-rw-r--r-- | lib/csv_builders/stream.rb | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/lib/csv_builders/stream.rb b/lib/csv_builders/stream.rb new file mode 100644 index 00000000000..a2b9fca84cb --- /dev/null +++ b/lib/csv_builders/stream.rb @@ -0,0 +1,17 @@ +# frozen_string_literal: true + +module CsvBuilders + class Stream < CsvBuilder + def render(max_rows = 100_000) + max_rows_including_header = max_rows + 1 + + Enumerator.new do |csv| + csv << CSV.generate_line(headers) + + each do |object| + csv << CSV.generate_line(row(object)) + end + end.lazy.take(max_rows_including_header) # rubocop: disable CodeReuse/ActiveRecord + end + end +end |