Method write_table

c_glib/test/helper/writable.rb:20–41

(table, output, type: :file)

module Helper
  module Writable
    def write_table(table, output, type: :file)
      if output.is_a?(Arrow::Buffer)
        output_stream = Arrow::BufferOutputStream.new(output)
      else
        output_stream = Arrow::FileOutputStream.new(output, false)
      end
      begin
        if type == :file
          writer_class = Arrow::RecordBatchFileWriter
        else
          writer_class = Arrow::RecordBatchStreamWriter
        end
        writer = writer_class.new(output_stream, table.schema)
        begin
          writer.write_table(table)
        ensure
          writer.close
        end
      ensure
        output_stream.close
      end
    end
  end
end