Method copyKeyAndInitGroup

bolt/exec/GroupingSet.cpp:1266–1319

  if (spillConfig_->rowBasedSpillMode == common::RowBasedSpillMode::DISABLE) {
    if (isDistinct()) {
      return mergeNextWithoutAggregates(maxOutputRows, result);
    } else {
      return mergeNextWithAggregates(maxOutputRows, maxOutputBytes, result);
    }
  } else {
    if (supportRowBasedOutput_) {
      bool ret = mergeNextWithRowBasedSpillInRowFormat(
          maxOutputRows, maxOutputBytes, result);
      return ret;
    } else {
      return mergeNextWithRowBasedSpill(maxOutputRows, maxOutputBytes, result);
    }
  }
}

void GroupingSet::copyKeyAndInitGroup(
    std::vector<char*>& distinctRows,
    std::vector<char*>& groups,
    size_t& initGroupCount,
    const RowContainer* container,
    const RowVectorPtr& result,
    const vector_size_t resultOffset) {
  NanosecondTimer aggTimer(&stats_.aggOutputUpdateTimeNs);
  auto numNewGroups = distinctRows.size() - initGroupCount;
  // has new groups to handle
  if (numNewGroups) {
    // for RowBased output, do not resize!!!
    if (!supportRowBasedOutput_) {
      result->resize(distinctRows.size());
    }
    // key direct copy into result vector
    for (auto i = 0; i < keyChannels_.size(); ++i) {
      rowToColumnVector(
          distinctRows.data() + initGroupCount,
          distinctRows.size() - initGroupCount,
          container->columns().at(i),
          resultOffset,
          result->childAt(i));
    }
    if (!isDistinct()) {
      BOLT_CHECK_EQ(distinctRows.size(), groups.size());
      // init from initGroupCount to groups.size()
      ensureGroupIndices(groups.size());

      if (supportRowBasedOutput_) {
        // need to store keys in columnar format
        SelectivityVector selectivity(result->size(), true);

        for (auto row = 0; row < numNewGroups; ++row) {
          for (auto i = 0; i < keyChannels_.size(); ++i) {
            DecodedVector decoded(*result->childAt(i), selectivity);
            char* group = groups[groupIndicesOfRows_[initGroupCount + row]];
            mergeRows_->store(decoded, resultOffset + row, group, i);
          }
        }
      }