Function compileExprToJitVectorFunc

bolt/jit/expression/ExprJitCompiler.cpp:401–448

};

JitBoltVectorFunction compileExprToJitVectorFunc(const exec::Expr* expr) {
  JitBoltVectorFunction result;
  ExprConverter converter;

  try {
    auto jitExpr = converter.convertExprToJitExpr(expr);
    bool compilable = thrust::jit::compilableExpr(jitExpr.get());
    if (!compilable) {
      return result;
    }

    auto fields = expr->distinctFields();
    std::vector<thrust::jit::ColumnDataMeta> jitMetas;

    for (auto&& f : fields) {
      auto jitType = converter.convertType(f->type());
      // TODO complex type
      // if (f->inputs().size() > 0) {
      // }

      thrust::jit::ColumnDataMeta jitMeta;
      jitMeta.type = jitType->kind;
      jitMeta.nullable = true; // presto
      jitMeta.name = f->name();
      jitMetas.emplace_back(std::move(jitMeta));
    }

    auto jitResourceHolder =
        thrust::jit::compileExprForVector(jitExpr.get(), jitMetas);

    result.jitFunc = (JitBoltVectorFunction::JitedFuncAddress)
                         jitResourceHolder->getFunction();

    // std::function<void(const std::vector<VectorPtr>& args, VectorPtr*
    // result)> func;
    FunctionResourceWrapper wrapper;
    wrapper.jitHolder = jitResourceHolder;
    result.func = std::bind(
        bolt_call_function_wrapper,
        wrapper,
        std::placeholders::_1,
        std::placeholders::_2);

  } catch (std::exception& e) {
    // log
  }
  return result;
}

} // namespace bytedance::bolt::jit
#endif