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));