org.apache.drill.exec.physical.impl.limit

Class PartitionLimitRecordBatch

java.lang.Object

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

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

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

org.apache.drill.exec.physical.impl.limit.PartitionLimitRecordBatch

All Implemented Interfaces:

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

public class PartitionLimitRecordBatch
extends AbstractSingleRecordBatch<PartitionLimit>

Helps to perform limit in a partition within a record batch. Currently only integer type of partition column is supported. This is mainly used for Lateral/Unnest subquery where each output batch from Unnest will contain an implicit column for rowId for each row.

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

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
PartitionLimitRecordBatch(PartitionLimit popConfig, FragmentContext context, RecordBatch incoming)

Method Summary

Modifier and Type	Method and Description
void	close()
protected RecordBatch.IterOutcome	doWork()
void	dump() Perform dump of this batch's state to logs.
protected RecordBatch.IterOutcome	getFinalOutcome(boolean hasRemainder) Gets the outcome to return from super implementation and then in case of EMIT outcome it refreshes the state of operator.
int	getRecordCount() Get the number of records.
SelectionVector2	getSelectionVector2()
protected boolean	setupNewSchema()

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

getIncoming

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

cancelIncoming, getLastKnownOutcome, handleNullInput, innerNext, setLastKnownOutcome

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

buildSchema, cancel, checkContinue, getContainer, getContext, getOutgoingContainer, getPopConfig, getRecordBatchStatsContext, getSchema, 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

Constructor Detail

PartitionLimitRecordBatch

public PartitionLimitRecordBatch(PartitionLimit popConfig,
                                 FragmentContext context,
                                 RecordBatch incoming)
                          throws OutOfMemoryException

Throws:

OutOfMemoryException

Method Detail

getSelectionVector2

public SelectionVector2 getSelectionVector2()

Specified by:

getSelectionVector2 in interface VectorAccessible

Overrides:

getSelectionVector2 in class AbstractRecordBatch<PartitionLimit>

getRecordCount

public int getRecordCount()

Description copied from interface: VectorAccessible

Get the number of records.

Returns:

number of records

close

public void close()

Specified by:

close in interface AutoCloseable

Overrides:

close in class AbstractRecordBatch<PartitionLimit>

setupNewSchema

protected boolean setupNewSchema()

Specified by:

setupNewSchema in class AbstractUnaryRecordBatch<PartitionLimit>

getFinalOutcome

protected RecordBatch.IterOutcome getFinalOutcome(boolean hasRemainder)

Gets the outcome to return from super implementation and then in case of EMIT outcome it refreshes the state of operator. Refresh is done to again apply limit on all the future incoming batches which will be part of next record boundary.

Overrides:

getFinalOutcome in class AbstractSingleRecordBatch<PartitionLimit>

Parameters:

hasRemainder -

Returns:

- IterOutcome to send downstream

doWork

protected RecordBatch.IterOutcome doWork()

Specified by:

doWork in class AbstractUnaryRecordBatch<PartitionLimit>

dump

public void dump()

Description copied from interface: RecordBatch

Perform dump of this batch's state to logs.