smart-executor/src/main/java/org/gcube/vremanagement/executor/persistence/CouchDBPersistenceConnector.java

/**
 * 
 */
package org.gcube.vremanagement.executor.persistence;

import java.net.URL;
import java.util.UUID;

import org.codehaus.jackson.JsonNode;
import org.codehaus.jackson.map.ObjectMapper;
import org.codehaus.jackson.node.ObjectNode;
import org.ektorp.CouchDbConnector;
import org.ektorp.CouchDbInstance;
import org.ektorp.ViewQuery;
import org.ektorp.ViewResult;
import org.ektorp.http.HttpClient;
import org.ektorp.http.StdHttpClient;
import org.ektorp.http.StdHttpClient.Builder;
import org.ektorp.impl.StdCouchDbConnector;
import org.ektorp.impl.StdCouchDbInstance;
import org.gcube.vremanagement.executor.plugin.PluginState;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Luca Frosini (ISTI - CNR) http://www.lucafrosini.com/
 *
 */
public class CouchDBPersistenceConnector extends PersistenceConnector {

	private static final Logger logger = LoggerFactory.getLogger(CouchDBPersistenceConnector.class);
	
	protected CouchDbInstance couchDbInstance;
	protected CouchDbConnector couchDbConnector;
	
	protected static final String DB_NAME = "dbName";
	
	public CouchDBPersistenceConnector(SmartExecutorPersistenceConfiguration configuration) throws Exception {
		prepareConnection(configuration);
	}
	
	protected HttpClient initHttpClient(URL url, String username, String password){
		Builder builder = new StdHttpClient.Builder().url(url); 
		builder.username(username).password(password);
		HttpClient httpClient = builder.build();
		return httpClient;
	}
	
	protected void prepareConnection(SmartExecutorPersistenceConfiguration configuration) throws Exception {
		logger.debug("Preparing Connection for {}", this.getClass().getSimpleName());
		HttpClient httpClient = initHttpClient(configuration.getUri().toURL(), configuration.getUsername(), configuration.getPassword());
		couchDbInstance = new StdCouchDbInstance(httpClient);
		couchDbConnector = new StdCouchDbConnector(configuration.getProperty(DB_NAME), couchDbInstance);
	}
	
	protected ViewResult query(ViewQuery query){
		ViewResult result = couchDbConnector.queryView(query);
		return result;
	}
	
	
	@Override
	public void close() throws Exception {
		couchDbConnector.getConnection().shutdown();
	}
	
	protected void createItem(JsonNode node, String id) throws Exception {
		if(id!=null && id.compareTo("")!=0){
			couchDbConnector.create(id, node);
		}else{
			couchDbConnector.create(node);
		}
	}
	
	public final static String UUID_FIELD = "uuid";
	public final static String ITERATION_FIELD = "iteration";
	public final static String PLUGIN_NAME_FIELD = "pluginName";
	public final static String TIMESTAMP_FIELD = "timestamp";
	public final static String STATE_FIELD = "state";
	
	/**
	 * {@inheritDoc}
	 */
	@Override
	public void pluginStateEvolution(UUID uuid, int iteration, long timestamp,
			String pluginName, PluginState pluginState) throws Exception {
		ObjectMapper objectMapper = new ObjectMapper();
		ObjectNode objectNode = objectMapper.createObjectNode();
		objectNode.put(UUID_FIELD, uuid.toString());
		objectNode.put(ITERATION_FIELD, iteration);
		objectNode.put(TIMESTAMP_FIELD, timestamp);
		objectNode.put(PLUGIN_NAME_FIELD, pluginName);
		objectNode.put(STATE_FIELD, pluginState.toString());
		createItem(objectNode, null);
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public PluginState getPluginInstanceState(UUID uuid, int iterationNumber)
			throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public PluginState getLastPluginInstanceState(UUID uuid) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}
	
}