org.apache.arrow.vector

Class ExtensionTypeVector<T extends ValueVector & FieldVector>

java.lang.Object

org.apache.arrow.vector.BaseValueVector

org.apache.arrow.vector.ExtensionTypeVector<T>

Type Parameters:

T - The wrapped vector type.

All Implemented Interfaces:

Closeable, AutoCloseable, Iterable<ValueVector>, FieldVector, ValueVector

public abstract class ExtensionTypeVector<T extends ValueVector & FieldVector>
extends BaseValueVector
implements FieldVector

A vector that wraps an underlying vector, used to help implement extension types.

Field Summary

Fields inherited from class org.apache.arrow.vector.BaseValueVector

allocator, fieldReader, INITIAL_VALUE_ALLOCATION, MAX_ALLOCATION_SIZE, MAX_ALLOCATION_SIZE_PROPERTY

Constructor Summary

Constructors

Constructor and Description
ExtensionTypeVector(Field field, BufferAllocator allocator, T underlyingVector) Instantiate an extension type vector.
ExtensionTypeVector(String name, BufferAllocator allocator, T underlyingVector) Instantiate an extension type vector.

Method Summary

Modifier and Type	Method and Description
<OUT,IN> OUT	accept(VectorVisitor<OUT,IN> visitor, IN value) Accept a generic VectorVisitor and return the result.
void	allocateNew() Allocate new buffers.
boolean	allocateNewSafe() Allocates new buffers.
void	clear() Release any owned ArrowBuf and reset the ValueVector to the initial state.
void	close() Alternative to clear().
BufferAllocator	getAllocator()
ArrowBuf[]	getBuffers(boolean clear) Return the underlying buffers associated with this vector.
int	getBufferSize() Get the number of bytes used by this vector.
int	getBufferSizeFor(int valueCount) Returns the number of bytes that is used by this vector if it holds the given number of values.
List<FieldVector>	getChildrenFromFields() The returned list is the same size as the list passed to initializeChildrenFromFields.
ArrowBuf	getDataBuffer() Gets the underlying buffer associated with data vector.
long	getDataBufferAddress() Gets the starting address of the underlying buffer associated with data vector.
Field	getField() Get information about how this field is materialized.
List<ArrowBuf>	getFieldBuffers() Get the buffers of the fields, (same size as getFieldVectors() since it is their content).
List<BufferBacked>	getFieldInnerVectors() Deprecated. This API will be removed as the current implementations no longer support inner vectors.
Types.MinorType	getMinorType()
String	getName() Gets the name of the vector.
int	getNullCount() Returns number of null elements in the vector.
abstract Object	getObject(int index) Get the extension object at the specified index.
ArrowBuf	getOffsetBuffer() Gets the underlying buffer associated with offset vector.
long	getOffsetBufferAddress() Gets the starting address of the underlying buffer associated with offset vector.
protected FieldReader	getReaderImpl() Each vector has a different reader that implements the FieldReader interface.
TransferPair	getTransferPair(BufferAllocator allocator) To transfer quota responsibility.
TransferPair	getTransferPair(String ref, BufferAllocator allocator)
TransferPair	getTransferPair(String ref, BufferAllocator allocator, CallBack callBack)
T	getUnderlyingVector() Get the underlying vector.
ArrowBuf	getValidityBuffer() Gets the underlying buffer associated with validity vector.
long	getValidityBufferAddress() Gets the starting address of the underlying buffer associated with validity vector.
int	getValueCapacity() Returns the maximum number of values that can be stored in this vector instance.
int	getValueCount() Gets the number of values.
void	initializeChildrenFromFields(List<Field> children) Initializes the child vectors to be later loaded with loadBuffers.
boolean	isNull(int index) Check whether an element in the vector is null.
Iterator<ValueVector>	iterator()
void	loadFieldBuffers(ArrowFieldNode fieldNode, List<ArrowBuf> ownBuffers) Loads data in the vectors.
TransferPair	makeTransferPair(ValueVector target) Makes a new transfer pair used to transfer underlying buffers.
void	reAlloc() Allocate new buffer with double capacity, and copy data into the new buffer.
void	reset() Reset the ValueVector to the initial state without releasing any owned ArrowBuf.
void	setInitialCapacity(int numRecords) Set the initial record capacity.
void	setNull(int index) Set the element at the given index to null.
void	setValueCount(int valueCount) Set number of values in the vector.

Methods inherited from class org.apache.arrow.vector.BaseValueVector

checkBufRefs, copyFrom, copyFromSafe, getReader, getValidityBufferSizeFromCount, releaseBuffer, toString, transferBuffer

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.apache.arrow.vector.ValueVector

copyFrom, copyFromSafe, getReader, hashCode, hashCode, validate, validateFull

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Constructor Detail

ExtensionTypeVector

public ExtensionTypeVector(String name,
                           BufferAllocator allocator,
                           T underlyingVector)

Instantiate an extension type vector.

Parameters:

name - name of the vector

allocator - allocator for memory management

underlyingVector - underlying filed vector

ExtensionTypeVector

public ExtensionTypeVector(Field field,
                           BufferAllocator allocator,
                           T underlyingVector)

Instantiate an extension type vector.

Parameters:

field - field materialized by this vector.

allocator - allocator for memory management

underlyingVector - underlying filed vector

Method Detail

getName

public String getName()

Description copied from interface: ValueVector

Gets the name of the vector.

Specified by:

getName in interface ValueVector

Specified by:

getName in class BaseValueVector

Returns:

the name of the vector.

getUnderlyingVector

public T getUnderlyingVector()

Get the underlying vector.

allocateNew

public void allocateNew()
                 throws OutOfMemoryException

Description copied from interface: ValueVector

Allocate new buffers. ValueVector implements logic to determine how much to allocate.

Specified by:

allocateNew in interface ValueVector

Throws:

OutOfMemoryException - Thrown if no memory can be allocated.

allocateNewSafe

public boolean allocateNewSafe()

Description copied from interface: ValueVector

Allocates new buffers. ValueVector implements logic to determine how much to allocate.

Specified by:

allocateNewSafe in interface ValueVector

Returns:

Returns true if allocation was successful.

reAlloc

public void reAlloc()

Description copied from interface: ValueVector

Allocate new buffer with double capacity, and copy data into the new buffer. Replace vector's buffer with new buffer, and release old one

Specified by:

reAlloc in interface ValueVector

setInitialCapacity

public void setInitialCapacity(int numRecords)

Description copied from interface: ValueVector

Set the initial record capacity.

Specified by:

setInitialCapacity in interface ValueVector

Parameters:

numRecords - the initial record capacity.

getValueCapacity

public int getValueCapacity()

Description copied from interface: ValueVector

Returns the maximum number of values that can be stored in this vector instance.

Specified by:

getValueCapacity in interface ValueVector

Returns:

the maximum number of values that can be stored in this vector instance.

reset

public void reset()

Description copied from interface: ValueVector

Reset the ValueVector to the initial state without releasing any owned ArrowBuf. Buffer capacities will remain unchanged and any previous data will be zeroed out. This includes buffers for data, validity, offset, etc. If the vector has any child vectors, they will also be reset.

Specified by:

reset in interface ValueVector

getField

public Field getField()

Description copied from interface: ValueVector

Get information about how this field is materialized.

Specified by:

getField in interface ValueVector

Returns:

the field corresponding to this vector

getMinorType

public Types.MinorType getMinorType()

Specified by:

getMinorType in interface ValueVector

getTransferPair

public TransferPair getTransferPair(String ref,
                                    BufferAllocator allocator)

Specified by:

getTransferPair in interface ValueVector

getTransferPair

public TransferPair getTransferPair(String ref,
                                    BufferAllocator allocator,
                                    CallBack callBack)

Specified by:

getTransferPair in interface ValueVector

makeTransferPair

public TransferPair makeTransferPair(ValueVector target)

Description copied from interface: ValueVector

Makes a new transfer pair used to transfer underlying buffers.

Specified by:

makeTransferPair in interface ValueVector

Parameters:

target - the target for the transfer

Returns:

a new transfer pair that is used to transfer underlying buffers into the target vector.

getReaderImpl

protected FieldReader getReaderImpl()

Description copied from class: BaseValueVector

Each vector has a different reader that implements the FieldReader interface. Overridden methods must make sure to return the correct concrete reader implementation.

Specified by:

getReaderImpl in class BaseValueVector

Returns:

Returns a lambda that initializes a reader when called.

getBufferSize

public int getBufferSize()

Description copied from interface: ValueVector

Get the number of bytes used by this vector.

Specified by:

getBufferSize in interface ValueVector

Returns:

the number of bytes that is used by this vector instance.

getBufferSizeFor

public int getBufferSizeFor(int valueCount)

Description copied from interface: ValueVector

Returns the number of bytes that is used by this vector if it holds the given number of values. The result will be the same as if setValueCount() were called, followed by calling getBufferSize(), but without any of the closing side-effects that setValueCount() implies wrt finishing off the population of a vector. Some operations might wish to use this to determine how much memory has been used by a vector so far, even though it is not finished being populated.

Specified by:

getBufferSizeFor in interface ValueVector

Parameters:

valueCount - the number of values to assume this vector contains

Returns:

the buffer size if this vector is holding valueCount values

getBuffers

public ArrowBuf[] getBuffers(boolean clear)

Description copied from interface: ValueVector

Return the underlying buffers associated with this vector. Note that this doesn't impact the reference counts for this buffer so it only should be used for in-context access. Also note that this buffer changes regularly thus external classes shouldn't hold a reference to it (unless they change it).

Specified by:

getBuffers in interface ValueVector

Parameters:

clear - Whether to clear vector before returning; the buffers will still be refcounted; but the returned array will be the only reference to them

Returns:

The underlying buffers that is used by this vector instance.

getValidityBuffer

public ArrowBuf getValidityBuffer()

Description copied from interface: ValueVector

Gets the underlying buffer associated with validity vector.

Specified by:

getValidityBuffer in interface ValueVector

Returns:

buffer

getDataBuffer

public ArrowBuf getDataBuffer()

Description copied from interface: ValueVector

Gets the underlying buffer associated with data vector.

Specified by:

getDataBuffer in interface ValueVector

Returns:

buffer

getOffsetBuffer

public ArrowBuf getOffsetBuffer()

Description copied from interface: ValueVector

Gets the underlying buffer associated with offset vector.

Specified by:

getOffsetBuffer in interface ValueVector

Returns:

buffer

getValueCount

public int getValueCount()

Description copied from interface: ValueVector

Gets the number of values.

Specified by:

getValueCount in interface ValueVector

Returns:

number of values in the vector

setValueCount

public void setValueCount(int valueCount)

Description copied from interface: ValueVector

Set number of values in the vector.

Specified by:

setValueCount in interface ValueVector

getObject

public abstract Object getObject(int index)

Get the extension object at the specified index.

Generally, this should access the underlying vector and construct the corresponding Java object from the raw data.

Specified by:

getObject in interface ValueVector

Parameters:

index - index of object to get

Returns:

friendly type object

getNullCount

public int getNullCount()

Description copied from interface: ValueVector

Returns number of null elements in the vector.

Specified by:

getNullCount in interface ValueVector

Returns:

number of null elements

isNull

public boolean isNull(int index)

Description copied from interface: ValueVector

Check whether an element in the vector is null.

Specified by:

isNull in interface ValueVector

Parameters:

index - index to check for null

Returns:

true if element is null

setNull

public void setNull(int index)

Description copied from interface: FieldVector

Set the element at the given index to null.

Specified by:

setNull in interface FieldVector

Parameters:

index - the value to change

initializeChildrenFromFields

public void initializeChildrenFromFields(List<Field> children)

Description copied from interface: FieldVector

Initializes the child vectors to be later loaded with loadBuffers.

Specified by:

initializeChildrenFromFields in interface FieldVector

Parameters:

children - the schema

getChildrenFromFields

public List<FieldVector> getChildrenFromFields()

Description copied from interface: FieldVector

The returned list is the same size as the list passed to initializeChildrenFromFields.

Specified by:

getChildrenFromFields in interface FieldVector

Returns:

the children according to schema (empty for primitive types)

loadFieldBuffers

public void loadFieldBuffers(ArrowFieldNode fieldNode,
                             List<ArrowBuf> ownBuffers)

Description copied from interface: FieldVector

Loads data in the vectors. (ownBuffers must be the same size as getFieldVectors())

Specified by:

loadFieldBuffers in interface FieldVector

Parameters:

fieldNode - the fieldNode

ownBuffers - the buffers for this Field (own buffers only, children not included)

getFieldBuffers

public List<ArrowBuf> getFieldBuffers()

Description copied from interface: FieldVector

Get the buffers of the fields, (same size as getFieldVectors() since it is their content).

Specified by:

getFieldBuffers in interface FieldVector

Returns:

the buffers containing the data for this vector (ready for reading)

getFieldInnerVectors

@Deprecated
public List<BufferBacked> getFieldInnerVectors()

Deprecated. This API will be removed as the current implementations no longer support inner vectors.

Get the inner vectors.

Specified by:

getFieldInnerVectors in interface FieldVector

Returns:

the inner vectors for this field as defined by the TypeLayout

getValidityBufferAddress

public long getValidityBufferAddress()

Description copied from interface: FieldVector

Gets the starting address of the underlying buffer associated with validity vector.

Specified by:

getValidityBufferAddress in interface FieldVector

Returns:

buffer address

getDataBufferAddress

public long getDataBufferAddress()

Description copied from interface: FieldVector

Gets the starting address of the underlying buffer associated with data vector.

Specified by:

getDataBufferAddress in interface FieldVector

Returns:

buffer address

getOffsetBufferAddress

public long getOffsetBufferAddress()

Description copied from interface: FieldVector

Gets the starting address of the underlying buffer associated with offset vector.

Specified by:

getOffsetBufferAddress in interface FieldVector

Returns:

buffer address

clear

public void clear()

Description copied from interface: ValueVector

Release any owned ArrowBuf and reset the ValueVector to the initial state. If the vector has any child vectors, they will also be cleared.

Specified by:

clear in interface ValueVector

Overrides:

clear in class BaseValueVector

close

public void close()

Description copied from interface: ValueVector

Alternative to clear(). Allows use as an AutoCloseable in try-with-resources.

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

Specified by:

close in interface ValueVector

Overrides:

close in class BaseValueVector

getTransferPair

public TransferPair getTransferPair(BufferAllocator allocator)

Description copied from interface: ValueVector

To transfer quota responsibility.

Specified by:

getTransferPair in interface ValueVector

Overrides:

getTransferPair in class BaseValueVector

Parameters:

allocator - the target allocator

Returns:

a transfer pair, creating a new target vector of the same type.

iterator

public Iterator<ValueVector> iterator()

Specified by:

iterator in interface Iterable<ValueVector>

Overrides:

iterator in class BaseValueVector

getAllocator

public BufferAllocator getAllocator()

Specified by:

getAllocator in interface ValueVector

Overrides:

getAllocator in class BaseValueVector

accept

public <OUT,IN> OUT accept(VectorVisitor<OUT,IN> visitor,
                           IN value)

Description copied from interface: ValueVector

Accept a generic VectorVisitor and return the result.

Specified by:

accept in interface ValueVector

Type Parameters:

OUT - the output result type.

IN - the input data together with visitor.