org.apache.drill.exec.ops

Class BaseOperatorContext

java.lang.Object

org.apache.drill.exec.ops.BaseOperatorContext

All Implemented Interfaces:

OperatorContext

Direct Known Subclasses:

OperatorFixture.MockOperatorContext

public abstract class BaseOperatorContext
extends Object
implements OperatorContext

Implementation of OperatorContext that provides services needed by most run-time operators. Excludes services that need the entire Drillbit. This class provides services common to the test-time version of the operator context and the full production-time context that includes network services.

Field Summary

Fields

Modifier and Type	Field and Description
protected BufferAllocator	allocator
protected FragmentContext	context
protected BufferManager	manager
protected PhysicalOperator	popConfig

Constructor Summary

Constructors

Constructor and Description
BaseOperatorContext(FragmentContext context, BufferAllocator allocator, PhysicalOperator popConfig)

Method Summary

Modifier and Type	Method and Description
void	close()
BufferAllocator	getAllocator() Return the memory allocator for this operator.
ExecutionControls	getExecutionControls()
ExecutorService	getExecutor()
FragmentContext	getFragmentContext()
ControlsInjector	getInjector() Returns the fault injection mechanism used to introduce faults at runtime for testing.
DrillBuf	getManagedBuffer()
DrillBuf	getManagedBuffer(int size)
String	getName()
<T extends PhysicalOperator> T	getOperatorDefn() Return the physical operator definition created by the planner and passed into the Drillbit executing the query.
ExecutorService	getScanDecodeExecutor()
ExecutorService	getScanExecutor()
<T extends Throwable> void	injectChecked(String desc, Class<T> exceptionClass) Insert a checked fault (exception) of the given class.
void	injectUnchecked(String desc) Insert an unchecked fault (exception).
DrillFileSystem	newFileSystem(org.apache.hadoop.conf.Configuration conf) Creates DrillFileSystem that automatically tracks operator stats.
DrillFileSystem	newNonTrackingFileSystem(org.apache.hadoop.conf.Configuration conf) Creates a DrillFileSystem that does not automatically track operator stats.
DrillBuf	replace(DrillBuf old, int newSize)
void	setInjector(ControlsInjector injector)

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.OperatorContext

getStats, runCallableAs

Field Detail

context

protected final FragmentContext context

allocator

protected final BufferAllocator allocator

popConfig

protected final PhysicalOperator popConfig

manager

protected final BufferManager manager

Constructor Detail

BaseOperatorContext

public BaseOperatorContext(FragmentContext context,
                           BufferAllocator allocator,
                           PhysicalOperator popConfig)

Method Detail

getFragmentContext

public FragmentContext getFragmentContext()

Specified by:

getFragmentContext in interface OperatorContext

getOperatorDefn

public <T extends PhysicalOperator> T getOperatorDefn()

Description copied from interface: OperatorContext

Return the physical operator definition created by the planner and passed into the Drillbit executing the query.

Specified by:

getOperatorDefn in interface OperatorContext

Returns:

the physical operator definition

getName

public String getName()

replace

public DrillBuf replace(DrillBuf old,
                        int newSize)

Specified by:

replace in interface OperatorContext

getManagedBuffer

public DrillBuf getManagedBuffer()

Specified by:

getManagedBuffer in interface OperatorContext

getManagedBuffer

public DrillBuf getManagedBuffer(int size)

Specified by:

getManagedBuffer in interface OperatorContext

getExecutionControls

public ExecutionControls getExecutionControls()

Specified by:

getExecutionControls in interface OperatorContext

getAllocator

public BufferAllocator getAllocator()

Description copied from interface: OperatorContext

Return the memory allocator for this operator.

Specified by:

getAllocator in interface OperatorContext

Returns:

the per-operator memory allocator

getExecutor

public ExecutorService getExecutor()

Specified by:

getExecutor in interface OperatorContext

getScanExecutor

public ExecutorService getScanExecutor()

Specified by:

getScanExecutor in interface OperatorContext

getScanDecodeExecutor

public ExecutorService getScanDecodeExecutor()

Specified by:

getScanDecodeExecutor in interface OperatorContext

setInjector

public void setInjector(ControlsInjector injector)

Specified by:

setInjector in interface OperatorContext

getInjector

public ControlsInjector getInjector()

Description copied from interface: OperatorContext

Returns the fault injection mechanism used to introduce faults at runtime for testing.

Specified by:

getInjector in interface OperatorContext

Returns:

the fault injector

injectUnchecked

public void injectUnchecked(String desc)

Description copied from interface: OperatorContext

Insert an unchecked fault (exception). Handles the details of checking if fault injection is enabled and this particular fault is selected.

Specified by:

injectUnchecked in interface OperatorContext

Parameters:

desc - the description of the fault used to match a fault injection parameter to determine if the fault should be injected

injectChecked

public <T extends Throwable> void injectChecked(String desc,
                                                Class<T> exceptionClass)
                                         throws T extends Throwable

Description copied from interface: OperatorContext

Insert a checked fault (exception) of the given class. Handles the details of checking if fault injection is enabled and this particular fault is selected.

Specified by:

injectChecked in interface OperatorContext

Parameters:

desc - the description of the fault used to match a fault injection parameter to determine if the fault should be injected

exceptionClass - the class of exeception to be thrown

Throws:

T - if the fault is enabled

T extends Throwable

close

public void close()

Specified by:

close in interface OperatorContext

newFileSystem

public DrillFileSystem newFileSystem(org.apache.hadoop.conf.Configuration conf)
                              throws IOException

Creates DrillFileSystem that automatically tracks operator stats. Only one tracking and no non-tracking file system per operator context.

Specified by:

newFileSystem in interface OperatorContext

Throws:

IOException

newNonTrackingFileSystem

public DrillFileSystem newNonTrackingFileSystem(org.apache.hadoop.conf.Configuration conf)
                                         throws IOException

Creates a DrillFileSystem that does not automatically track operator stats. Multiple non-tracking file system are allowed.

Specified by:

newNonTrackingFileSystem in interface OperatorContext

Throws:

IOException