Function RunBenchmark

extensions/sets_functions_benchmark_test.cc:204–242

}

void RunBenchmark(const TestCase& test_case, benchmark::State& state) {
  bool lists_overlap = test_case.result.BoolOrDie();

  std::string expr = test_case.expr;
  if (test_case.list_impl == ListImpl::kRhsConstant) {
    expr = absl::StrReplaceAll(
        expr, {{"y", ConstantList(lists_overlap, test_case.size)}});
  }
  ASSERT_OK_AND_ASSIGN(ParsedExpr parsed_expr, Parse(expr));

  google::protobuf::Arena arena;

  InterpreterOptions options;
  options.constant_folding = true;
  options.constant_arena = &arena;
  options.enable_qualified_identifier_rewrites = true;
  auto builder = CreateCelExpressionBuilder(options);
  ASSERT_OK(RegisterBuiltinFunctions(builder->GetRegistry(), options));
  ASSERT_OK(RegisterSetsFunctions(builder->GetRegistry()->InternalGetRegistry(),
                                  cel::RuntimeOptions{}));
  ASSERT_OK_AND_ASSIGN(
      auto cel_expr, builder->CreateExpression(&(parsed_expr.expr()), nullptr));

  Activation activation;
  ASSERT_OK_AND_ASSIGN(
      auto storage,
      RegisterLists(test_case.result.BoolOrDie(), test_case.size,
                    test_case.list_impl == ListImpl::kWrappedModern, &arena,
                    activation));

  state.SetLabel(test_case.MakeLabel(test_case.size));
  for (auto _ : state) {
    ASSERT_OK_AND_ASSIGN(CelValue result,
                         cel_expr->Evaluate(activation, &arena));
    ASSERT_TRUE(result.IsBool());
    ASSERT_EQ(result.BoolOrDie(), test_case.result.BoolOrDie())
        << test_case.test_name;
  }
}

void BM_SetsIntersectsTrue(benchmark::State& state) {
  ListImpl impl = FromNumber(state.range(0));