Function BM_ReadOffsetIndex

cpp/src/parquet/page_index_benchmark.cc:37–58

}

void BM_ReadOffsetIndex(::benchmark::State& state) {
  auto builder = OffsetIndexBuilder::Make();
  const int num_pages = static_cast<int>(state.range(0));
  constexpr int64_t page_size = 1024;
  constexpr int64_t first_row_index = 10000;
  for (int i = 0; i < num_pages; ++i) {
    builder->AddPage(page_size * i, page_size, first_row_index * i);
  }
  constexpr int64_t final_position = 4096;
  builder->Finish(final_position);
  auto sink = CreateOutputStream();
  builder->WriteTo(sink.get());
  auto buffer = sink->Finish().ValueOrDie();
  ReaderProperties properties;
  for (auto _ : state) {
    auto offset_index = OffsetIndex::Make(
        buffer->data() + 0, static_cast<uint32_t>(buffer->size()), properties);
    ::benchmark::DoNotOptimize(offset_index);
  }
  state.SetBytesProcessed(state.iterations() * buffer->size());
  state.SetItemsProcessed(state.iterations() * num_pages);
}

BENCHMARK(BM_ReadOffsetIndex)->Apply(PageIndexSetArgs);