org.apache.drill.exec.rpc.control

Class ControlClient

java.lang.Object

org.apache.drill.exec.rpc.RpcBus<T,CC>

org.apache.drill.exec.rpc.BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>

org.apache.drill.exec.rpc.control.ControlClient

All Implemented Interfaces:

Closeable, AutoCloseable

public class ControlClient
extends BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.drill.exec.rpc.RpcBus

RpcBus.ChannelClosedHandler, RpcBus.InboundHandler

Field Summary

Fields inherited from class org.apache.drill.exec.rpc.RpcBus

local, remote, rpcConfig

Constructor Summary

Constructors

Constructor and Description
ControlClient(ControlConnectionConfig config, CoordinationProtos.DrillbitEndpoint remoteEndpoint, ReconnectingConnection.CloseHandlerCreator closeHandlerFactory)

Method Summary

Modifier and Type	Method and Description
protected void	finalizeConnection(BitControl.BitControlHandshake handshake, ControlConnection connection)
protected io.netty.util.concurrent.GenericFutureListener<io.netty.channel.ChannelFuture>	getCloseHandler(io.netty.channel.socket.SocketChannel ch, ControlConnection clientConnection)
ProtobufLengthDecoder	getDecoder(BufferAllocator allocator)
com.google.protobuf.MessageLite	getResponseDefaultInstance(int rpcType)
protected void	handle(ControlConnection connection, int rpcType, io.netty.buffer.ByteBuf pBody, io.netty.buffer.ByteBuf dBody, ResponseSender sender)
protected ControlConnection	initRemoteConnection(io.netty.channel.socket.SocketChannel channel)
protected void	prepareSaslHandshake(RpcConnectionHandler<ControlConnection> connectionHandler, List<String> serverAuthMechanisms) Creates various instances needed to start the SASL handshake.
protected List<String>	validateHandshake(BitControl.BitControlHandshake handshake)

Methods inherited from class org.apache.drill.exec.rpc.BasicClient

close, connectAsClient, isActive, isAuthComplete, isSslEnabled, send, send, send, setAuthComplete, setAutoRead, setSslChannel, setupSSL, startSaslHandshake

Methods inherited from class org.apache.drill.exec.rpc.RpcBus

get, send, send, send, setAddresses

Methods inherited from class java.lang.Object

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

Constructor Detail

ControlClient

public ControlClient(ControlConnectionConfig config,
                     CoordinationProtos.DrillbitEndpoint remoteEndpoint,
                     ReconnectingConnection.CloseHandlerCreator closeHandlerFactory)

Method Detail

initRemoteConnection

protected ControlConnection initRemoteConnection(io.netty.channel.socket.SocketChannel channel)

Overrides:

initRemoteConnection in class BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>

getCloseHandler

protected io.netty.util.concurrent.GenericFutureListener<io.netty.channel.ChannelFuture> getCloseHandler(io.netty.channel.socket.SocketChannel ch,
                                                                                                         ControlConnection clientConnection)

Overrides:

getCloseHandler in class RpcBus<BitControl.RpcType,ControlConnection>

getResponseDefaultInstance

public com.google.protobuf.MessageLite getResponseDefaultInstance(int rpcType)
                                                           throws RpcException

Specified by:

getResponseDefaultInstance in class RpcBus<BitControl.RpcType,ControlConnection>

Throws:

RpcException

handle

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

Specified by:

handle in class RpcBus<BitControl.RpcType,ControlConnection>

Throws:

RpcException

prepareSaslHandshake

protected void prepareSaslHandshake(RpcConnectionHandler<ControlConnection> connectionHandler,
                                    List<String> serverAuthMechanisms)

Description copied from class: BasicClient

Creates various instances needed to start the SASL handshake. This is called from BasicClient.validateHandshake(MessageLite) if authentication is required from server side.

Specified by:

prepareSaslHandshake in class BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>

Parameters:

connectionHandler - - Connection handler used by client's to know about success/failure conditions.

serverAuthMechanisms - - List of auth mechanisms configured on server side

validateHandshake

protected List<String> validateHandshake(BitControl.BitControlHandshake handshake)
                                  throws RpcException

Specified by:

validateHandshake in class BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>

Throws:

RpcException

finalizeConnection

protected void finalizeConnection(BitControl.BitControlHandshake handshake,
                                  ControlConnection connection)

Overrides:

finalizeConnection in class BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>

getDecoder

public ProtobufLengthDecoder getDecoder(BufferAllocator allocator)

Specified by:

getDecoder in class BasicClient<BitControl.RpcType,ControlConnection,BitControl.BitControlHandshake,BitControl.BitControlHandshake>