Function BenchmarkSegmentedAggregate

cpp/src/arrow/acero/aggregate_benchmark.cc:895–910

//

static void BenchmarkSegmentedAggregate(
    benchmark::State& state, int64_t num_rows, std::vector<Aggregate> aggregates,
    const std::vector<std::shared_ptr<Array>>& arguments,
    const std::vector<std::shared_ptr<Array>>& keys, int64_t num_segment_keys,
    int64_t num_segments) {
  ASSERT_GT(num_segments, 0);

  auto rng = random::RandomArrayGenerator(42);
  auto segment_key = rng.Int64(num_rows, /*min=*/0, /*max=*/num_segments - 1);
  int64_t* values = segment_key->data()->GetMutableValues<int64_t>(1);
  std::sort(values, values + num_rows);
  // num_segment_keys copies of the segment key.
  ArrayVector segment_keys(num_segment_keys, segment_key);

  BenchmarkAggregate(state, std::move(aggregates), arguments, keys, segment_keys);
}

static void CountScalarSegmentedByInts(benchmark::State& state) {
  constexpr int64_t num_rows = 32 * 1024;