Function exportValues

bolt/vector/arrow/Bridge.cpp:840–914

}

void exportValues(
    const BaseVector& vec,
    const Selection& rows,
    const ArrowOptions& options,
    ArrowArray& out,
    memory::MemoryPool* pool,
    BoltToArrowBridgeHolder& holder) {
  const auto& type = vec.type();
  out.n_buffers = 2;

  if (!vec.values()) {
    // all values are null.
    BOLT_CHECK_EQ(out.null_count, out.length);

    switch (vec.typeKind()) {
      case TypeKind::BOOLEAN:
      case TypeKind::TINYINT:
        holder.setBuffer(1, AlignedBuffer::allocate<uint8_t>(vec.size(), pool));
        break;
      case TypeKind::SMALLINT:
        holder.setBuffer(
            1, AlignedBuffer::allocate<uint16_t>(vec.size(), pool));
        break;
      case TypeKind::INTEGER:
        holder.setBuffer(
            1, AlignedBuffer::allocate<uint32_t>(vec.size(), pool));
        break;
      case TypeKind::BIGINT:
        holder.setBuffer(
            1, AlignedBuffer::allocate<uint64_t>(vec.size(), pool));
        break;
      case TypeKind::HUGEINT:
        holder.setBuffer(
            1, AlignedBuffer::allocate<uint128_t>(vec.size(), pool));
        break;
      case TypeKind::REAL:
        holder.setBuffer(1, AlignedBuffer::allocate<float>(vec.size(), pool));
        break;
      case TypeKind::DOUBLE:
        holder.setBuffer(1, AlignedBuffer::allocate<double>(vec.size(), pool));
        break;
      case TypeKind::TIMESTAMP:
        holder.setBuffer(
            1, AlignedBuffer::allocate<uint128_t>(vec.size(), pool));
        break;
      default:
        BOLT_UNREACHABLE();
    }
    return;
  }
  if (!rows.changed() && isFlatScalarZeroCopy(type)) {
    // Arrow does not allow a nullptr for the values buffer.
    // The all-null (no values buffer) case is handled above.
    BOLT_DCHECK(vec.values());
    holder.setBuffer(1, vec.values());
    return;
  }