org.apache.drill.exec.physical.impl.TopN

Class TopNBatch

java.lang.Object

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

org.apache.drill.exec.physical.impl.TopN.TopNBatch

All Implemented Interfaces:

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

public class TopNBatch
extends AbstractRecordBatch<TopN>

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TopNBatch.SimpleSV4RecordBatch

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
MappingSet	leftMapping
MappingSet	mainMapping
MappingSet	rightMapping

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

Method Summary

Methods

Modifier and Type	Method and Description
void	buildSchema()
void	close()
static MappingSet	createLeftMappingSet()
static MappingSet	createMainMappingSet()
static PriorityQueue	createNewPriorityQueue(MappingSet mainMapping, MappingSet leftMapping, MappingSet rightMapping, OptionSet optionSet, FunctionLookupContext functionLookupContext, CodeCompiler codeCompiler, List<Order.Ordering> orderings, VectorAccessible batch, boolean unionTypeEnabled, boolean codegenDump, int limit, BufferAllocator allocator, BatchSchema.SelectionVectorMode mode)
static MappingSet	createRightMappingSet()
int	getRecordCount() Gets the number of records that are within this record.
SelectionVector2	getSelectionVector2()
SelectionVector4	getSelectionVector4()
WritableBatch	getWritableBatch() Gets a writable version of this batch.
RecordBatch.IterOutcome	innerNext()
protected void	killIncoming(boolean sendUpstream)
void	purgeAndResetPriorityQueue() Handle schema changes during execution.

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

getContext, getOutgoingContainer, getPopConfig, getSchema, getValueAccessorById, getValueVectorId, iterator, kill, next, next, next

Methods inherited from class java.lang.Object

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

Field Detail

mainMapping

public final MappingSet mainMapping

leftMapping

public final MappingSet leftMapping

rightMapping

public final MappingSet rightMapping

Constructor Detail

TopNBatch

public TopNBatch(TopN popConfig,
         FragmentContext context,
         RecordBatch incoming)
          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

getSelectionVector2

public SelectionVector2 getSelectionVector2()

Specified by:

getSelectionVector2 in interface VectorAccessible

Overrides:

getSelectionVector2 in class AbstractRecordBatch<TopN>

getSelectionVector4

public SelectionVector4 getSelectionVector4()

Specified by:

getSelectionVector4 in interface VectorAccessible

Overrides:

getSelectionVector4 in class AbstractRecordBatch<TopN>

close

public void close()

Specified by:

close in interface AutoCloseable

Overrides:

close in class AbstractRecordBatch<TopN>

buildSchema

public void buildSchema()
                 throws SchemaChangeException

Overrides:

buildSchema in class AbstractRecordBatch<TopN>

Throws:

SchemaChangeException

innerNext

public RecordBatch.IterOutcome innerNext()

Specified by:

innerNext in class AbstractRecordBatch<TopN>

createMainMappingSet

public static MappingSet createMainMappingSet()

createLeftMappingSet

public static MappingSet createLeftMappingSet()

createRightMappingSet

public static MappingSet createRightMappingSet()

createNewPriorityQueue

public static PriorityQueue createNewPriorityQueue(MappingSet mainMapping,
                                   MappingSet leftMapping,
                                   MappingSet rightMapping,
                                   OptionSet optionSet,
                                   FunctionLookupContext functionLookupContext,
                                   CodeCompiler codeCompiler,
                                   List<Order.Ordering> orderings,
                                   VectorAccessible batch,
                                   boolean unionTypeEnabled,
                                   boolean codegenDump,
                                   int limit,
                                   BufferAllocator allocator,
                                   BatchSchema.SelectionVectorMode mode)
                                            throws ClassTransformationException,
                                                   IOException,
                                                   SchemaChangeException

Throws:

ClassTransformationException

IOException

SchemaChangeException

purgeAndResetPriorityQueue

public void purgeAndResetPriorityQueue()
                                throws SchemaChangeException,
                                       ClassTransformationException,
                                       IOException

Handle schema changes during execution. 1. Purge existing batches 2. Promote newly created container for new schema. 3. Recreate priority queue and reset with coerced container.

Throws:

SchemaChangeException

ClassTransformationException

IOException

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

Overrides:

getWritableBatch in class AbstractRecordBatch<TopN>

killIncoming

protected void killIncoming(boolean sendUpstream)

Specified by:

killIncoming in class AbstractRecordBatch<TopN>