1 files changed, 165 insertions, 0 deletions
diff --git a/src/main/java/parallelai/spyglass/hbase/HBaseTableSplitBase.java b/src/main/java/parallelai/spyglass/hbase/HBaseTableSplitBase.java
new file mode 100644
index 0000000..2f6e7b5
--- /dev/null
+++ b/src/main/java/parallelai/spyglass/hbase/HBaseTableSplitBase.java
@@ -0,0 +1,165 @@
+package parallelai.spyglass.hbase;
+
+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 java.io.DataInput;
+import java.io.DataOutput;
+import java.io.IOException;
+import java.io.Serializable;
+import java.util.TreeSet;
+
+/**
+ * Created with IntelliJ IDEA.
+ * User: chand_000
+ * Date: 29/08/13
+ * Time: 16:18
+ * To change this template use File | Settings | File Templates.
+ */
+public abstract class HBaseTableSplitBase implements InputSplit,
+        Comparable<HBaseTableSplitBase>, Serializable {
+
+    private final Log LOG = LogFactory.getLog(HBaseTableSplitBase.class);
+
+
+    protected byte[] m_tableName = null;
+    protected byte[] m_startRow = null;
+    protected byte[] m_endRow = null;
+    protected String m_regionLocation = null;
+    protected TreeSet<String> m_keyList = null;
+    protected HBaseConstants.SourceMode m_sourceMode = HBaseConstants.SourceMode.EMPTY;
+    protected boolean m_endRowInclusive = true;
+    protected int m_versions = 1;
+    protected boolean m_useSalt = false;
+
+
+    /** @return table name */
+    public byte[] getTableName() {
+        return this.m_tableName;
+    }
+
+    /** @return starting row key */
+    public byte[] getStartRow() {
+        return this.m_startRow;
+    }
+
+    /** @return end row key */
+    public byte[] getEndRow() {
+        return this.m_endRow;
+    }
+
+    public boolean getEndRowInclusive() {
+        return m_endRowInclusive;
+    }
+
+    public void setEndRowInclusive(boolean isInclusive) {
+        m_endRowInclusive = isInclusive;
+    }
+
+    /** @return list of keys to get */
+    public TreeSet<String> getKeyList() {
+        return m_keyList;
+    }
+
+    public int getVersions() {
+        return m_versions;
+    }
+
+    /** @return get the source mode */
+    public HBaseConstants.SourceMode getSourceMode() {
+        return m_sourceMode;
+    }
+
+    public boolean getUseSalt() {
+        return m_useSalt;
+    }
+
+    /** @return the region's hostname */
+    public String getRegionLocation() {
+        LOG.debug("REGION GETTER : " + m_regionLocation);
+
+        return this.m_regionLocation;
+    }
+
+    public String[] getLocations() {
+        LOG.debug("REGION ARRAY : " + m_regionLocation);
+
+        return new String[] { this.m_regionLocation };
+    }
+
+
+    public void copy(HBaseTableSplitBase that) {
+        this.m_endRow = that.m_endRow;
+        this.m_endRowInclusive = that.m_endRowInclusive;
+        this.m_keyList = that.m_keyList;
+        this.m_sourceMode = that.m_sourceMode;
+        this.m_startRow = that.m_startRow;
+        this.m_tableName = that.m_tableName;
+        this.m_useSalt = that.m_useSalt;
+        this.m_versions = that.m_versions;
+    }
+
+    @Override
+    public void readFields(DataInput in) throws IOException {
+        LOG.debug("READ ME : " + in.toString());
+
+        this.m_tableName = Bytes.readByteArray(in);
+        this.m_regionLocation = Bytes.toString(Bytes.readByteArray(in));
+        this.m_sourceMode = HBaseConstants.SourceMode.valueOf(Bytes.toString(Bytes
+                .readByteArray(in)));
+        this.m_useSalt = Bytes.toBoolean(Bytes.readByteArray(in));
+
+        switch (this.m_sourceMode) {
+            case SCAN_RANGE:
+                this.m_startRow = Bytes.readByteArray(in);
+                this.m_endRow = Bytes.readByteArray(in);
+                this.m_endRowInclusive = Bytes.toBoolean(Bytes.readByteArray(in));
+                break;
+
+            case GET_LIST:
+                this.m_versions = Bytes.toInt(Bytes.readByteArray(in));
+                this.m_keyList = new TreeSet<String>();
+
+                int m = Bytes.toInt(Bytes.readByteArray(in));
+
+                for (int i = 0; i < m; i++) {
+                    this.m_keyList.add(Bytes.toString(Bytes.readByteArray(in)));
+                }
+                break;
+        }
+
+        LOG.debug("READ and CREATED : " + this);
+    }
+
+    @Override
+    public void write(DataOutput out) throws IOException {
+        LOG.debug("WRITE : " + this);
+
+        Bytes.writeByteArray(out, this.m_tableName);
+        Bytes.writeByteArray(out, Bytes.toBytes(this.m_regionLocation));
+        Bytes.writeByteArray(out, Bytes.toBytes(this.m_sourceMode.name()));
+        Bytes.writeByteArray(out, Bytes.toBytes(this.m_useSalt));
+
+        switch (this.m_sourceMode) {
+            case SCAN_RANGE:
+                Bytes.writeByteArray(out, this.m_startRow);
+                Bytes.writeByteArray(out, this.m_endRow);
+                Bytes.writeByteArray(out, Bytes.toBytes(this.m_endRowInclusive));
+                break;
+
+            case GET_LIST:
+                Bytes.writeByteArray(out, Bytes.toBytes(m_versions));
+                Bytes.writeByteArray(out, Bytes.toBytes(this.m_keyList.size()));
+
+                for (String k : this.m_keyList) {
+                    Bytes.writeByteArray(out, Bytes.toBytes(k));
+                }
+                break;
+        }
+
+        LOG.debug("WROTE : " + out.toString());
+    }
+}