JAVA操作Hbase基础例子

package com.cma.hbase.test;

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HColumnDescriptor;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.KeyValue;
import org.apache.hadoop.hbase.MasterNotRunningException;
import org.apache.hadoop.hbase.ZooKeeperConnectionException;
import org.apache.hadoop.hbase.client.Delete;
import org.apache.hadoop.hbase.client.Get;
import org.apache.hadoop.hbase.client.HBaseAdmin;
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 com.cma.hbase.Constants;
import com.cma.hbase.entity.DataModel;
import com.cma.hbase.tools.HbaseUtils;

public class HbaseTest {

    public static void main(String[] args) {
        创建一张表
        createTable("hello_baby",new String[]{"code","ws","wd","t","ps","rh","vis","r"});
        写入一条数据
        writeRecord("hello_baby","row1","code","","code");
        writeRecord("hello_baby","row1","ws","","ws");
        writeRecord("hello_baby","row1","wd","","wd");
        writeRecord("hello_baby","row1","t","","t");
        writeRecord("hello_baby","row1","ps","","ps");
        writeRecord("hello_baby","row1","rh","","rh");
        writeRecord("hello_baby","row1","vis","","vis");
        writeRecord("hello_baby","row1","r","","r");
        写入一组数据
        writeRecordList("hello_baby");
        //查询出一条数据
        getRecord("hello_baby","row1");
        删除一行
        deleteRecord("hello_baby","row1");
        删除一张表
        dropTable("hello_baby");
        清空一张表
        clearTable("hello_baby");
    }

    /**
     * 清空一张表
     * @param string
     */
    private static void clearTable(String tableName) {
        Configuration cfg = HbaseUtils.getCfg();
        
        try {
            HBaseAdmin admin = new HBaseAdmin(cfg);
            
        } catch (MasterNotRunningException e) {
            e.printStackTrace();
        } catch (ZooKeeperConnectionException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            System.out.println("tableName: "+tableName+" drop over!");
        }
    }

    /**
     * 写入一组数据
     * @param tableName
     */
    private static void writeRecordList(String tableName) {
        Long start = System.currentTimeMillis();
        Configuration cfg = HbaseUtils.getCfg();
        try {
            HTable htable = new HTable(cfg,tableName);
            List<Put> puts = getPuts();
            System.out.println(puts.size());
            htable.put(puts);
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            System.out.println("tableName:"+tableName+" write over!");
        }
        Long end = System.currentTimeMillis();
        System.out.println("cost time: "+(end -start));
    }

    private static List<Put> getPuts() {
        List<Put> putList = new ArrayList<Put>();
        List<Put> putRecord = null;
        try {
            List<String> lines = FileUtils.readLines(new File("/home/guest/data/201307310800.csv"));
            for(int i = 1 ;i < lines.size();i++){
                putRecord = getPutsByLine(lines.get(i));
                putList.addAll(putRecord);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return putList;
    }

    /**
     * 获得一组Put
     * @param line
     * @return
     */
    private static List<Put> getPutsByLine(String line) {
        List<Put> puts = new ArrayList<Put>();
        Put put = null;
        
        if(StringUtils.isNotBlank(line)){
            String[] columns = line.split(",");
            String[] families = Constants.FAMILIES;
            String rowKey = "201307310800"+columns[0];
            for(int i = 0;i < columns.length;i++){
                String family = families[i];
                String qualifier = "";
                String value = columns[i];
                put = getPut(rowKey,family,qualifier,value);
                puts.add(put);
            }
        }
        return puts;
    }

    /**
     * 组装一个Put
     * @param rowKey
     * @param family
     * @param qualifier
     * @param value
     * @return
     */
    private static Put getPut(String rowKey, String family, String qualifier,
            String value) {
        Put put = new Put(Bytes.toBytes(rowKey));
        put.add(Bytes.toBytes(family), Bytes.toBytes(qualifier), Bytes.toBytes(value));
        return put;
    }

    /**
     * 查询出一条数据
     * @param tableName
     * @param rowKey
     */
    private static void getRecord(String tableName, String rowKey) {
        Configuration cfg = HbaseUtils.getCfg();
        try {
            HTable htable = new HTable(cfg,tableName);
            Get get = new Get(Bytes.toBytes(rowKey));
            Result rs = htable.get(get);
            
            for(KeyValue kv : rs.raw()){
                System.out.print(new String(kv.getRow())+" *** ");
                System.out.print(new String(kv.getFamily())+" *** ");
                System.out.print(new String(kv.getQualifier())+" *** ");
                System.out.print(new String(kv.getValue()));
                System.out.println();
            }
            
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 删除一张表
     * @param tableName
     */
    private static void dropTable(String tableName) {
        Configuration cfg = HbaseUtils.getCfg();
        
        try {
            HBaseAdmin admin = new HBaseAdmin(cfg);
            admin.disableTable(tableName);
            admin.deleteTable(tableName);
        } catch (MasterNotRunningException e) {
            e.printStackTrace();
        } catch (ZooKeeperConnectionException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            System.out.println("tableName: "+tableName+" drop over!");
        }
    }

    /**
     * 删除一行
     * @param tableName
     * @param rowKey
     */
    private static void deleteRecord(String tableName, String rowKey) {
        Configuration cfg = HbaseUtils.getCfg();
        
        try {
            HTable htable = new HTable(cfg,tableName);
            Delete del = new Delete(Bytes.toBytes(rowKey));
            htable.delete(del);
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            System.out.println("rowKey: "+rowKey+" delete over!");
        }
    }

    /**
     * 存储一列
     * @param tableName
     * @param rowKey
     * @param family
     * @param qualifier
     * @param value
     */
    private static void writeRecord(String tableName,String rowKey,String family,String qualifier,String value) {
        Configuration cfg = HbaseUtils.getCfg();
        try {
            HTable htable = new HTable(cfg,tableName);
            Put put = new Put(Bytes.toBytes(rowKey));
            put.add(Bytes.toBytes(family), Bytes.toBytes(qualifier), Bytes.toBytes(value));
            htable.put(put);
            
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            System.out.println("family: "+family+" put over!");
        }
    }

    /**
     * 创建一张表
     * @param tableName
     * @param families
     */
    private static void createTable(String tableName,String[] families) {
        HBaseAdmin admin = null;
        try {
            Configuration cfg = HbaseUtils.getCfg();
            System.out.println(cfg);
            admin = new HBaseAdmin(cfg);
            if(admin.tableExists(tableName)){
                System.out.println("表："+tableName+" 已经存在！");
            }else{
                HTableDescriptor tableDesc = new HTableDescriptor(tableName);
                for(String column : families){
                    tableDesc.addFamily(new HColumnDescriptor(column));
                }
                admin.createTable(tableDesc);
            }
        } catch (MasterNotRunningException e) {
            e.printStackTrace();
        } catch (ZooKeeperConnectionException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            System.out.println("over!");
        }
    }
    

}