class AthenaCsv::Output

Attributes

file_path[R]
query[R]

Public Class Methods

new(query, file_path) click to toggle source
# File lib/athena_csv/output.rb, line 7
def initialize(query, file_path)
  @query = query
  @file_path = file_path
end

Public Instance Methods

generate_csv() click to toggle source
# File lib/athena_csv/output.rb, line 20
def generate_csv
  CSV.open(file_path, "wb") do |csv|
    query_results.each do |row|
      csv << values(row)
    end
  end
end
query_results() click to toggle source
# File lib/athena_csv/output.rb, line 12
def query_results
  @query_result_rows ||= Client.new.run(query)
end
values(row) click to toggle source
# File lib/athena_csv/output.rb, line 16
def values(row)
      row.data.map {|cell| cell.var_char_value}
end