Function AddPassFactory

cpp/src/arrow/engine/substrait/serde_test.cc:94–116

namespace engine {

Status AddPassFactory(
    const std::string& factory_name,
    acero::ExecFactoryRegistry* registry = acero::default_exec_factory_registry()) {
  using acero::ExecNode;
  using acero::ExecNodeOptions;
  using acero::ExecPlan;
  using compute::ExecBatch;
  struct PassNode : public acero::MapNode {
    static Result<ExecNode*> Make(ExecPlan* plan, std::vector<ExecNode*> inputs,
                                  const acero::ExecNodeOptions& options) {
      RETURN_NOT_OK(ValidateExecNodeInputs(plan, inputs, 1, "PassNode"));
      return plan->EmplaceNode<PassNode>(plan, inputs, inputs[0]->output_schema());
    }

    PassNode(ExecPlan* plan, std::vector<ExecNode*> inputs,
             std::shared_ptr<Schema> output_schema)
        : MapNode(plan, inputs, output_schema) {}

    const char* kind_name() const override { return "PassNode"; }
    Result<ExecBatch> ProcessBatch(ExecBatch batch) override { return batch; }
  };
  return registry->AddFactory(factory_name, PassNode::Make);
}

const auto kNullConsumer = std::make_shared<acero::NullSinkNodeConsumer>();