com.weicoder.hbase.impl.HBaseImpl Maven / Gradle / Ivy

Go to download
package com.weicoder.hbase.impl;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.client.Admin;
import org.apache.hadoop.hbase.client.ColumnFamilyDescriptorBuilder;
import org.apache.hadoop.hbase.client.ConnectionFactory;
import org.apache.hadoop.hbase.client.TableDescriptorBuilder;

import com.weicoder.common.constants.StringConstants;
import com.weicoder.common.factory.FactoryKey;
import com.weicoder.common.lang.Conversion;
import com.weicoder.common.log.Logs;
import com.weicoder.hbase.HBase;
import com.weicoder.hbase.HBaseDao;

/**
 * HBase实现
 * 
 * @author WD
 */
public final class HBaseImpl extends FactoryKey implements HBase {
	// 配置
	private Configuration cfg;
	// HBase管理
	private Admin admin;

	/**
	 * 构造方法
	 * 
	 * @param host 主机
	 * @param port 端口
	 */
	public HBaseImpl(String host, int port) {
		try {
			// 创建配置
			cfg = HBaseConfiguration.create();
			// 设置主机
			cfg.set("hbase.master", host);
			cfg.set("hbase.zookeeper.quorum", host);
			// 设置端口
			cfg.set("hbase.zookeeper.property.clientPort", Conversion.toString(port));
			cfg.set("hbase.master.port", Conversion.toString(port));
			// 实例化管理
			admin = ConnectionFactory.createConnection(cfg).getAdmin();
		} catch (Exception e) {
			Logs.error(e);
		}
	}

	/**
	 * 获得Dao
	 * 
	 * @param  tableName 表名
	 * @return           HBaseDao
	 */
	public HBaseDao getDao(String tableName) {
		return getInstance(tableName);
	}

	/**
	 * 创建表
	 * 
	 * @param  tableName 表名
	 * @return           HBaseDao
	 */
	public HBaseDao createTable(String tableName, String... cols) {
		try {
			// 表不存在
			TableName tn = TableName.valueOf(tableName);
			if (!admin.tableExists(tn)) {
				// 声明表对象
				TableDescriptorBuilder tableDesc = TableDescriptorBuilder.newBuilder(tn);
				// 添加列
				for (int i = 0; i < cols.length; i++) {
					tableDesc.setColumnFamily(ColumnFamilyDescriptorBuilder.of(cols[i]));
				}
				// 创建表
				admin.createTable(tableDesc.build());
			}
		} catch (Exception e) {
			Logs.warn(e);
		}
		// 返回表Dao
		return getDao(tableName);
	}

	/**
	 * 删除表
	 * 
	 * @param tableName 表名
	 */
	public void deleteTable(String tableName) {
		try {
			// 生成表对象
			TableName name = TableName.valueOf(tableName);
			// 表存在
			if (admin.tableExists(name)) {
				// 使表失效
				admin.disableTable(name);
				// 删除表
				admin.deleteTable(name);
			}
		} catch (Exception e) {
			Logs.warn(e);
		}
	}

	/**
	 * 实例化一个新对象
	 */
	public HBaseDao newInstance(String tableName) {
		return new HBaseDaoImpl(cfg, tableName);
	}

	/**
	 * 实例化一个新对象
	 */
	public HBaseDao newInstance() {
		return newInstance(StringConstants.EMPTY);
	}
}