org.cloudgraph.hbase.connect.Connection Maven / Gradle / Ivy

Go to download
/**
 *        CloudGraph Community Edition (CE) License
 * 
 * This is a community release of CloudGraph, a dual-license suite of
 * Service Data Object (SDO) 2.1 services designed for relational and 
 * big-table style "cloud" databases, such as HBase and others. 
 * This particular copy of the software is released under the 
 * version 2 of the GNU General Public License. CloudGraph was developed by 
 * TerraMeta Software, Inc.
 * 
 * Copyright (c) 2013, TerraMeta Software, Inc. All rights reserved.
 * 
 * General License information can be found below.
 * 
 * This distribution may include materials developed by third
 * parties. For license and attribution notices for these
 * materials, please refer to the documentation that accompanies
 * this distribution (see the "Licenses for Third-Party Components"
 * appendix) or view the online documentation at 
 * . 
 */
package org.cloudgraph.hbase.connect;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ExecutorService;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.commons.pool2.ObjectPool;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.client.Admin;
import org.apache.hadoop.hbase.client.BufferedMutator;
import org.apache.hadoop.hbase.client.BufferedMutatorParams;
import org.apache.hadoop.hbase.client.RegionLocator;
import org.apache.hadoop.hbase.client.Table;
import org.cloudgraph.store.service.GraphServiceException;

/**
 * Connection wrapper. 
 * 
 * The new HBase Client API changes removed the existing connection pool implementation and placed 
 * the responsibility of managing the lifecycle of connections on the caller. A pool is necessary as 
 * creating connections to the cluster via. zookeeper is fairly expensive. 
 * @author Scott Cinnamond
 * @since 0.6.3
 */
public class Connection {
	private static Log log = LogFactory.getLog(Connection.class);
	private org.apache.hadoop.hbase.client.Connection con;
	private ObjectPool pool;

	public Connection(org.apache.hadoop.hbase.client.Connection con, ObjectPool pool) {
		super();
		this.con = con;
		this.pool = pool;
	}

	public void close() throws IOException {
		if (log.isDebugEnabled())
			log.debug("closing wrapped connection, " + this.con);
		try {
			this.pool.returnObject(this);
		} catch (Exception e) {
			throw new GraphServiceException(e);
		} 
	}

	public boolean isClosed() {
		return con.isClosed();
	}
	
	public void abort(String why, Throwable e) {
		con.abort(why, e);
	}

	public boolean isAborted() {
		return con.isAborted();
	}

	public Configuration getConfiguration() {
		return con.getConfiguration();
	}

	public boolean tableExists(TableName tableName) throws IOException {
		boolean exists = getAdmin().tableExists(tableName);;
		return exists;
	}

	public Table getTable(TableName tableName) throws IOException {
		Table result = con.getTable(tableName);
		return result;
	}

	public Table getTable(TableName tableName, ExecutorService pool)
			throws IOException {
		Table result = con.getTable(tableName, pool);
		return result;
	}

	public BufferedMutator getBufferedMutator(TableName tableName)
			throws IOException {
		return con.getBufferedMutator(tableName);
	}

	public BufferedMutator getBufferedMutator(BufferedMutatorParams params)
			throws IOException {
		return con.getBufferedMutator(params);
	}

	public RegionLocator getRegionLocator(TableName tableName)
			throws IOException {
		return con.getRegionLocator(tableName);
	}

	public Admin getAdmin() throws IOException {
		return con.getAdmin();
	}

}