Function BenchmarkWriteCsv

cpp/src/arrow/csv/writer_benchmark.cc:96–111

}

void BenchmarkWriteCsv(benchmark::State& state, const WriteOptions& options,
                       const RecordBatch& batch) {
  int64_t total_size = 0;

  for (auto _ : state) {
    auto out = io::BufferOutputStream::Create().ValueOrDie();
    ABORT_NOT_OK(WriteCSV(batch, options, out.get()));
    auto buffer = out->Finish().ValueOrDie();
    total_size += buffer->size();
  }

  // byte size of the generated csv dataset
  state.SetBytesProcessed(total_size);
  state.SetItemsProcessed(state.iterations() * batch.num_columns() * batch.num_rows());
  state.counters["null_percent"] = static_cast<double>(state.range(0));
}

// Exercises UnQuotedColumnPopulator with integer
void WriteCsvNumeric(benchmark::State& state) {