org.apache.drill.exec.physical.impl.unnest

Class MockLateralJoinBatch

java.lang.Object

org.apache.drill.exec.physical.impl.unnest.MockLateralJoinBatch

All Implemented Interfaces:

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

public class MockLateralJoinBatch
extends Object
implements LateralContract, CloseableRecordBatch

A mock lateral join implementation for testing unnest. This ignores all the other input and simply puts the unnest output into a results hypervector. Since Unnest returns an empty batch when it encounters a schema change, this implementation will also output an empty batch when it sees a schema change from unnest

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.drill.exec.record.RecordBatch

RecordBatch.IterOutcome

Field Summary

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

MAX_BATCH_ROW_COUNT

Constructor Summary

Constructors

Constructor and Description
MockLateralJoinBatch(FragmentContext context, OperatorContext oContext, RecordBatch incoming)

Method Summary

Modifier and Type	Method and Description
void	cancel() Informs child operators that no more data is needed.
void	close()
void	dump() Perform dump of this batch's state to logs.
VectorContainer	getContainer() Return the internal vector container
FragmentContext	getContext() Gets the FragmentContext of the current query fragment.
RecordBatch	getIncoming() Get reference to left side incoming of LateralJoinBatch.
RecordBatch.IterOutcome	getLeftOutcome() Get the current outcome of left incoming batch.
VectorContainer	getOutgoingContainer()
int	getRecordCount() Get the number of records.
int	getRecordIndex() Get current record index in incoming to be processed.
List<ValueVector>	getResultList()
BatchSchema	getSchema() Gets the current schema of this record batch.
SelectionVector2	getSelectionVector2()
SelectionVector4	getSelectionVector4()
RecordBatch	getUnnest()
VectorWrapper<?>	getValueAccessorById(Class<?> clazz, int... ids)
TypedFieldId	getValueVectorId(SchemaPath path) Gets the value vector type and ID for the given schema path.
WritableBatch	getWritableBatch() Gets a writable version of this batch.
boolean	isCompleted()
Iterator<VectorWrapper<?>>	iterator()
void	moveToNextRecord()
RecordBatch.IterOutcome	next() Updates the data in each Field reading interface for the next range of records.
void	reset()
void	setUnnest(RecordBatch unnest)
void	setUnnestLimit(int limit)

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

MockLateralJoinBatch

public MockLateralJoinBatch(FragmentContext context,
                            OperatorContext oContext,
                            RecordBatch incoming)

Method Detail

getIncoming

public RecordBatch getIncoming()

Description copied from interface: LateralContract

Get reference to left side incoming of LateralJoinBatch.

Specified by:

getIncoming in interface LateralContract

Returns:

The incoming RecordBatch

getRecordIndex

public int getRecordIndex()

Description copied from interface: LateralContract

Get current record index in incoming to be processed.

Specified by:

getRecordIndex in interface LateralContract

Returns:

The current record index in incoming to be processed.

getLeftOutcome

public RecordBatch.IterOutcome getLeftOutcome()

Description copied from interface: LateralContract

Get the current outcome of left incoming batch.

Specified by:

getLeftOutcome in interface LateralContract

Returns:

The current outcome of left incoming batch.

moveToNextRecord

public void moveToNextRecord()

reset

public void reset()

setUnnest

public void setUnnest(RecordBatch unnest)

setUnnestLimit

public void setUnnestLimit(int limit)

getUnnest

public RecordBatch getUnnest()

next

public RecordBatch.IterOutcome next()

Description copied from interface: RecordBatch

Updates the data in each Field reading interface for the next range of records.

Once a RecordBatch's next() has returned RecordBatch.IterOutcome.NONE or IterOutcome#STOP, the consumer should no longer call next(). Behavior at this point is undefined and likely to throw an exception.

See RecordBatch.IterOutcome for the protocol (possible sequences of return values).

Specified by:

next in interface RecordBatch

Returns:

An IterOutcome describing the result of the iteration.

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

getResultList

public List<ValueVector> getResultList()

close

public void close()
           throws Exception

Specified by:

close in interface AutoCloseable

Throws:

Exception

dump

public void dump()

Description copied from interface: RecordBatch

Perform dump of this batch's state to logs.

Specified by:

dump in interface RecordBatch

getRecordCount

public int getRecordCount()

Description copied from interface: VectorAccessible

Get the number of records.

Specified by:

getRecordCount in interface VectorAccessible

Returns:

number of records

getSelectionVector2

public SelectionVector2 getSelectionVector2()

Specified by:

getSelectionVector2 in interface VectorAccessible

getSelectionVector4

public SelectionVector4 getSelectionVector4()

Specified by:

getSelectionVector4 in interface VectorAccessible

getContext

public FragmentContext getContext()

Description copied from interface: RecordBatch

Gets the FragmentContext of the current query fragment. Useful for reporting failure information or other query-level information.

Specified by:

getContext in interface RecordBatch

getSchema

public BatchSchema getSchema()

Description copied from interface: RecordBatch

Gets the current schema of this record batch.

May be called only when the most recent call to RecordBatch.next(), if any, returned RecordBatch.IterOutcome.OK_NEW_SCHEMA or RecordBatch.IterOutcome.OK.

The schema changes when and only when RecordBatch.next() returns RecordBatch.IterOutcome.OK_NEW_SCHEMA.

Specified by:

getSchema in interface RecordBatch

Specified by:

getSchema in interface VectorAccessible

Returns:

schema of the current batch

cancel

public void cancel()

Description copied from interface: RecordBatch

Informs child operators that no more data is needed. Only called for "normal" cancellation to avoid unnecessary compute in any worker threads. For the error case, the fragment executor will call close() on each child automatically.

The operator which triggers the cancel MUST send a NONE status downstream, or throw an exception. It is not legal to call next() on an operator after calling its cancel() method.

Specified by:

cancel in interface RecordBatch

getOutgoingContainer

public VectorContainer getOutgoingContainer()

Specified by:

getOutgoingContainer in interface RecordBatch

getValueVectorId

public TypedFieldId getValueVectorId(SchemaPath path)

Description copied from interface: RecordBatch

Gets the value vector type and ID for the given schema path. The TypedFieldId should store a fieldId which is the same as the ordinal position of the field within the Iterator provided this class's implementation of Iterable<ValueVector>.

Specified by:

getValueVectorId in interface RecordBatch

Specified by:

getValueVectorId in interface VectorAccessible

Parameters:

path - The path where the vector should be located.

Returns:

The local field id associated with this vector. If no field matches this path, this will return a null TypedFieldId

getValueAccessorById

public VectorWrapper<?> getValueAccessorById(Class<?> clazz,
                                             int... ids)

Specified by:

getValueAccessorById in interface RecordBatch

Specified by:

getValueAccessorById in interface VectorAccessible

isCompleted

public boolean isCompleted()

getContainer

public VectorContainer getContainer()

Description copied from interface: RecordBatch

Return the internal vector container

Specified by:

getContainer in interface RecordBatch

Returns:

The internal vector container

iterator

public Iterator<VectorWrapper<?>> iterator()

Specified by:

iterator in interface Iterable<VectorWrapper<?>>