Method SetLookupPrepare

be/src/exprs/in-predicate.h:366–388

template <typename T, typename SetType>
void InPredicate::SetLookupPrepare(
    FunctionContext* ctx, FunctionContext::FunctionStateScope scope) {
  if (scope != FunctionContext::FRAGMENT_LOCAL) return;

  SetLookupState<SetType>* state = new SetLookupState<SetType>;
  state->type = ctx->GetArgType(0);
  state->contains_null = false;
  // Collect all values in a vector to use the bulk insert API to avoid N^2 behavior
  // with flat_set.
  std::vector<SetType> element_list;
  element_list.reserve(ctx->GetNumArgs() - 1);
  for (int i = 1; i < ctx->GetNumArgs(); ++i) {
    DCHECK(ctx->IsArgConstant(i));
    T* arg = reinterpret_cast<T*>(ctx->GetConstantArg(i));
    if (arg->is_null) {
      state->contains_null = true;
    } else {
      element_list.push_back(GetVal<T, SetType>(state->type, *arg));
    }
  }
  state->val_set.insert(element_list.begin(), element_list.end());
  ctx->SetFunctionState(scope, state);
}

template <typename SetType>
void InPredicate::SetLookupClose(