Method apply

bolt/python/Function.cpp:53–85

  }

  void apply(
      const SelectivityVector& rows,
      std::vector<VectorPtr>& args,
      const TypePtr& outputType,
      exec::EvalCtx& context,
      VectorPtr& result) const override {
    pybind11::list pyArgs;
    if (mapBatch_) {
      BOLT_CHECK_EQ(defaultArgs_.size(), 0)
      std::shared_ptr<RowVector> rowVector =
          bolt::python::combineInputColumns(args, context);
      pyArgs.append(pybind11::cast(rowVector));
    } else {
      pyArgs = bolt::python::asPyArgs(nArgs_, args, defaultArgs_);
    }

    bolt::python::pyTry<void>(
        [&]() { result = function_(*pyArgs).cast<VectorPtr>(); },
        [&]() {
          return fmt::format(
              "Failed to evaluate function '{}' with "
              "arguments: '{}'",
              bolt::python::pyTypeStr(function_),
              bolt::python::toString(pyArgs));
        });
    // `result` will be allocated from the pool of the `pybolt` python
    // module. Its lifetime is the same as the one of that module.
    if (outputType->isPrimitiveType()) {
      BOLT_CHECK_EQ(result->type(), outputType);
    } else {
      BOLT_CHECK(result->type()->equivalent(*outputType))
    }
  }
};
} // namespace