hbaseDB.java

package implementations;
/**
 * Copyright 2011 Thibault Dory
 * Licensed under the GPL Version 3 license
 */

import hbase_mapreduce.MapReduceHbaseDB;
import java.io.IOException;

import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.client.Get;
import org.apache.hadoop.hbase.client.HTable;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.util.Bytes;

import core.BenchDB;

/**
 * 
 * @author Thibault Dory
 *
 * Use the following commands in HBase shell to create the needed tables and column families
 * create 'myTable', 'myColumnFamily'
 * create 'result', 'resultF'
 * create 'result2', 'resultF'
 */

public class hbaseDB extends BenchDB{
	HTable table;
	HBaseConfiguration config;
	
	@Override
	public int connectNode(String nodeAddress) {
		int ret;
		config = new HBaseConfiguration();
		try {
			table = new HTable(config, "myTable");
			ret = 1;
		} catch (IOException e) {
			e.printStackTrace();
			ret = -1;
		}
		return ret;
	}
	@Override
	public String readDB(String ID) {
		String ret;
		Get g = new Get(Bytes.toBytes(ID));
		try {
			Result r = table.get(g);
			byte [] value = r.getValue(Bytes.toBytes("myColumnFamily"),Bytes.toBytes("value"));
			ret = Bytes.toString(value);
		} catch (IOException e) {
			e.printStackTrace();
			ret = null;
		}
		return ret;
	}

	@Override
	public int updateDB(String ID, String newValue) {
		return writeDB(ID, newValue);
	}
	@Override
	public int writeDB(String ID, String Value) {
		int ret = 0;
		Put p = new Put(Bytes.toBytes(ID));
		try{
			p.add(Bytes.toBytes("myColumnFamily"), Bytes.toBytes("value"), Bytes.toBytes(Value));
			table.put(p);
			ret = 1;
		}catch(Exception e){
			e.printStackTrace();
			ret = -1;			
		}
		return ret;
	}
	
	@Override
	public void close(){
		try {
			table.close();
			System.out.println("================== Table closed ====================================================");
			config.clear();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	@Override
	public void searchDB(String keyword){
		
		String[] args = {"master"};
		try {
			MapReduceHbaseDB.main(args);
		} catch (Exception e) {
			e.printStackTrace();
		}
		
	}

}