Function createPartitionFunctionSpec

bolt/exec/tests/utils/PlanBuilder.cpp:1174–1204

namespace {
core::PartitionFunctionSpecPtr createPartitionFunctionSpec(
    const RowTypePtr& inputType,
    const std::vector<core::TypedExprPtr>& keys,
    memory::MemoryPool* pool) {
  if (keys.empty()) {
    return std::make_shared<core::GatherPartitionFunctionSpec>();
  } else {
    std::vector<column_index_t> keyIndices;
    keyIndices.reserve(keys.size());

    std::vector<VectorPtr> constValues;
    constValues.reserve(keys.size());

    for (const auto& key : keys) {
      if (auto field =
              std::dynamic_pointer_cast<const core::FieldAccessTypedExpr>(
                  key)) {
        keyIndices.push_back(inputType->getChildIdx(field->name()));
      } else if (
          auto constant =
              std::dynamic_pointer_cast<const core::ConstantTypedExpr>(key)) {
        keyIndices.push_back(kConstantChannel);
        constValues.push_back(constant->toConstantVector(pool));
      } else {
        BOLT_UNREACHABLE();
      }
    }
    return std::make_shared<HashPartitionFunctionSpec>(
        inputType, std::move(keyIndices), std::move(constValues));
  }
}

RowTypePtr concat(const RowTypePtr& a, const RowTypePtr& b) {
  std::vector<std::string> names = a->names();