1 files changed, 10 insertions, 2 deletions
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