summaryrefslogtreecommitdiff
path: root/lib/csv_builders/stream.rb
blob: a2b9fca84cb4bed3450edef503397ae5917116fd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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