How to use

setImpersonator

method

in

org.apache.sentry.core.common.utils.ThriftUtil

 @Override
 public boolean process(TProtocol in, TProtocol out) throws TException {
  ThriftUtil.setIpAddress(in);
  ThriftUtil.setImpersonator(in);
  return super.process(in, out);
 }
}

 @Override
 public boolean process(TProtocol in, TProtocol out) throws TException {
  // set the ip and impersonator for audit log
  ThriftUtil.setIpAddress(in);
  ThriftUtil.setImpersonator(in);
  return super.process(in, out);
 }
}

public static void setImpersonator(final TProtocol in) {
 try {
  TTransport transport = in.getTransport();
  if (transport instanceof TSaslServerTransport) {
   String impersonator = ((TSaslServerTransport) transport).getSaslServer()
     .getAuthorizationID();
   setImpersonator(impersonator);
  }
 } catch (Exception e) {
  // If there has exception when get impersonator info, log the error information.
  LOGGER.warn("There is an error when get the impersonator:" + e.getMessage());
 }
}

@BeforeClass
public static void init() {
 conf = new Configuration();
 conf.set(ServerConfig.SENTRY_SERVICE_NAME, ServerConfig.SENTRY_SERVICE_NAME_DEFAULT);
 ThriftUtil.setIpAddress(TEST_IP);
 ThriftUtil.setImpersonator(TEST_IMPERSONATOR);
 TEST_PRIVILEGES_MAP.put("resourceType1", "resourceName1");
 TEST_PRIVILEGES_MAP.put("resourceType2", "resourceName2");
 TEST_PRIVILEGES_MAP.put("resourceType3", "resourceName3");
}

@BeforeClass
public static void init() {
 conf = new Configuration();
 conf.set(ServerConfig.SENTRY_SERVICE_NAME,
   ServerConfig.SENTRY_SERVICE_NAME_DEFAULT);
 ThriftUtil.setIpAddress(TEST_IP);
 ThriftUtil.setImpersonator(TEST_IMPERSONATOR);
}