hbase客户端协议修改

时间:2023-01-31 08:31:04

hbase 的客户端 协议修改

hbase 的客户端通过 RpcClientImpl 进行远程的rpc调用

客户端连接到远程是通过RpcClientImpl 进行管理的,通过创建 createSocket 进行连接到远程的服务端
通过 connectionmanager.getStubKey 拿到远程链接,创建到远程对应的ip端口

 NetUtils.connect(this.socket, remoteId.getAddress(), connectTO)

然后把数据写到远端去

IPCUtil.write(this.out, header, call.param, cellBlock);

默认上传用户信息到远端 AuthMethod.SIMPLE在ConnectionId 中有个ticket,
这个ticket就是一个user类对象,保存了客户端的信息,UserGroupInformation 中进行拿取,
并可以根据不同的auth方法,进行不同的权限校验,支持 DIGEST,KERBEROS 和SIMPLE
可以直接修改 UserGroupInformation 里面拿用户信息的类 ,替换掉 KERBEROS中拿取的信息
通过客户端创建的 socket ,传输protobuf的协议数据到远端中。如

IPCUtil.write(this.out, header, call.param, cellBlock);

远程一般为hmaster和regionserver。
可修改 /hbase-protocol/src/main/protobuf 下面的proto文件,就可以修改了调用请求接口了。
window下面可以通过

mvn compile -Dcompile-protobuf -Dprotoc.path=E:/realtime/protoc/protoc-2.5.0-win32/protoc.exe 

生成相关的新类
/hbase-protocol/src/main/protobuf/Master.proto

   message GetTableDescriptorsRequest {
      repeated TableName table_names = 1;
      optional string regex = 2;
      optional bool include_sys_tables = 3 [default=false];
      optional string namespace = 4;
      optional string localip = 5;
      optional string user = 6;
      optional string password = 7;
    }

执行上面的命令后,就可以增加接口参数了。

然后在类 RequestConverter 中可以增加请求的客户端信息

  /**
   * Creates a protocol buffer GetTableDescriptorsRequest for a single table
   *增加请求的本地用户的信息
   * @param tableName the table name
   * @return a GetTableDescriptorsRequest
   */
  public static GetTableDescriptorsRequest buildGetTableDescriptorsRequest(
  final TableName tableName) {
return GetTableDescriptorsRequest.newBuilder()
  .addTableNames(ProtobufUtil.toProtoTableName(tableName)).setLocalip(ClientUtils.getLocalHost())
  .setUser(ClientUtils.getLocalUser()).setPassword(ClientUtils.getLocalPassowrd())
  .build();
  }

在服务端 MasterRpcServices.getTableDescriptors中就可以拿到请求的信息了

  @Override
  public GetTableDescriptorsResponse getTableDescriptors(RpcController c,
  GetTableDescriptorsRequest req) throws ServiceException {
try {
  master.checkInitialized();

  String remoteip = req.getLocalip();
  String user = req.getUser();
  String password = req.getPassword();
  LOG.info(String.format("get tableDescriptors remoteip=[%s] ,user=[%s] , regex=[%s]",remoteip,user, password));