org.apache.drill.exec.record

Class AbstractRecordBatch<T extends PhysicalOperator>

java.lang.Object

org.apache.drill.exec.record.AbstractRecordBatch<T>

All Implemented Interfaces:

AutoCloseable, Iterable<VectorWrapper<?>>, CloseableRecordBatch, RecordBatch, VectorAccessible

Direct Known Subclasses:

AbstractBinaryRecordBatch, AbstractSingleRecordBatch, ExternalSortBatch, ExternalSortBatch, HashAggBatch, MergeJoinBatch, MergingRecordBatch, OrderedPartitionRecordBatch, ProducerConsumerBatch, SortBatch, StreamingAggBatch, TopNBatch, WindowFrameRecordBatch, WriterRecordBatch

public abstract class AbstractRecordBatch<T extends PhysicalOperator>
extends Object
implements CloseableRecordBatch

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static class	AbstractRecordBatch.BatchState

Nested classes/interfaces inherited from interface org.apache.drill.exec.record.RecordBatch

RecordBatch.IterOutcome

Field Summary

Fields

Modifier and Type	Field and Description
protected VectorContainer	container
protected FragmentContext	context
protected OperatorContext	oContext
protected T	popConfig
protected AbstractRecordBatch.BatchState	state
protected OperatorStats	stats
protected boolean	unionTypeEnabled

Fields inherited from interface org.apache.drill.exec.record.RecordBatch

MAX_BATCH_SIZE

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	AbstractRecordBatch(T popConfig, FragmentContext context)
protected	AbstractRecordBatch(T popConfig, FragmentContext context, boolean buildSchema)
protected	AbstractRecordBatch(T popConfig, FragmentContext context, boolean buildSchema, OperatorContext oContext)

Method Summary

Methods

Modifier and Type	Method and Description
protected void	buildSchema()
void	close()
FragmentContext	getContext() Gets the FragmentContext of the current query fragment.
VectorContainer	getOutgoingContainer()
PhysicalOperator	getPopConfig()
BatchSchema	getSchema() Gets the current schema of this record batch.
SelectionVector2	getSelectionVector2()
SelectionVector4	getSelectionVector4()
VectorWrapper<?>	getValueAccessorById(Class<?> clazz, int... ids)
TypedFieldId	getValueVectorId(SchemaPath path) Gets the value vector type and ID for the given schema path.
WritableBatch	getWritableBatch() Gets a writable version of this batch.
abstract RecordBatch.IterOutcome	innerNext()
Iterator<VectorWrapper<?>>	iterator()
void	kill(boolean sendUpstream) Informs child nodes that this query should be terminated.
protected abstract void	killIncoming(boolean sendUpstream)
RecordBatch.IterOutcome	next() Updates the data in each Field reading interface for the next range of records.
RecordBatch.IterOutcome	next(int inputIndex, RecordBatch b)
RecordBatch.IterOutcome	next(RecordBatch b)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.drill.exec.record.RecordBatch

getRecordCount

Field Detail

container

protected final VectorContainer container

popConfig

protected final T extends PhysicalOperator popConfig

context

protected final FragmentContext context

oContext

protected final OperatorContext oContext

stats

protected final OperatorStats stats

unionTypeEnabled

protected final boolean unionTypeEnabled

state

protected AbstractRecordBatch.BatchState state

Constructor Detail

AbstractRecordBatch

protected AbstractRecordBatch(T popConfig,
                   FragmentContext context)
                       throws OutOfMemoryException

Throws:

OutOfMemoryException

AbstractRecordBatch

protected AbstractRecordBatch(T popConfig,
                   FragmentContext context,
                   boolean buildSchema)
                       throws OutOfMemoryException

Throws:

OutOfMemoryException

AbstractRecordBatch

protected AbstractRecordBatch(T popConfig,
                   FragmentContext context,
                   boolean buildSchema,
                   OperatorContext oContext)

Method Detail

iterator

public Iterator<VectorWrapper<?>> iterator()

Specified by:

iterator in interface Iterable<VectorWrapper<?>>

getContext

public FragmentContext getContext()

Description copied from interface: RecordBatch

Gets the FragmentContext of the current query fragment. Useful for reporting failure information or other query-level information.

Specified by:

getContext in interface RecordBatch

getPopConfig

public PhysicalOperator getPopConfig()

next

public final RecordBatch.IterOutcome next(RecordBatch b)

next

public final RecordBatch.IterOutcome next(int inputIndex,
                           RecordBatch b)

next

public final RecordBatch.IterOutcome next()

Description copied from interface: RecordBatch

Updates the data in each Field reading interface for the next range of records.

Once a RecordBatch's next() has returned RecordBatch.IterOutcome.NONE or RecordBatch.IterOutcome.STOP, the consumer should no longer call next(). Behavior at this point is undefined and likely to throw an exception.

See RecordBatch.IterOutcome for the protocol (possible sequences of return values).

Specified by:

next in interface RecordBatch

Returns:

An IterOutcome describing the result of the iteration.

innerNext

public abstract RecordBatch.IterOutcome innerNext()

getSchema

public BatchSchema getSchema()

Description copied from interface: RecordBatch

Gets the current schema of this record batch.

May be called only when the most recent call to RecordBatch.next(), if any, returned #OK_NEW_SCHEMA or #OK.

The schema changes when and only when RecordBatch.next() returns #OK_NEW_SCHEMA.

Specified by:

getSchema in interface RecordBatch

Specified by:

getSchema in interface VectorAccessible

Returns:

schema of the current batch

buildSchema

protected void buildSchema()
                    throws SchemaChangeException

Throws:

SchemaChangeException

kill

public void kill(boolean sendUpstream)

Description copied from interface: RecordBatch

Informs child nodes that this query should be terminated. Child nodes should use the QueryContext to determine what has happened.

Specified by:

kill in interface RecordBatch

killIncoming

protected abstract void killIncoming(boolean sendUpstream)

close

public void close()

Specified by:

close in interface AutoCloseable

getSelectionVector2

public SelectionVector2 getSelectionVector2()

Specified by:

getSelectionVector2 in interface VectorAccessible

getSelectionVector4

public SelectionVector4 getSelectionVector4()

Specified by:

getSelectionVector4 in interface VectorAccessible

getValueVectorId

public TypedFieldId getValueVectorId(SchemaPath path)

Description copied from interface: RecordBatch

Gets the value vector type and ID for the given schema path. The TypedFieldId should store a fieldId which is the same as the ordinal position of the field within the Iterator provided this class's implementation of Iterable.

Specified by:

getValueVectorId in interface RecordBatch

Specified by:

getValueVectorId in interface VectorAccessible

Parameters:

path - The path where the vector should be located.

Returns:

The local field id associated with this vector. If no field matches this path, this will return a null TypedFieldId

getValueAccessorById

public VectorWrapper<?> getValueAccessorById(Class<?> clazz,
                                    int... ids)

Specified by:

getValueAccessorById in interface RecordBatch

Specified by:

getValueAccessorById in interface VectorAccessible

getWritableBatch

public WritableBatch getWritableBatch()

Description copied from interface: RecordBatch

Gets a writable version of this batch. Takes over ownership of existing buffers.

Specified by:

getWritableBatch in interface RecordBatch

getOutgoingContainer

public VectorContainer getOutgoingContainer()

Specified by:

getOutgoingContainer in interface RecordBatch