org.apache.drill.exec.rpc.user

Class UserClient

java.lang.Object

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

org.apache.drill.exec.rpc.BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

org.apache.drill.exec.rpc.user.UserClient

All Implemented Interfaces:

Closeable, AutoCloseable

public class UserClient
extends BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	UserClient.UserToBitConnection

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

connection

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

local, remote, rpcConfig

Constructor Summary

Constructors

Constructor and Description
UserClient(String clientName, DrillConfig config, Properties properties, boolean supportComplexTypes, BufferAllocator allocator, io.netty.channel.EventLoopGroup eventLoopGroup, Executor eventExecutor, CoordinationProtos.DrillbitEndpoint endpoint)

Method Summary

Methods

Modifier and Type	Method and Description
void	connect(CoordinationProtos.DrillbitEndpoint endpoint, DrillProperties properties, UserBitShared.UserCredentials credentials) Connects, and if required, authenticates.
ProtobufLengthDecoder	getDecoder(BufferAllocator allocator)
protected com.google.protobuf.MessageLite	getResponseDefaultInstance(int rpcType)
UserProtos.RpcEndpointInfos	getServerInfos()
Set<UserProtos.RpcType>	getSupportedMethods()
protected void	handle(UserClient.UserToBitConnection connection, int rpcType, ByteBuf pBody, ByteBuf dBody, ResponseSender sender)
protected UserClient.UserToBitConnection	initRemoteConnection(io.netty.channel.socket.SocketChannel channel)
protected boolean	isSslEnabled()
DrillRpcFuture<UserProtos.QueryPlanFragments>	planQuery(UserProtos.GetQueryPlanFragments req) planQuery is an API to plan a query without query execution
protected <SEND extends com.google.protobuf.MessageLite,RECEIVE extends com.google.protobuf.MessageLite> void	send(RpcOutcomeListener<RECEIVE> listener, UserProtos.RpcType rpcType, SEND protobufBody, Class<RECEIVE> clazz, boolean allowInEventLoop, ByteBuf... dataBodies)
protected void	setupSSL(io.netty.channel.ChannelPipeline pipe, ConnectionMultiListener.SSLHandshakeListener sslHandshakeListener)
void	submitQuery(UserResultsListener resultsListener, UserProtos.RunQuery query)
protected void	validateHandshake(UserProtos.BitToUserHandshake inbound)

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

close, connectAsClient, finalizeConnection, getInitialCommand, isActive, send, send, send, setAutoRead, setSslChannel

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

get, getCloseHandler, send, send, send, setAddresses

Methods inherited from class java.lang.Object

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

Constructor Detail

UserClient

public UserClient(String clientName,
          DrillConfig config,
          Properties properties,
          boolean supportComplexTypes,
          BufferAllocator allocator,
          io.netty.channel.EventLoopGroup eventLoopGroup,
          Executor eventExecutor,
          CoordinationProtos.DrillbitEndpoint endpoint)
           throws NonTransientRpcException

Throws:

NonTransientRpcException

Method Detail

setupSSL

protected void setupSSL(io.netty.channel.ChannelPipeline pipe,
            ConnectionMultiListener.SSLHandshakeListener sslHandshakeListener)

Overrides:

setupSSL in class BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

isSslEnabled

protected boolean isSslEnabled()

Overrides:

isSslEnabled in class BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

getServerInfos

public UserProtos.RpcEndpointInfos getServerInfos()

getSupportedMethods

public Set<UserProtos.RpcType> getSupportedMethods()

submitQuery

public void submitQuery(UserResultsListener resultsListener,
               UserProtos.RunQuery query)

connect

public void connect(CoordinationProtos.DrillbitEndpoint endpoint,
           DrillProperties properties,
           UserBitShared.UserCredentials credentials)
             throws RpcException

Connects, and if required, authenticates. This method blocks until both operations are complete.

Parameters:

endpoint - endpoint to connect to

properties - properties

credentials - credentials

Throws:

RpcException - if either connection or authentication fails

send

protected <SEND extends com.google.protobuf.MessageLite,RECEIVE extends com.google.protobuf.MessageLite> void send(RpcOutcomeListener<RECEIVE> listener,
                                                                                                       UserProtos.RpcType rpcType,
                                                                                                       SEND protobufBody,
                                                                                                       Class<RECEIVE> clazz,
                                                                                                       boolean allowInEventLoop,
                                                                                                       ByteBuf... dataBodies)

getResponseDefaultInstance

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

Specified by:

getResponseDefaultInstance in class RpcBus<UserProtos.RpcType,UserClient.UserToBitConnection>

Throws:

RpcException

handle

protected void handle(UserClient.UserToBitConnection connection,
          int rpcType,
          ByteBuf pBody,
          ByteBuf dBody,
          ResponseSender sender)
               throws RpcException

Specified by:

handle in class RpcBus<UserProtos.RpcType,UserClient.UserToBitConnection>

Throws:

RpcException

validateHandshake

protected void validateHandshake(UserProtos.BitToUserHandshake inbound)
                          throws RpcException

Specified by:

validateHandshake in class BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

Throws:

RpcException

initRemoteConnection

protected UserClient.UserToBitConnection initRemoteConnection(io.netty.channel.socket.SocketChannel channel)

Overrides:

initRemoteConnection in class BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

getDecoder

public ProtobufLengthDecoder getDecoder(BufferAllocator allocator)

Specified by:

getDecoder in class BasicClient<UserProtos.RpcType,UserClient.UserToBitConnection,UserProtos.UserToBitHandshake,UserProtos.BitToUserHandshake>

planQuery

public DrillRpcFuture<UserProtos.QueryPlanFragments> planQuery(UserProtos.GetQueryPlanFragments req)

planQuery is an API to plan a query without query execution

Parameters:

req - - data necessary to plan query

Returns:

list of PlanFragments that can later on be submitted for execution