Method append

bolt/exec/AddressableNonNullValueList.cpp:35–71

namespace bytedance::bolt::aggregate::prestosql {

AddressableNonNullValueList::Entry AddressableNonNullValueList::append(
    const DecodedVector& decoded,
    vector_size_t index,
    HashStringAllocator* allocator) {
  ByteOutputStream stream(allocator);
  if (!firstHeader_) {
    // An array_agg or related begins with an allocation of 5 words and
    // 4 bytes for header. This is compact for small arrays (up to 5
    // bigints) and efficient if needs to be extended (stores 4 bigints
    // and a next pointer. This could be adaptive, with smaller initial
    // sizes for lots of small arrays.
    static constexpr int kInitialSize = 44;

    currentPosition_ = allocator->newWrite(stream, kInitialSize);
    firstHeader_ = currentPosition_.header;
  } else {
    allocator->extendWrite(currentPosition_, stream);
  }

  const auto hash = decoded.base()->hashValueAt(decoded.index(index));

  const auto originalSize = stream.size();

  // Write value.
  exec::ContainerRowSerdeOptions options{};
  exec::ContainerRowSerde::serialize(
      *decoded.base(), decoded.index(index), stream, options);

  ++size_;

  auto startAndFinish = allocator->finishWrite(stream, 1024);
  currentPosition_ = startAndFinish.second;

  const auto writtenSize = stream.size() - originalSize;

  return {startAndFinish.first, writtenSize, hash};
}

namespace {