org.apache.drill.exec.physical.impl.project

Class ProjectRecordBatch

java.lang.Object

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

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

org.apache.drill.exec.record.AbstractSingleRecordBatch<Project>

org.apache.drill.exec.physical.impl.project.ProjectRecordBatch

All Implemented Interfaces:

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

public class ProjectRecordBatch
extends AbstractSingleRecordBatch<Project>

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.drill.exec.record.AbstractRecordBatch

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 List<ValueVector>	allocationVectors
protected List<FieldReference>	complexFieldReferencesList
protected List<BaseWriter.ComplexWriter>	complexWriters
protected ProjectMemoryManager	memoryManager

Fields inherited from class org.apache.drill.exec.record.AbstractSingleRecordBatch

incoming

Fields inherited from class org.apache.drill.exec.record.AbstractUnaryRecordBatch

callBack

Fields inherited from class org.apache.drill.exec.record.AbstractRecordBatch

batchStatsContext, container, context, oContext, popConfig, state, stats, unionTypeEnabled

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

MAX_BATCH_ROW_COUNT

Constructor Summary

Constructors

Constructor and Description
ProjectRecordBatch(Project pop, RecordBatch incoming, FragmentContext context)

Method Summary

Modifier and Type	Method and Description
void	addComplexWriter(BaseWriter.ComplexWriter writer)
protected void	cancelIncoming()
protected RecordBatch.IterOutcome	doWork()
void	dump() Perform dump of this batch's state to logs.
protected RecordBatch.IterOutcome	getFinalOutcome(boolean hasMoreRecordInBoundary) Based on lastKnownOutcome and if there are more records to be output for current record boundary detected by EMIT outcome, this method returns EMIT or OK outcome.
VectorContainer	getOutgoingContainer()
int	getRecordCount() Get the number of records.
protected RecordBatch.IterOutcome	handleNullInput() Handle Null input specially when Project operator is for query output.
RecordBatch.IterOutcome	innerNext()
protected boolean	setupNewSchema()

Methods inherited from class org.apache.drill.exec.record.AbstractSingleRecordBatch

getIncoming

Methods inherited from class org.apache.drill.exec.record.AbstractUnaryRecordBatch

getLastKnownOutcome, setLastKnownOutcome

Methods inherited from class org.apache.drill.exec.record.AbstractRecordBatch

buildSchema, cancel, checkContinue, close, getContainer, getContext, getPopConfig, getRecordBatchStatsContext, getSchema, getSelectionVector2, getSelectionVector4, getValueAccessorById, getValueVectorId, getWritableBatch, isRecordBatchStatsLoggingEnabled, iterator, next, next, next, schemaChangeException, schemaChangeException

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Field Detail

allocationVectors

protected List<ValueVector> allocationVectors

complexWriters

protected List<BaseWriter.ComplexWriter> complexWriters

complexFieldReferencesList

protected List<FieldReference> complexFieldReferencesList

memoryManager

protected ProjectMemoryManager memoryManager

Constructor Detail

ProjectRecordBatch

public ProjectRecordBatch(Project pop,
                          RecordBatch incoming,
                          FragmentContext context)

Method Detail

getRecordCount

public int getRecordCount()

Description copied from interface: VectorAccessible

Get the number of records.

Returns:

number of records

cancelIncoming

protected void cancelIncoming()

Overrides:

cancelIncoming in class AbstractUnaryRecordBatch<Project>

innerNext

public RecordBatch.IterOutcome innerNext()

Overrides:

innerNext in class AbstractUnaryRecordBatch<Project>

getOutgoingContainer

public VectorContainer getOutgoingContainer()

Specified by:

getOutgoingContainer in interface RecordBatch

Overrides:

getOutgoingContainer in class AbstractRecordBatch<Project>

doWork

protected RecordBatch.IterOutcome doWork()

Specified by:

doWork in class AbstractUnaryRecordBatch<Project>

addComplexWriter

public void addComplexWriter(BaseWriter.ComplexWriter writer)

setupNewSchema

protected boolean setupNewSchema()

Specified by:

setupNewSchema in class AbstractUnaryRecordBatch<Project>

handleNullInput

protected RecordBatch.IterOutcome handleNullInput()

Handle Null input specially when Project operator is for query output. This happens when the input returns no batches (returns a FAST NONE directly).

Project operator has to return a batch with schema derived using the following 3 rules:

• Case 1: * ==> expand into an empty list of columns.
• Case 2: regular column reference ==> treat as nullable-int column
• Case 3: expressions => Call ExpressionTreeMaterialization over an empty vector contain. Once the expression is materialized without error, use the output type of materialized expression.

The batch is constructed with the above rules, and recordCount = 0. Returned with OK_NEW_SCHEMA to down-stream operator.

Overrides:

handleNullInput in class AbstractUnaryRecordBatch<Project>

Returns:

IterOutcome.NONE.

getFinalOutcome

protected RecordBatch.IterOutcome getFinalOutcome(boolean hasMoreRecordInBoundary)

Description copied from class: AbstractSingleRecordBatch

Based on lastKnownOutcome and if there are more records to be output for current record boundary detected by EMIT outcome, this method returns EMIT or OK outcome.

Overrides:

getFinalOutcome in class AbstractSingleRecordBatch<Project>

Returns:

EMIT - If the lastknownOutcome was EMIT and output records corresponding to all the incoming records in current record boundary is already produced. OK - otherwise

dump

public void dump()

Description copied from interface: RecordBatch

Perform dump of this batch's state to logs.