org.apache.drill.exec.physical.impl.partitionsender

Class PartitionerTemplate

java.lang.Object

org.apache.drill.exec.physical.impl.partitionsender.PartitionerTemplate

All Implemented Interfaces:

Partitioner

public abstract class PartitionerTemplate
extends Object
implements Partitioner

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	PartitionerTemplate.OutgoingRecordBatch

Field Summary

Fields inherited from interface org.apache.drill.exec.physical.impl.partitionsender.Partitioner

TEMPLATE_DEFINITION

Constructor Summary

Constructors

Constructor and Description
PartitionerTemplate()

Method Summary

Methods

Modifier and Type	Method and Description
void	clear()
abstract int	doEval(int inIndex)
abstract void	doSetup(FragmentContext context, RecordBatch incoming, PartitionerTemplate.OutgoingRecordBatch[] outgoing)
void	flushOutgoingBatches(boolean isLastBatch, boolean schemaChanged) Flush each outgoing record batch, and optionally reset the state of each outgoing record batch (on schema change).
PartitionOutgoingBatch	getOutgoingBatch(int index) Method to get PartitionOutgoingBatch based on the fact that there can be > 1 Partitioner
List<? extends PartitionOutgoingBatch>	getOutgoingBatches()
OperatorStats	getStats()
void	initialize()
protected PartitionerTemplate.OutgoingRecordBatch	newOutgoingRecordBatch(OperatorStats stats, HashPartitionSender operator, AccountingDataTunnel tunnel, FragmentContext context, BufferAllocator allocator, int oppositeMinorFragmentId) Shim method to be overridden in plain-old Java mode by the subclass to instantiate the generated inner class.
void	partitionBatch(RecordBatch incoming)
void	setup(FragmentContext context, RecordBatch incoming, HashPartitionSender popConfig, OperatorStats stats, OperatorContext oContext, int start, int end)

Methods inherited from class java.lang.Object

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

Constructor Detail

PartitionerTemplate

public PartitionerTemplate()

Method Detail

getOutgoingBatches

public List<? extends PartitionOutgoingBatch> getOutgoingBatches()

Specified by:

getOutgoingBatches in interface Partitioner

getOutgoingBatch

public PartitionOutgoingBatch getOutgoingBatch(int index)

Description copied from interface: Partitioner

Method to get PartitionOutgoingBatch based on the fact that there can be > 1 Partitioner

Specified by:

getOutgoingBatch in interface Partitioner

Returns:

PartitionOutgoingBatch that matches index within Partitioner. This method can return null if index does not fall within boundary of this Partitioner

setup

public final void setup(FragmentContext context,
         RecordBatch incoming,
         HashPartitionSender popConfig,
         OperatorStats stats,
         OperatorContext oContext,
         int start,
         int end)
                 throws SchemaChangeException

Specified by:

setup in interface Partitioner

Throws:

SchemaChangeException

newOutgoingRecordBatch

protected PartitionerTemplate.OutgoingRecordBatch newOutgoingRecordBatch(OperatorStats stats,
                                                             HashPartitionSender operator,
                                                             AccountingDataTunnel tunnel,
                                                             FragmentContext context,
                                                             BufferAllocator allocator,
                                                             int oppositeMinorFragmentId)

Shim method to be overridden in plain-old Java mode by the subclass to instantiate the generated inner class. Byte-code manipulation appears to fix up the byte codes directly. The name is special, it must be "new" + inner class name.

getStats

public OperatorStats getStats()

Specified by:

getStats in interface Partitioner

flushOutgoingBatches

public void flushOutgoingBatches(boolean isLastBatch,
                        boolean schemaChanged)
                          throws IOException

Flush each outgoing record batch, and optionally reset the state of each outgoing record batch (on schema change). Note that the schema is updated based on incoming at the time this function is invoked.

Specified by:

flushOutgoingBatches in interface Partitioner

Parameters:

isLastBatch - true if this is the last incoming batch

schemaChanged - true if the schema has changed

Throws:

IOException

partitionBatch

public void partitionBatch(RecordBatch incoming)
                    throws IOException

Specified by:

partitionBatch in interface Partitioner

Throws:

IOException

initialize

public void initialize()

Specified by:

initialize in interface Partitioner

clear

public void clear()

Specified by:

clear in interface Partitioner

doSetup

public abstract void doSetup(@Named(value="context")
           FragmentContext context,
           @Named(value="incoming")
           RecordBatch incoming,
           @Named(value="outgoing")
           PartitionerTemplate.OutgoingRecordBatch[] outgoing)
                      throws SchemaChangeException

Throws:

SchemaChangeException

doEval

public abstract int doEval(@Named(value="inIndex")
         int inIndex)
                    throws SchemaChangeException

Throws:

SchemaChangeException