Method Merge

bolt/exec/Merge.cpp:40–78

namespace bytedance::bolt::exec {

Merge::Merge(
    int32_t operatorId,
    DriverCtx* driverCtx,
    RowTypePtr outputType,
    const std::vector<std::shared_ptr<const core::FieldAccessTypedExpr>>&
        sortingKeys,
    const std::vector<core::SortOrder>& sortingOrders,
    const std::string& planNodeId,
    const std::string& operatorType,
    const std::optional<common::SpillConfig>& spillConfig)
    : SourceOperator(
          driverCtx,
          std::move(outputType),
          operatorId,
          planNodeId,
          operatorType,
          spillConfig),
      maxOutputBatchRows_{outputBatchRows()},
      maxOutputBatchBytes_{
          driverCtx->queryConfig().preferredOutputBatchBytes()},
      sortingKeys_([&]() {
        auto numKeys = sortingKeys.size();
        std::vector<SpillSortKey> keys;
        keys.reserve(numKeys);
        for (int i = 0; i < numKeys; ++i) {
          auto channel = exprToChannel(sortingKeys[i].get(), outputType_);
          BOLT_CHECK_NE(
              channel,
              kConstantChannel,
              "Merge doesn't allow constant grouping keys");
          keys.emplace_back(
              channel,
              CompareFlags{
                  sortingOrders[i].isNullsFirst(),
                  sortingOrders[i].isAscending(),
                  false});
        }
        return keys;
      }()) {}

void Merge::initialize() {
  Operator::initialize();