Method nextPartition

bolt/exec/TopNRowNumber.cpp:392–422

    estimatedOutputRowSize_ = rowSize;
  }
}

TopNRowNumber::TopRows* TopNRowNumber::nextPartition() {
  if (!table_) {
    if (!currentPartition_) {
      currentPartition_ = 0;
      return singlePartition_.get();
    }
    return nullptr;
  }

  if (!currentPartition_) {
    numPartitions_ = table_->listAllRows(
        &partitionIt_,
        partitions_.size(),
        RowContainer::kUnlimited,
        partitions_.data());
    if (numPartitions_ == 0) {
      // No more partitions.
      return nullptr;
    }

    currentPartition_ = 0;
  } else {
    ++currentPartition_.value();
    if (currentPartition_ >= numPartitions_) {
      currentPartition_.reset();
      return nextPartition();
    }
  }

  return &currentPartition();
}