Method createWindowFunctionsForSpillableWindowBuild

bolt/exec/Window.cpp:372–407

}

void Window::createWindowFunctionsForSpillableWindowBuild() {
  const auto& inputType = windowNode_->sources()[0]->outputType();
  std::vector<std::unique_ptr<exec::WindowFunction>> windowFunctions;
  for (const auto& windowNodeFunction : windowNode_->windowFunctions()) {
    std::vector<WindowFunctionArg> functionArgs;
    functionArgs.reserve(windowNodeFunction.functionCall->inputs().size());
    for (auto& arg : windowNodeFunction.functionCall->inputs()) {
      auto channel = exprToChannel(arg.get(), inputType);
      if (channel == kConstantChannel) {
        auto constantArg = core::TypedExprs::asConstant(arg);
        functionArgs.push_back(
            {arg->type(),
             constantArg->toConstantVector(memory::spillMemoryPool()),
             std::nullopt});
      } else {
        functionArgs.push_back({arg->type(), nullptr, channel});
      }
    }

    windowFunctions.push_back(WindowFunction::create(
        windowNodeFunction.functionCall->name(),
        functionArgs,
        windowNodeFunction.functionCall->type(),
        windowNodeFunction.ignoreNulls,
        memory::spillMemoryPool(),
        &stringAllocator_,
        operatorCtx_->driverCtx()->queryConfig()));
  }
  if (needSort_) {
    dynamic_cast<SpillableWindowBuild<true>*>(windowBuild_.get())
        ->setWindowFunction(std::move(windowFunctions));
  } else {
    dynamic_cast<SpillableWindowBuild<false>*>(windowBuild_.get())
        ->setWindowFunction(std::move(windowFunctions));
  }
}

// Support 'rank' and
// 'row_number' functions and the agg window function with default frame.