Method CodegenInsert

be/src/exec/filter-context.cc:381–599

An example of the generated code the following query: RF001[min_max] -> a.smallint_col set minmax_filter_threshold=1.0; set minmax_filtering_level=PAGE; select straight_join count(a.id) from alltypes a join [SHUFFLE] alltypes b on a.smallint_col = b.bigint_col; ; Function Attrs: noinline nounwind define internal fastcc void @InsertRuntimeFilters(%"struct.impala::FilterContext" nocapture readonl

//   ret void
// }
Status FilterContext::CodegenInsert(LlvmCodeGen* codegen, ScalarExpr* filter_expr,
    const TRuntimeFilterDesc& filter_desc, llvm::Function** fn) {
  llvm::LLVMContext& context = codegen->context();
  LlvmBuilder builder(context);

  *fn = nullptr;
  llvm::PointerType* this_type = codegen->GetStructPtrType<FilterContext>();
  llvm::PointerType* tuple_row_ptr_type = codegen->GetStructPtrType<TupleRow>();
  LlvmCodeGen::FnPrototype prototype(
      codegen, "FilterContextInsert", codegen->void_type());
  prototype.AddArgument(LlvmCodeGen::NamedVariable("this", this_type));
  prototype.AddArgument(LlvmCodeGen::NamedVariable("row", tuple_row_ptr_type));

  llvm::Value* args[2];
  llvm::Function* insert_filter_fn = prototype.GeneratePrototype(&builder, args);
  llvm::Value* this_arg = args[0];
  llvm::Value* row_arg = args[1];

  llvm::Value* local_filter_arg;
  // The function for inserting into the in-list filter.
  llvm::Function* insert_in_list_filter_fn = nullptr;
  if (filter_desc.type == TRuntimeFilterType::BLOOM) {
    // Load 'local_bloom_filter' from 'this_arg' FilterContext object.
    llvm::Value* local_bloom_filter_ptr =
        builder.CreateStructGEP(nullptr, this_arg, 3, "local_bloom_filter_ptr");
    local_filter_arg =
        builder.CreateLoad(local_bloom_filter_ptr, "local_bloom_filter_arg");
  } else if (filter_desc.type == TRuntimeFilterType::MIN_MAX) {
    // Load 'local_min_max_filter' from 'this_arg' FilterContext object.
    llvm::Value* local_min_max_filter_ptr =
        builder.CreateStructGEP(nullptr, this_arg, 4, "local_min_max_filter_ptr");
    llvm::PointerType* min_max_filter_type =
        codegen->GetNamedPtrType(MinMaxFilter::GetLlvmClassName(
        filter_expr->type().type))->getPointerTo();
    local_min_max_filter_ptr = builder.CreatePointerCast(
        local_min_max_filter_ptr, min_max_filter_type, "cast_min_max_filter_ptr");
    local_filter_arg =
        builder.CreateLoad(local_min_max_filter_ptr, "local_min_max_filter_arg");
  } else {
    DCHECK(filter_desc.type == TRuntimeFilterType::IN_LIST);
    // Load 'local_in_list_filter' from 'this_arg' FilterContext object.
    llvm::Value* local_in_list_filter_ptr =
        builder.CreateStructGEP(nullptr, this_arg, 5, "local_in_list_filter_ptr");
    switch (filter_expr->type().type) {
      case TYPE_TINYINT:
        insert_in_list_filter_fn = codegen->GetFunction(
            IRFunction::TINYINT_IN_LIST_FILTER_INSERT, false);
        break;
      case TYPE_SMALLINT:
        insert_in_list_filter_fn = codegen->GetFunction(
            IRFunction::SMALLINT_IN_LIST_FILTER_INSERT, false);
        break;
      case TYPE_INT:
        insert_in_list_filter_fn = codegen->GetFunction(
            IRFunction::INT_IN_LIST_FILTER_INSERT, false);
        break;
      case TYPE_BIGINT:
        insert_in_list_filter_fn = codegen->GetFunction(