Method copyRangesImpl

bolt/vector/ComplexVector.cpp:519–641

}

void ArrayVectorBase::copyRangesImpl(
    const BaseVector* source,
    const folly::Range<const BaseVector::CopyRange*>& ranges,
    VectorPtr* targetValues,
    VectorPtr* targetKeys) {
  if (isAllNullVector(*source)) {
    BaseVector::setNulls(mutableRawNulls(), ranges, true);
    return;
  }

  const BaseVector* sourceValues;
  const BaseVector* sourceKeys = nullptr;

  auto leafSource = source->wrappedVector();
  BOLT_CHECK_EQ(leafSource->encoding(), encoding());

  if (typeKind_ == TypeKind::ARRAY) {
    sourceValues = leafSource->as<ArrayVector>()->elements().get();
  } else {
    sourceValues = leafSource->as<MapVector>()->mapValues().get();
    sourceKeys = leafSource->as<MapVector>()->mapKeys().get();
  }

  if (targetKeys) {
    BaseVector::ensureWritable(
        SelectivityVector::empty(),
        targetKeys->get()->type(),
        pool(),
        *targetKeys);
  } else {
    BaseVector::ensureWritable(
        SelectivityVector::empty(),
        targetValues->get()->type(),
        pool(),
        *targetValues);
  }

  auto sourceArray = leafSource->asUnchecked<ArrayVectorBase>();
  auto setNotNulls = mayHaveNulls() || source->mayHaveNulls();
  auto* mutableOffsets = offsets_->asMutable<vector_size_t>();
  auto* mutableSizes = sizes_->asMutable<vector_size_t>();
  vector_size_t childSize = targetValues->get()->size();
  if (ranges.size() == 1 && ranges.back().count == 1) {
    auto& range = ranges.back();
    if (range.count == 0) {
      return;
    }
    BOLT_DCHECK(BaseVector::length_ >= range.targetIndex + range.count);
    // Fast path if we're just copying a single array.
    if (source->isNullAt(range.sourceIndex)) {
      setNull(range.targetIndex, true);
    } else {
      if (setNotNulls) {
        setNull(range.targetIndex, false);
      }

      vector_size_t wrappedIndex = source->wrappedIndex(range.sourceIndex);
      vector_size_t copySize = sourceArray->sizeAt(wrappedIndex);