org.apache.drill.exec.work.batch

Class ControlMessageHandler

java.lang.Object

org.apache.drill.exec.work.batch.ControlMessageHandler

All Implemented Interfaces:

RequestHandler<ControlConnection>

public class ControlMessageHandler
extends Object
implements RequestHandler<ControlConnection>

Constructor Summary

Constructors

Constructor and Description
ControlMessageHandler(WorkManager.WorkerBee bee)

Method Summary

Modifier and Type	Method and Description
GeneralRPCProtos.Ack	cancelFragment(ExecProtos.FragmentHandle handle)
CustomHandlerRegistry	getHandlerRegistry()
void	handle(ControlConnection connection, int rpcType, io.netty.buffer.ByteBuf pBody, io.netty.buffer.ByteBuf dBody, ResponseSender sender) Handle request of given type (rpcType) with message (pBody) and optional data (dBody) on the connection, and return the appropriate response.
GeneralRPCProtos.Ack	initializeFragment(BitControl.InitializeFragments fragments)
GeneralRPCProtos.Ack	receivingFragmentFinished(BitControl.FinishedReceiver finishedReceiver)
GeneralRPCProtos.Ack	requestFragmentStatus(BitControl.FragmentStatus status)
GeneralRPCProtos.Ack	requestQueryCancel(UserBitShared.QueryId queryId)
UserBitShared.QueryProfile	requestQueryStatus(UserBitShared.QueryId queryId)
GeneralRPCProtos.Ack	resumeFragment(ExecProtos.FragmentHandle handle)
void	startNewFragment(BitControl.PlanFragment fragment, DrillbitContext drillbitContext) Start a new fragment on this node.

Methods inherited from class java.lang.Object

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

Constructor Detail

ControlMessageHandler

public ControlMessageHandler(WorkManager.WorkerBee bee)

Method Detail

handle

public void handle(ControlConnection connection,
                   int rpcType,
                   io.netty.buffer.ByteBuf pBody,
                   io.netty.buffer.ByteBuf dBody,
                   ResponseSender sender)
            throws RpcException

Description copied from interface: RequestHandler

Handle request of given type (rpcType) with message (pBody) and optional data (dBody) on the connection, and return the appropriate response. The method must do one of three things: + use send the response + throw UserRpcException, in which case a response will be sent using send + throw an Exception, in which case, the connection will be dropped

Specified by:

handle in interface RequestHandler<ControlConnection>

Parameters:

connection - remote connection

rpcType - rpc type

pBody - message

dBody - data, maybe null

sender - used to send the response

Throws:

RpcException

startNewFragment

public void startNewFragment(BitControl.PlanFragment fragment,
                             DrillbitContext drillbitContext)
                      throws UserRpcException

Start a new fragment on this node. These fragments can be leaf or intermediate fragments which are scheduled by remote or local Foreman node.

Parameters:

fragment -

Throws:

UserRpcException

cancelFragment

public GeneralRPCProtos.Ack cancelFragment(ExecProtos.FragmentHandle handle)

resumeFragment

public GeneralRPCProtos.Ack resumeFragment(ExecProtos.FragmentHandle handle)

receivingFragmentFinished

public GeneralRPCProtos.Ack receivingFragmentFinished(BitControl.FinishedReceiver finishedReceiver)

requestFragmentStatus

public GeneralRPCProtos.Ack requestFragmentStatus(BitControl.FragmentStatus status)

requestQueryCancel

public GeneralRPCProtos.Ack requestQueryCancel(UserBitShared.QueryId queryId)

initializeFragment

public GeneralRPCProtos.Ack initializeFragment(BitControl.InitializeFragments fragments)
                                        throws RpcException

Throws:

RpcException

requestQueryStatus

public UserBitShared.QueryProfile requestQueryStatus(UserBitShared.QueryId queryId)
                                              throws RpcException

Throws:

RpcException

getHandlerRegistry

public CustomHandlerRegistry getHandlerRegistry()