Method sortPartitions

bolt/exec/WindowBuild.cpp:222–409

}

void WindowBuild::sortPartitions() {
  sortRows_.resize(numRows_);
  RowContainerIterator iter;
  data_->listRows(&iter, numRows_, sortRows_.data());

#ifdef ENABLE_BOLT_JIT
  if (cmp_ == nullptr && enableJit_) {
    if (data_->JITable(data_->keyTypes())) {
      auto [jitMod, rowRowCmpfn] = data_->codegenCompare(
          data_->keyTypes(),
          sortSpillCompareFlags_,
          bytedance::bolt::jit::CmpType::SORT_LESS,
          true);
      jitModule_ = std::move(jitMod);
      cmp_ = (RowRowCompare)jitModule_->getFuncPtr(rowRowCmpfn);
    }
  }
  if (cmp_) {
    sorter_.sort(sortRows_.begin(), sortRows_.end(), cmp_);
  } else {
#endif

#ifdef ENABLE_META_SORT
    MetaRowsSorterWraper<BufferRows>::MetaCodegenSort(
        sortedRows_,
        data_.get(),
        sorter_,
        data_->keyIndices(),
        sortCompareFlags_);
#else
  sorter_.sort(
      sortRows_.begin(),
      sortRows_.end(),
      [this](const char* leftRow, const char* rightRow) {
        for (vector_size_t index = 0; index < sortSpillCompareFlags_.size();
             ++index) {
          if (auto result = data_->compare(
                  leftRow, rightRow, index, sortSpillCompareFlags_[index])) {
            return result < 0;
          }
        }
        return false;
      });
#endif

#ifdef ENABLE_BOLT_JIT
  }
#endif
}

void WindowBuild::ensureInputFits(const RowVectorPtr& input) {
  if (spillConfig_ == nullptr) {
    return;
  }

  const int64_t numRows = data_->numRows();
  if (numRows == 0) {
    // 'data_' is empty. Nothing to spill.