dremio jdbc 客户端实际上包含了基本上两大类,一个是传统jdbc 的,一个是基于apache arrow flight sql jdbc 的
当前主要说明传统jdbc 客户端的,内部上dremio 基于了calcite 的avatica 进行jdbc driver 的包装,实际客户端的
链接处理是基于了dremio 自己包装的java client (通过rpc 服务,直接的是DremioClient,内部实际上调用的是UserClient)
关于avatica 部分就不说明了,主要说明下rpc 关联部分
DremioClient 提供的能力
UserClient 提供的能力
内部网络部分是基于了netty,从下图可以看出依赖了BasicClient以及RpcBus,jdbc 消息处理以及查询执行,就是基于此提供的包装能力
比如处理查询,连接管理,处理响应
dremio 对于client rpc 的处理基于netty,自己实现了编码器以及解码器以及消息处理的handler
BasicClient 构造函数部分
b = new Bootstrap()
.group(eventLoopGroup)
.channel(TransportCheck.getClientSocketChannel())
.option(ChannelOption.ALLOCATOR, alloc)
.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 30 * 1000)
.option(ChannelOption.SO_REUSEADDR, true)
.option(ChannelOption.SO_RCVBUF, SO_BUF_SZ)
.option(ChannelOption.SO_SNDBUF, SO_BUF_SZ)
.option(ChannelOption.TCP_NODELAY, true)
.handler(new ChannelInitializer<SocketChannel>() {
@Override
protected void initChannel(SocketChannel ch) throws Exception {
connection = initRemoteConnection(ch);
// each handler in the pipeline is created per connection
ch.closeFuture().addListener(newCloseListener(ch, connection));
final ChannelPipeline pipe = ch.pipeline();
// rpc 编解码器
pipe.addLast(PROTOCOL_ENCODER, new RpcEncoder("c-" + rpcConfig.getName()));
pipe.addLast(PROTOCOL_DECODER, newDecoder(connection.getAllocator()));
pipe.addLast(HANDSHAKE_HANDLER, new ClientHandshakeHandler());
if (timeoutInMillis != -1) {
pipe.addLast(IDLE_STATE_HANDLER, new IdlePingHandler(timeoutInMillis));
}
// 消息处理handler
pipe.addLast(MESSAGE_HANDLER, new InboundHandler(connection));
pipe.addLast(EXCEPTION_HANDLER, new RpcExceptionHandler<>(connection));
}
});
当进入rpc服务之后,后续的就是标准的sql 处理了,包含了sql 校验,关系转换,逻辑计划,物理计划,计划执行。。。
参考部分执行调用图
说明
以上只是一个简单的说明,实际内部还是很复杂的,毕竟需要关联整个dremio 的东西,
jdbc 处理部分基于了protobuf 处理消息 protocol以及services/base-rpc 模块里边关于消息是定义是值得看看的
参考资料
client/jdbc/src/main/java/com/dremio/jdbc/impl/DremioJdbc41Factory.java
client/jdbc/src/main/java/com/dremio/jdbc/impl/DremioConnectionImpl.java
client/base/src/main/java/com/dremio/exec/client/DremioClient.java
services/coordinator/src/main/java/com/dremio/service/coordinator/ClusterCoordinator.java
services/base-rpc/src/main/java/com/dremio/exec/rpc/RpcBus.java