diff options
Diffstat (limited to 'src/main')
-rw-r--r-- | src/main/resources/pom.xml | 2 | ||||
-rw-r--r-- | src/main/scala/parallelai/spyglass/base/JobRunner.scala | 12 |
2 files changed, 11 insertions, 3 deletions
diff --git a/src/main/resources/pom.xml b/src/main/resources/pom.xml index e8d4fc0..d0fa51a 100644 --- a/src/main/resources/pom.xml +++ b/src/main/resources/pom.xml @@ -8,7 +8,7 @@ <description>Cascading and Scalding wrapper for HBase with advanced features</description> <groupId>parallelai</groupId> <artifactId>parallelai.spyglass</artifactId> - <version>2.10_0.10_4.3</version> + <version>2.10_0.10_4.4</version> <packaging>jar</packaging> <organization> diff --git a/src/main/scala/parallelai/spyglass/base/JobRunner.scala b/src/main/scala/parallelai/spyglass/base/JobRunner.scala index b3a9af0..3069051 100644 --- a/src/main/scala/parallelai/spyglass/base/JobRunner.scala +++ b/src/main/scala/parallelai/spyglass/base/JobRunner.scala @@ -3,7 +3,10 @@ package parallelai.spyglass.base import org.apache.hadoop.conf.Configuration import com.twitter.scalding.Tool import org.apache.hadoop - +import org.apache.hadoop.hbase.security.token.TokenUtil +import org.apache.hadoop.security.UserGroupInformation +import org.apache.hadoop.hbase.security.User + object JobRunner { def main(args : Array[String]) { val conf: Configuration = new Configuration @@ -17,7 +20,12 @@ object JobRunner { } AppConfig.jobConfig = conf - + + if (User.isHBaseSecurityEnabled(conf)) { + println("Obtaining token for HBase security."); + TokenUtil.obtainAndCacheToken(conf, UserGroupInformation.getCurrentUser()); + } + hadoop.util.ToolRunner.run(conf, new Tool, args); } }
\ No newline at end of file |