summaryrefslogtreecommitdiff
path: root/lib/csv_builders/stream.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/csv_builders/stream.rb')
-rw-r--r--lib/csv_builders/stream.rb17
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