Method outputUniqueGroups

bolt/exec/GroupingSet.cpp:1355–1402

  SelectivityVector selections(groupOfRows.size());
  auto& accumulators = container->accumulators();
  BOLT_CHECK_EQ(accumulators.size(), aggregates_.size());
  for (auto i = 0; i < aggregates_.size(); ++i) {
    // extract accumulate
    std::vector<VectorPtr> args{BaseVector::create(
        accumulators[i].spillType(), groupOfRows.size(), &pool_)};
    accumulators[i].extractForSpill(
        folly::Range<char**>(rows.data(), groupOfRows.size()), args[0]);
    // update accumulate to groups
    aggregates_[i].function->addIntermediateResults(
        groupOfRows.data(), selections, args, false);
  }
  rows.clear();
  groupOfRows.clear();
}

void GroupingSet::outputUniqueGroups(
    std::vector<char*>& uniqueRows,
    const RowVectorPtr& uniqueRes,
    size_t& uniqueCount,
    size_t& estimateUniqueBytesPerRow) {
  if (!uniqueRows.empty()) {
    NanosecondTimer aggTimer(&stats_.aggOutputUpdateTimeNs);
    if (uniqueRows.size() + uniqueCount > uniqueRes->size()) {
      uniqueRes->resize(uniqueRows.size() + uniqueCount);
    }
    auto* container = spiller_->container();
    for (auto i = 0; i < keyChannels_.size(); ++i) {
      rowToColumnVector(
          uniqueRows.data(),
          uniqueRows.size(),
          container->columns().at(i),
          uniqueCount,
          uniqueRes->childAt(i));
    }

    auto& accumulators = container->accumulators();
    BOLT_CHECK_EQ(accumulators.size(), aggregates_.size());
    for (auto i = 0; i < aggregates_.size(); ++i) {
      // extract accumulate
      auto aggCol = uniqueRes->childAt(i + keyChannels_.size());
      VectorPtr args = BaseVector::create(
          isPartial_ ? accumulators[i].spillType()
                     : accumulators[i].finalOutputType(),
          uniqueRows.size(),
          &pool_);
      BOLT_CHECK(aggCol->type()->equivalent(*args->type()));
      isPartial_
          ? accumulators[i].extractForSpill(
                folly::Range<char**>(uniqueRows.data(), uniqueRows.size()),
                args)
          : accumulators[i].extractForOutput(