Method extractGroups

bolt/exec/GroupingSet.cpp:867–908

  NanosecondTimer timer(&stats_.aggExtractGroupsTimeNs);
  if (groups.empty()) {
    return;
  }
  // 4 bytes for rowSize
  static constexpr int32_t headerSize = sizeof(RowSizeType);
  const auto& rowInfo = rowInfo_.value();
  char *rowStart = nullptr, *newRow = nullptr;
  RowSizeType rowSize = 0, actualRowSize = 0;
  const auto& compositeResult =
      std::dynamic_pointer_cast<CompositeRowVector>(result);
  BOLT_CHECK(compositeResult);
  compositeResult->setRowOffset(sizeof(RowSizeType));

  convertRowsFromContainerRows(compositeResult, groups, resultRanges);
}

void GroupingSet::extractGroups(
    folly::Range<char**> groups,
    const RowVectorPtr& result,
    bool excludeKey) {
  NanosecondTimer timer(&stats_.aggExtractGroupsTimeNs);
  result->resize(groups.size());
  if (groups.empty()) {
    return;
  }
  RowContainer& rows = *table_->rows();
  auto totalKeys = rows.keyTypes().size();
  if (!excludeKey) {
    for (int32_t i = 0; i < totalKeys; ++i) {
      auto keyVector = result->childAt(i);
      rows.extractColumn(groups.data(), groups.size(), i, keyVector);
    }
  }
  for (int32_t i = 0; i < aggregates_.size(); ++i) {
    if (!aggregates_[i].sortingKeys.empty()) {
      continue;
    }

    auto& function = aggregates_[i].function;
    auto& aggregateVector = result->childAt(i + totalKeys);
    if (isPartial_) {
      function->extractAccumulators(
          groups.data(), groups.size(), &aggregateVector);
    } else {
      function->extractValues(groups.data(), groups.size(), &aggregateVector);
    }