Method getOutputFromSpilledPartition

bolt/exec/Window.cpp:986–1025

}

RowVectorPtr Window::getOutputFromSpilledPartition() {
  RowVectorPtr spilledOutput;
  bool isEnd;
  if (LIKELY(currentPartition_->getBatch(spilledOutput, isEnd))) {
    auto numOutputRows = spilledOutput->size();
    BOLT_DCHECK(numOutputRows > 0);
    std::vector<VectorPtr> children;
    // input columns
    children.insert(
        children.end(),
        spilledOutput->children().begin(),
        spilledOutput->children().end());
    // window function columns
    for (auto w = 0; w < windowResultTypes_.size(); ++w) {
      auto child = BaseVector::create(
          windowResultTypes_[w], numOutputRows, operatorCtx_->pool());
      currentPartition_->getWindowFunctionResults(
          child, numOutputRows, w, windowBuild_->isAggWindowFunc());
      children.emplace_back(std::move(child));
    }
    numProcessedRows_ += numOutputRows;
    if (isEnd) {
      windowBuild_->resetSpiller();
      callResetPartition();
    }
    return std::make_shared<RowVector>(
        operatorCtx_->pool(),
        outputType_,
        BufferPtr(nullptr),
        numOutputRows,
        std::move(children));
  } else {
    // no more batch from spilled file
    // for SpillableWindowBuild, need to set spiller_ to nullptr
    // or needsInput will not consume more data
    windowBuild_->resetSpiller();
    callResetPartition();
    return nullptr;
  }
}

bool Window::isSpillableWindowBuild() {
  static std::map<std::string, bool> supportedAggFunc{