1 files changed, 10 insertions, 0 deletions
diff --git a/src/main/java/parallelai/spyglass/hbase/HBaseTap.java b/src/main/java/parallelai/spyglass/hbase/HBaseTap.java
index 3576e96..e4a3f78 100644
--- a/src/main/java/parallelai/spyglass/hbase/HBaseTap.java
+++ b/src/main/java/parallelai/spyglass/hbase/HBaseTap.java
@@ -404,9 +404,11 @@ public class HBaseTap extends Tap<JobConf, RecordReader, OutputCollector> {
   private static class SinkConfig implements Serializable {
 	  public String tableName = null;
 	  public boolean useSalt = false;
+	  public boolean autoFlush = false;
 	  
 	  public void configure(Configuration jobConf) {
           jobConf.setBoolean(String.format(HBaseConstants.USE_SALT, tableName), useSalt);
+          jobConf.setBoolean(String.format(HBaseConstants.AUTO_FLUSH, tableName), autoFlush);
 	  }
   }
   
@@ -459,6 +461,14 @@ public class HBaseTap extends Tap<JobConf, RecordReader, OutputCollector> {
 	
     sinkConfigList.add(sc);
   }
+
+  public void setHBaseSinkParms( boolean useSalt, boolean autoFlush ) {
+    SinkConfig sc = new SinkConfig();
+    sc.tableName = tableName;
+    sc.autoFlush = autoFlush;
+    sc.useSalt = useSalt;
+    sinkConfigList.add(sc);
+  }
   
   private void setPrefixList(SourceConfig sc, String prefixList ) {
 	  prefixList = (prefixList == null || prefixList.length() == 0) ? HBaseSalter.DEFAULT_PREFIX_LIST : prefixList;