Add latest changes from gitlab-org/gitlab@13-6-stable-eev13.6.0-rc42

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