Hadoop tips - 如何获得客户请求的来源IP

Hadoop服务端所有的客户请求(除了HTTP接口)都使用RPC框架的Server来实现.

RPC Server处理请求的是Handler线程, 对每一个请求(Call对象, 包含了客户来源IP,端口等信息)的处理过程中包含如下代码

CurCall.set(call);

其中CurCall是一个ThreadLocal对象. 在处理的过程中如果要获得客户的IP, 可以调用如下方法, RPC Server已经实现了该方法:

  public static InetAddress getRemoteIp() {
    Call call = CurCall.get();
    if (call != null) {
      return call.connection.socket.getInetAddress();
    }
    return null;
  }

Hadoop在其他地方也用过ThreadLocal对象, 比如保存用户信息的UserGroupInformation类, 我们可以设置当前请求的UGI:

UserGroupInformation.setCurrentUGI(call.connection.ticket);

然后在Handler线程的其他地方就能得到当前用户信息:

UserGroupInformation.getCurrentUGI();