apache/kyuubi
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

use session ugi to collect tokens

Browse Source
This commit is contained in:
Kent Yao 2018-11-16 14:02:31 +08:00
parent 6368cb1ebe
commit e5ed088181
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
kyuubi-server/src/main/scala/yaooqinn/kyuubi/session/KyuubiSession.scala
View File

@ -36,6 +36,7 @@ import yaooqinn.kyuubi.operation.{KyuubiOperation, OperationHandle, OperationMan
import yaooqinn.kyuubi.schema.RowSet
import yaooqinn.kyuubi.session.security.TokenCollector
import yaooqinn.kyuubi.spark.SparkSessionWithUGI
import yaooqinn.kyuubi.utils.KyuubiHadoopUtil
/**
* An Execution Session with [[SparkSession]] instance inside, which shares [[SparkContext]]
@ -75,7 +76,7 @@ private[kyuubi] class KyuubiSession(
currentUser.reloginFromKeytab()
}
val user = UserGroupInformation.createProxyUser(username, currentUser)
TokenCollector.obtainTokenIfRequired(conf, user.getCredentials)
KyuubiHadoopUtil.doAs(user)(TokenCollector.obtainTokenIfRequired(conf, user.getCredentials))
user
} else {
UserGroupInformation.createRemoteUser(username)
@ -139,7 +140,9 @@ private[kyuubi] class KyuubiSession(
def ugi: UserGroupInformation = {
val creds = new Credentials
TokenCollector.obtainTokenIfRequired(conf, creds)
KyuubiHadoopUtil.doAs(sessionUGI) {
TokenCollector.obtainTokenIfRequired(conf, creds)
}
sessionUGI.addCredentials(creds)
sessionUGI
}