1 files changed, 97 insertions, 0 deletions

diff --git a/src/main/java/parallelai/spyglass/hbase/HBaseTableSplitGranular.java b/src/main/java/parallelai/spyglass/hbase/HBaseTableSplitGranular.java
new file mode 100644
index 0000000..4de7153
--- /dev/null
+++ b/src/main/java/parallelai/spyglass/hbase/HBaseTableSplitGranular.java
@@ -0,0 +1,97 @@
+package parallelai.spyglass.hbase;
+
+import java.io.DataInput;
+import java.io.DataOutput;
+import java.io.IOException;
+import java.io.Serializable;
+import java.util.TreeSet;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.apache.hadoop.hbase.HConstants;
+import org.apache.hadoop.hbase.util.Bytes;
+import org.apache.hadoop.mapred.InputSplit;
+
+import parallelai.spyglass.hbase.HBaseConstants.SourceMode;
+
+public class HBaseTableSplitGranular extends HBaseTableSplitBase {
+
+	private final Log LOG = LogFactory.getLog(HBaseTableSplitGranular.class);
+
+    /** default constructor */
+    public HBaseTableSplitGranular() {
+        this(HConstants.EMPTY_BYTE_ARRAY, HConstants.EMPTY_BYTE_ARRAY,
+                HConstants.EMPTY_BYTE_ARRAY, "", HBaseConstants.SourceMode.EMPTY, false);
+    }
+
+    /**
+     * Constructor
+     *
+     * @param tableName
+     * @param startRow
+     * @param endRow
+     * @param location
+     */
+    public HBaseTableSplitGranular(final byte[] tableName, final byte[] startRow,
+                               final byte[] endRow, final String location,
+                               final HBaseConstants.SourceMode sourceMode, final boolean useSalt) {
+        this.m_tableName = tableName;
+        this.m_startRow = startRow;
+        this.m_endRow = endRow;
+        this.m_regionLocation = location;
+        this.m_sourceMode = sourceMode;
+        this.m_useSalt = useSalt;
+    }
+
+    public HBaseTableSplitGranular(final byte[] tableName,
+                               final TreeSet<String> keyList, int versions, final String location,
+                               final HBaseConstants.SourceMode sourceMode, final boolean useSalt) {
+        this.m_tableName = tableName;
+        this.m_keyList = keyList;
+        this.m_versions = versions;
+        this.m_sourceMode = sourceMode;
+        this.m_regionLocation = location;
+        this.m_useSalt = useSalt;
+    }
+
+
+	@Override
+	public long getLength() {
+		// Not clear how to obtain this... seems to be used only for sorting
+		// splits
+		return 0;
+	}
+
+
+	@Override
+	public String toString() {
+		return String
+				.format(
+						"Table Name (%s) Region (%s) Source Mode (%s) Start Key (%s) Stop Key (%s) Key List Size (%s) Versions (%s) Use Salt (%s)",
+						Bytes.toString(m_tableName), m_regionLocation, m_sourceMode,
+						Bytes.toString(m_startRow), Bytes.toString(m_endRow),
+						(m_keyList != null) ? m_keyList.size() : "EMPTY", m_versions,
+						m_useSalt);
+	}
+
+	@Override
+	public int compareTo(HBaseTableSplitBase o) {
+        if( ! (o instanceof HBaseTableSplitGranular) ) return -1;
+
+		switch (m_sourceMode) {
+			case SCAN_ALL:
+			case SCAN_RANGE:
+				return Bytes.compareTo(getStartRow(), o.getStartRow());
+
+			case GET_LIST:
+				return m_keyList.equals(o.getKeyList()) ? 0 : -1;
+
+			case EMPTY:
+				return 0;
+
+			default:
+				return -1;
+		}
+
+	}
+}
\ No newline at end of file