Function RunBenchmark

extensions/bindings_ext_benchmark_test.cc:178–204

}

void RunBenchmark(const BenchmarkCase& benchmark, benchmark::State& state) {
  std::vector<Macro> all_macros = Macro::AllMacros();
  std::vector<Macro> bindings_macros = cel::extensions::bindings_macros();
  all_macros.insert(all_macros.end(), bindings_macros.begin(),
                    bindings_macros.end());
  ASSERT_OK_AND_ASSIGN(
      auto expr, ParseWithMacros(benchmark.expression, all_macros, "<input>"));

  InterpreterOptions options;
  auto builder =
      google::api::expr::runtime::CreateCelExpressionBuilder(options);

  ASSERT_OK(google::api::expr::runtime::RegisterBuiltinFunctions(
      builder->GetRegistry()));

  ASSERT_OK_AND_ASSIGN(auto program, builder->CreateExpression(
                                         &expr.expr(), &expr.source_info()));

  Activation activation;
  google::protobuf::Arena arena;
  for (auto _ : state) {
    auto result = program->Evaluate(activation, &arena);
    benchmark::DoNotOptimize(result);
    ABSL_DCHECK_OK(result);
    ABSL_DCHECK(benchmark.matcher.Matches(*result));
  }
}

void BM_Simple(benchmark::State& state) {
  RunBenchmark(BenchmarkCases()[0], state);