Method WriteFile

cpp/src/parquet/metadata_benchmark.cc:67–93

  }

  std::shared_ptr<Buffer> WriteFile(benchmark::State* state) {
    PARQUET_ASSIGN_OR_THROW(auto sink, BufferOutputStream::Create());

    auto writer = ParquetFileWriter::Open(sink, schema_root_, writer_properties_);
    std::vector<int32_t> int32_values(1, 42);
    int64_t data_size = 0;
    for (int rg = 0; rg < num_row_groups_; ++rg) {
      auto row_group_writer = writer->AppendRowGroup();
      for (int col = 0; col < num_columns_; ++col) {
        auto col_writer = row_group_writer->NextColumn();
        ARROW_CHECK_EQ(col_writer->type(), Type::INT32);
        auto typed_col_writer = static_cast<Int32Writer*>(col_writer);
        typed_col_writer->WriteBatch(
            /*num_values=*/static_cast<int64_t>(int32_values.size()),
            /*def_levels=*/nullptr, /*rep_levels=*/nullptr, int32_values.data());
        typed_col_writer->Close();
      }
      row_group_writer->Close();
      data_size += row_group_writer->total_compressed_bytes_written();
    }
    writer->Close();
    PARQUET_ASSIGN_OR_THROW(auto buf, sink->Finish());
    state->counters["file_size"] = static_cast<double>(buf->size());
    // Note that "data_size" includes the Thrift page headers
    state->counters["data_size"] = static_cast<double>(data_size);
    return buf;
  }

  void ReadFile(std::shared_ptr<Buffer> contents) {
    auto source = std::make_shared<BufferReader>(contents);