Function GetFilteredFragments

cpp/src/arrow/dataset/file_benchmark.cc:72–87

}

static void GetFilteredFragments(benchmark::State& state, compute::Expression filter) {
  auto dataset = GetDataset();
  ASSERT_OK_AND_ASSIGN(filter, filter.Bind(*dataset.dataset->schema()));
  int64_t num_filtered_fragments = 0;
  for (auto _ : state) {
    num_filtered_fragments = 0;
    ASSERT_OK_AND_ASSIGN(auto fragments, dataset.dataset->GetFragments(filter));
    ABORT_NOT_OK(fragments.Visit([&](std::shared_ptr<Fragment>) {
      ++num_filtered_fragments;
      return Status::OK();
    }));
  }
  state.SetItemsProcessed(state.iterations() * dataset.num_fragments);
  state.counters["num_fragments"] = static_cast<double>(dataset.num_fragments);
  state.counters["num_filtered_fragments"] = static_cast<double>(num_filtered_fragments);
}

using compute::field_ref;
using compute::literal;