Method extractForSpill

bolt/exec/SortedAggregations.cpp:187–220

}

void SortedAggregations::extractForSpill(
    folly::Range<char**> groups,
    VectorPtr& result) const {
  auto* arrayVector = result->as<ArrayVector>();
  arrayVector->resize(groups.size());

  auto* rawOffsets =
      arrayVector->mutableOffsets(groups.size())->asMutable<vector_size_t>();
  auto* rawSizes =
      arrayVector->mutableSizes(groups.size())->asMutable<vector_size_t>();

  vector_size_t offset = 0;
  for (auto i = 0; i < groups.size(); ++i) {
    auto* accumulator = reinterpret_cast<RowPointers*>(groups[i] + offset_);
    rawSizes[i] = accumulator->size;
    rawOffsets[i] = offset;
    offset += accumulator->size;
  }

  std::vector<char*> groupRows(offset);

  offset = 0;
  for (auto i = 0; i < groups.size(); ++i) {
    auto* accumulator = reinterpret_cast<RowPointers*>(groups[i] + offset_);
    accumulator->read(
        folly::Range(groupRows.data() + offset, accumulator->size));
    offset += accumulator->size;
  }

  auto& elementsVector = arrayVector->elements();
  elementsVector->resize(offset);
  inputData_->extractSerializedRows(
      folly::Range(groupRows.data(), groupRows.size()), elementsVector);
}

void SortedAggregations::clear() {
  inputData_->clear();