org.apache.drill.test

Class PhysicalOpUnitTestBase.MockExecutorFragmentContext

java.lang.Object

org.apache.drill.exec.ops.BaseFragmentContext

org.apache.drill.test.OperatorFixture.MockFragmentContext

org.apache.drill.test.PhysicalOpUnitTestBase.MockExecutorFragmentContext

All Implemented Interfaces:

AutoCloseable, ExchangeFragmentContext, ExecutorFragmentContext, FragmentContext, RootFragmentContext, UdfUtilities

Enclosing class:

PhysicalOpUnitTestBase

protected static class PhysicalOpUnitTestBase.MockExecutorFragmentContext
extends OperatorFixture.MockFragmentContext
implements ExecutorFragmentContext

Note

The PhysicalOpUnitTestBase.MockExecutorFragmentContext should only be used in PhysicalOpUnitTestBase because PhysicalOpUnitTestBase needs a dummy ExecutorFragmentContext to be passed to batch creators. If you are unit testing operators and need a mock fragment context please use OperatorFixture.MockFragmentContext.

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.drill.exec.ops.FragmentContext

FragmentContext.ExecutorState

Field Summary

Fields inherited from interface org.apache.drill.exec.ops.UdfUtilities

INJECTABLE_GETTER_METHODS

Constructor Summary

Constructors

Constructor and Description
MockExecutorFragmentContext(FragmentContext fragmentContext)

Method Summary

Modifier and Type	Method and Description
void	addRuntimeFilter(RuntimeFilterWritable runtimeFilter) Add a RuntimeFilter when the RuntimeFilter receiver belongs to the same MinorFragment.
Collection<CoordinationProtos.DrillbitEndpoint>	getBits()
IncomingBuffers	getBuffers()
ClusterCoordinator	getClusterCoordinator()
Controller	getController()
AccountingDataTunnel	getDataTunnel(CoordinationProtos.DrillbitEndpoint endpoint)
CoordinationProtos.DrillbitEndpoint	getEndpoint()
CoordinationProtos.DrillbitEndpoint	getForemanEndpoint()
OperatorCreatorRegistry	getOperatorCreatorRegistry()
PhysicalPlanReader	getPlanReader()
QueryProfileStoreContext	getProfileStoreContext()
BufferAllocator	getRootAllocator() Returns the root allocator for the Drillbit.
RuntimeFilterWritable	getRuntimeFilter(long rfIdentifier)
RuntimeFilterWritable	getRuntimeFilter(long rfIdentifier, long maxWaitTime, TimeUnit timeUnit) Get the RuntimeFilter with a blocking wait, if the waiting option is enabled.
FragmentStats	getStats()
Set<Map.Entry<UserServer.BitToUserConnection,UserServer.BitToUserConnectionConfig>>	getUserConnections()
AccountingUserConnection	getUserDataTunnel()
WorkEventBus	getWorkEventBus()
boolean	isUserAuthenticationEnabled()
void	setBuffers(IncomingBuffers buffers)
void	setExecutorState(FragmentContext.ExecutorState executorState)
void	waitForSendComplete()

Methods inherited from class org.apache.drill.test.OperatorFixture.MockFragmentContext

close, getAliasRegistryProvider, getAllocator, getBufferManager, getCompiler, getConfig, getConstantValueHolder, getContextInformation, getExecutionControls, getExecutor, getExecutorState, getFragIdString, getFullRootSchema, getHandle, getMetastoreRegistry, getNewChildAllocator, getOptions, getPartitionExplorer, getQueryUserName, getScanDecodeExecutor, getScanExecutor, isImpersonationEnabled, newOperatorContext, newOperatorContext, requestMemory

Methods inherited from class org.apache.drill.exec.ops.BaseFragmentContext

getFunctionRegistry, getImplementationClass, getImplementationClass, getImplementationClass, getImplementationClass, getManagedBuffer, getManagedBuffer, getManagedBufferManager, getQueryId, getQueryIdString, getSQLStatementType, replace

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.drill.exec.ops.FragmentContext

close, getAliasRegistryProvider, getAllocator, getCompiler, getConfig, getExecutionControls, getExecutor, getExecutorState, getFragIdString, getFullRootSchema, getFunctionRegistry, getHandle, getImplementationClass, getImplementationClass, getImplementationClass, getImplementationClass, getManagedBuffer, getManagedBuffer, getManagedBufferManager, getMetastoreRegistry, getNewChildAllocator, getOptions, getQueryId, getQueryIdString, getQueryUserName, getScanDecodeExecutor, getScanExecutor, getSQLStatementType, isImpersonationEnabled, newOperatorContext, newOperatorContext, replace, requestMemory

Methods inherited from interface org.apache.drill.exec.ops.UdfUtilities

getConstantValueHolder, getContextInformation, getPartitionExplorer

Constructor Detail

MockExecutorFragmentContext

public MockExecutorFragmentContext(FragmentContext fragmentContext)

Method Detail

getRootAllocator

public BufferAllocator getRootAllocator()

Description copied from interface: ExecutorFragmentContext

Returns the root allocator for the Drillbit.

Specified by:

getRootAllocator in interface ExecutorFragmentContext

getPlanReader

public PhysicalPlanReader getPlanReader()

Specified by:

getPlanReader in interface ExecutorFragmentContext

getClusterCoordinator

public ClusterCoordinator getClusterCoordinator()

Specified by:

getClusterCoordinator in interface ExecutorFragmentContext

getForemanEndpoint

public CoordinationProtos.DrillbitEndpoint getForemanEndpoint()

Specified by:

getForemanEndpoint in interface ExecutorFragmentContext

getEndpoint

public CoordinationProtos.DrillbitEndpoint getEndpoint()

Specified by:

getEndpoint in interface ExecutorFragmentContext

getBits

public Collection<CoordinationProtos.DrillbitEndpoint> getBits()

Specified by:

getBits in interface ExecutorFragmentContext

getOperatorCreatorRegistry

public OperatorCreatorRegistry getOperatorCreatorRegistry()

Specified by:

getOperatorCreatorRegistry in interface ExecutorFragmentContext

setBuffers

public void setBuffers(IncomingBuffers buffers)

Specified by:

setBuffers in interface ExecutorFragmentContext

getProfileStoreContext

public QueryProfileStoreContext getProfileStoreContext()

Specified by:

getProfileStoreContext in interface ExecutorFragmentContext

getWorkEventBus

public WorkEventBus getWorkEventBus()

Specified by:

getWorkEventBus in interface ExecutorFragmentContext

getUserConnections

public Set<Map.Entry<UserServer.BitToUserConnection,UserServer.BitToUserConnectionConfig>> getUserConnections()

Specified by:

getUserConnections in interface ExecutorFragmentContext

waitForSendComplete

public void waitForSendComplete()

Specified by:

waitForSendComplete in interface ExchangeFragmentContext

getDataTunnel

public AccountingDataTunnel getDataTunnel(CoordinationProtos.DrillbitEndpoint endpoint)

Specified by:

getDataTunnel in interface ExchangeFragmentContext

getUserDataTunnel

public AccountingUserConnection getUserDataTunnel()

Specified by:

getUserDataTunnel in interface ExchangeFragmentContext

getController

public Controller getController()

Specified by:

getController in interface ExchangeFragmentContext

getBuffers

public IncomingBuffers getBuffers()

Specified by:

getBuffers in interface ExchangeFragmentContext

getStats

public FragmentStats getStats()

Specified by:

getStats in interface RootFragmentContext

setExecutorState

public void setExecutorState(FragmentContext.ExecutorState executorState)

Specified by:

setExecutorState in interface RootFragmentContext

isUserAuthenticationEnabled

public boolean isUserAuthenticationEnabled()

Specified by:

isUserAuthenticationEnabled in interface ExecutorFragmentContext

addRuntimeFilter

public void addRuntimeFilter(RuntimeFilterWritable runtimeFilter)

Description copied from interface: FragmentContext

Add a RuntimeFilter when the RuntimeFilter receiver belongs to the same MinorFragment.

Specified by:

addRuntimeFilter in interface FragmentContext

Overrides:

addRuntimeFilter in class OperatorFixture.MockFragmentContext

Parameters:

runtimeFilter - runtime filter

getRuntimeFilter

public RuntimeFilterWritable getRuntimeFilter(long rfIdentifier)

Specified by:

getRuntimeFilter in interface FragmentContext

Overrides:

getRuntimeFilter in class OperatorFixture.MockFragmentContext

getRuntimeFilter

public RuntimeFilterWritable getRuntimeFilter(long rfIdentifier,
                                              long maxWaitTime,
                                              TimeUnit timeUnit)

Description copied from interface: FragmentContext

Get the RuntimeFilter with a blocking wait, if the waiting option is enabled.

Specified by:

getRuntimeFilter in interface FragmentContext

Overrides:

getRuntimeFilter in class OperatorFixture.MockFragmentContext

Parameters:

rfIdentifier - runtime filter identifier

maxWaitTime - max wait time

timeUnit - time unit

Returns:

the RFW or null