EmbeddedHBaseLifecycleManager.java

package com.lordofthejars.nosqlunit.hbase;

import java.io.File;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.HBaseTestingUtility;
import org.apache.hadoop.hbase.HConstants;
import org.apache.hadoop.hbase.coprocessor.CoprocessorHost;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.lordofthejars.nosqlunit.core.AbstractLifecycleManager;
import com.lordofthejars.nosqlunit.core.IOUtils;

public class EmbeddedHBaseLifecycleManager extends AbstractLifecycleManager {

private static final Logger LOGGER = LoggerFactory.getLogger(EmbeddedHBase.class); 
	
	protected static final String LOCALHOST = "127.0.0.1";
	protected static final int PORT = HConstants.DEFAULT_MASTER_PORT;
	protected static final String TARGET_DIRECTORY = HBaseTestingUtility.DEFAULT_BASE_TEST_DIRECTORY;
	
	protected static final String DEFAULT_PERMISSIONS = "775";
	
	protected String filePermissions = DEFAULT_PERMISSIONS;
	protected Configuration configuration;
	
	protected EmbeddedHBaseStarter embeddedHBaseStarter;
	
	public EmbeddedHBaseLifecycleManager() {
		super();
		embeddedHBaseStarter = new EmbeddedHBaseStarter();
	}
	
	@Override
	public String getHost() {
		return LOCALHOST;
	}

	@Override
	public int getPort() {
		return PORT;
	}

	@Override
	public void doStart() throws Throwable {
		
		LOGGER.info("Starting Embedded HBase instance.");
		
		Configuration config = configuration();
		startMiniCluster(config);
		EmbeddedHBaseInstances.getInstance().addHBaseConfiguration(this.configuration, LOCALHOST+PORT);
		
		LOGGER.info("Started Embedded HBase instance.");
		
	}

	private void startMiniCluster(Configuration config) throws Exception {
		HBaseTestingUtility testUtil = embeddedHBaseStarter.startSingleMiniCluster(config);
		configuration = testUtil.getConfiguration();
	}

	private Configuration configuration() {
		Configuration config = HBaseConfiguration.create();
		config.set(CoprocessorHost.REGION_COPROCESSOR_CONF_KEY,
				"org.apache.hadoop.hbase.coprocessor.AggregateImplementation");
		config.set("dfs.datanode.data.dir.perm", filePermissions);
		return config;
	}

	@Override
	public void doStop() {
		
		LOGGER.info("Stopping Embedded HBase instance.");
		
		EmbeddedHBaseInstances.getInstance().removeHBaseConfiguration(LOCALHOST+PORT);
		shutdownMiniCluster();
		cleanTargetDirectory();
		
		LOGGER.info("Stopped Embedded HBase instance.");
	}

	private void shutdownMiniCluster() {
		embeddedHBaseStarter.stopMiniCluster();
	}
	
	private void cleanTargetDirectory() {
		File directory = new File(TARGET_DIRECTORY);
		
		if(directory.exists()) {
			IOUtils.deleteDir(directory);
		}
		
	}
	
	public String getFilePermissions() {
		return filePermissions;
	}
	
	public void setFilePermissions(String filePermissions) {
		this.filePermissions = filePermissions;
	}

	protected void setEmbeddedHBaseStarter(EmbeddedHBaseStarter embeddedHBaseStarter) {
		this.embeddedHBaseStarter = embeddedHBaseStarter;
	}

}