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

Class ProjectRecordBatch

java.lang.Object

org.apache.drill.exec.record.AbstractRecordBatch<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 inherited from class org.apache.drill.exec.record.AbstractSingleRecordBatch

callBack, incoming, outOfMemory

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

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

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

MAX_BATCH_SIZE

Constructor Summary

Constructors

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

Method Summary

Methods

Modifier and Type	Method and Description
void	addComplexWriter(BaseWriter.ComplexWriter writer)
protected RecordBatch.IterOutcome	doWork()
VectorContainer	getOutgoingContainer()
int	getRecordCount() Gets the number of records that are within this record.
protected RecordBatch.IterOutcome	handleNullInput() Handle Null input specially when Project operator is for query output.
RecordBatch.IterOutcome	innerNext()
protected void	killIncoming(boolean sendUpstream)
protected boolean	setupNewSchema()

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

getSchema

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

buildSchema, close, getContext, getPopConfig, getSelectionVector2, getSelectionVector4, getValueAccessorById, getValueVectorId, getWritableBatch, iterator, kill, next, next, next

Methods inherited from class java.lang.Object

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

Constructor Detail

ProjectRecordBatch

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

Throws:

OutOfMemoryException

Method Detail

getRecordCount

public int getRecordCount()

Description copied from interface: RecordBatch

Gets the number of records that are within this record.

Returns:

number of records

killIncoming

protected void killIncoming(boolean sendUpstream)

Overrides:

killIncoming in class AbstractSingleRecordBatch<Project>

innerNext

public RecordBatch.IterOutcome innerNext()

Overrides:

innerNext in class AbstractSingleRecordBatch<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 AbstractSingleRecordBatch<Project>

addComplexWriter

public void addComplexWriter(BaseWriter.ComplexWriter writer)

setupNewSchema

protected boolean setupNewSchema()
                          throws SchemaChangeException

Specified by:

setupNewSchema in class AbstractSingleRecordBatch<Project>

Throws:

SchemaChangeException

handleNullInput

protected RecordBatch.IterOutcome handleNullInput()

Handle Null input specially when Project operator is for query output. This happens when input return 0 batch (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 AbstractSingleRecordBatch<Project>

Returns:

IterOutcome.NONE.