From f6d5f6c03557ea3e9dbb34f7c7c2344b314905cc Mon Sep 17 00:00:00 2001 From: Bolke de Bruin Date: Tue, 2 Feb 2016 12:50:17 +0100 Subject: [PATCH] Use KerberosName instead of local implementation --- .../org/apache/hive/service/cli/thrift/ThriftCLIService.java | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java b/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java index 5a0f1c8..ae98e4a 100644 --- a/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java +++ b/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java @@ -31,6 +31,8 @@ import org.apache.commons.logging.LogFactory; import org.apache.hadoop.hive.conf.HiveConf; import org.apache.hadoop.hive.conf.HiveConf.ConfVars; +import org.apache.hadoop.hive.shims.HadoopShims.KerberosNameShim; +import org.apache.hadoop.hive.shims.ShimLoader; import org.apache.hive.service.AbstractService; import org.apache.hive.service.ServiceException; import org.apache.hive.service.ServiceUtils; @@ -343,7 +345,7 @@ private String getIpAddress() { * @return * @throws HiveSQLException */ - private String getUserName(TOpenSessionReq req) throws HiveSQLException { + private String getUserName(TOpenSessionReq req) throws HiveSQLException, IOException { String userName = null; // Kerberos if (isKerberosAuthMode()) { @@ -369,12 +371,12 @@ private String getUserName(TOpenSessionReq req) throws HiveSQLException { return effectiveClientUser; } - private String getShortName(String userName) { + private String getShortName(String userName) throws IOException { String ret = null; + if (userName != null) { - int indexOfDomainMatch = ServiceUtils.indexOfDomainMatch(userName); - ret = (indexOfDomainMatch <= 0) ? userName : - userName.substring(0, indexOfDomainMatch); + KerberosNameShim fullKerberosName = ShimLoader.getHadoopShims().getKerberosNameShim(userName); + ret = fullKerberosName.getShortName(); } return ret; -- 2.5.4 (Apple Git-61)