[hbase10] changes to be able to work with secured hbase cluster

brianfrankcooper · Jan 22, 2016 · 6cd94e8 · 6cd94e8
1 parent 025f3b4
commit 6cd94e8
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/hbase10/src/main/java/com/yahoo/ycsb/db/HBaseClient10.java b/hbase10/src/main/java/com/yahoo/ycsb/db/HBaseClient10.java
@@ -24,6 +24,7 @@
 import com.yahoo.ycsb.Status;
 import com.yahoo.ycsb.measurements.Measurements;
 
+import org.apache.hadoop.security.UserGroupInformation;
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.hbase.Cell;
 import org.apache.hadoop.hbase.CellUtil;
@@ -112,6 +113,22 @@ public void init() throws DBException {
           Durability.valueOf(getProperties().getProperty("durability"));
     }
 
+    if ("kerberos".equalsIgnoreCase(config.get("hbase.security.authentication"))) {
+      config.set("hadoop.security.authentication", "Kerberos");
+      UserGroupInformation.setConfiguration(config);
+    }
+
+    if ((getProperties().getProperty("principal")!=null) 
+        && (getProperties().getProperty("keytab")!=null)) {
+      try {
+        UserGroupInformation.loginUserFromKeytab(getProperties().getProperty("principal"), 
+              getProperties().getProperty("keytab"));
+      } catch (IOException e) {
+        System.err.println("Keytab file is not readable or not found");
+        throw new DBException(e);
+      }
+    }
+
     try {
       connection = ConnectionFactory.createConnection(config);
     } catch (java.io.IOException e) {