org.apache.drill.exec.rpc.data

Class TestBitRpc.MockFragmentManager

java.lang.Object

org.apache.drill.exec.rpc.data.TestBitRpc.MockFragmentManager

All Implemented Interfaces:

FragmentManager

Enclosing class:

TestBitRpc

public static class TestBitRpc.MockFragmentManager
extends Object
implements FragmentManager

Constructor Summary

Constructors

Constructor and Description
MockFragmentManager(BootStrapContext c)

Method Summary

Modifier and Type	Method and Description
void	cancel()
FragmentContext	getFragmentContext()
ExecProtos.FragmentHandle	getHandle()
FragmentExecutor	getRunnable() Get the fragment runner for this incoming fragment.
boolean	handle(IncomingDataBatch batch) Handle the next incoming record batch.
boolean	isCancelled() Find out if the FragmentManager has been cancelled.
boolean	isWaiting()
void	receivingFragmentFinished(ExecProtos.FragmentHandle handle)
void	unpause() If the executor is paused (for testing), this method should unpause the executor.

Methods inherited from class java.lang.Object

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

Constructor Detail

MockFragmentManager

public MockFragmentManager(BootStrapContext c)

Method Detail

handle

public boolean handle(IncomingDataBatch batch)
               throws FragmentSetupException,
                      IOException

Description copied from interface: FragmentManager

Handle the next incoming record batch.

Specified by:

handle in interface FragmentManager

Returns:

True if the fragment has enough incoming data to be able to be run.

Throws:

FragmentSetupException

IOException

getRunnable

public FragmentExecutor getRunnable()

Description copied from interface: FragmentManager

Get the fragment runner for this incoming fragment. Note, this can only be requested once.

Specified by:

getRunnable in interface FragmentManager

Returns:

cancel

public void cancel()

Specified by:

cancel in interface FragmentManager

isCancelled

public boolean isCancelled()

Description copied from interface: FragmentManager

Find out if the FragmentManager has been cancelled.

Specified by:

isCancelled in interface FragmentManager

Returns:

true if the FragmentManager has been cancelled.

unpause

public void unpause()

Description copied from interface: FragmentManager

If the executor is paused (for testing), this method should unpause the executor. This method should handle multiple calls.

Specified by:

unpause in interface FragmentManager

isWaiting

public boolean isWaiting()

Specified by:

isWaiting in interface FragmentManager

getHandle

public ExecProtos.FragmentHandle getHandle()

Specified by:

getHandle in interface FragmentManager

getFragmentContext

public FragmentContext getFragmentContext()

Specified by:

getFragmentContext in interface FragmentManager

receivingFragmentFinished

public void receivingFragmentFinished(ExecProtos.FragmentHandle handle)

Specified by:

receivingFragmentFinished in interface FragmentManager