prerna.engine.impl.datastax.DataStaxGraphEngine Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.engine.impl.datastax;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
import java.util.Vector;

import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;

import com.datastax.driver.dse.DseCluster;
import com.datastax.driver.dse.DseSession;
import com.datastax.driver.dse.graph.GraphOptions;
import com.datastax.dse.graph.api.DseGraph;
import com.fasterxml.jackson.databind.ObjectMapper;

import prerna.engine.impl.AbstractDatabaseEngine;
import prerna.query.interpreters.GremlinNoEdgeBindInterpreter;
import prerna.query.interpreters.IQueryInterpreter;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.execptions.SemossPixelException;
import prerna.sablecc2.om.nounmeta.NounMetadata;

import prerna.util.Constants;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class DataStaxGraphEngine extends AbstractDatabaseEngine {
	
	private static final Logger classLogger = LogManager.getLogger(DataStaxGraphEngine.class);

	private GraphTraversalSource graphTraversalSession;
	private Map typeMap = new HashMap();
	private Map nameMap = new HashMap();

	@Override
	public void open(Properties smssProp) throws Exception {
		super.open(smssProp);
		String host = this.smssProp.getProperty("HOST");
		String port = this.smssProp.getProperty("PORT");
		String username = this.smssProp.getProperty("USERNAME");
		String password = this.smssProp.getProperty("PASSWORD");
		String graphName = this.smssProp.getProperty("GRAPH_NAME");
		// get the type map
		String typeMapStr = this.smssProp.getProperty("TYPE_MAP");
		// get the name map
		String nameMapStr = this.smssProp.getProperty("NAME_MAP");

		DseCluster dseCluster = null;
		if(username != null && password != null) {
			dseCluster = DseCluster.builder().addContactPoint(host).withCredentials(username, password)
					.withPort(Integer.parseInt(port)).withGraphOptions(new GraphOptions().setGraphName(graphName))
					.build();
		} else {
			dseCluster = DseCluster.builder().addContactPoint(host).withPort(Integer.parseInt(port))
					.withGraphOptions(new GraphOptions().setGraphName(graphName)).build();
		}
		if(dseCluster != null) {
			DseSession dseSession = dseCluster.connect();
			if (typeMapStr != null && !typeMapStr.trim().isEmpty()) {
				try {
					this.typeMap = new ObjectMapper().readValue(typeMapStr, Map.class);
				} catch (IOException e) {
					classLogger.error(Constants.STACKTRACE, e);
				}
			}
			if(nameMapStr != null && !nameMapStr.trim().isEmpty()) {
				try {
					this.nameMap = new ObjectMapper().readValue(nameMapStr, Map.class);
				} catch (IOException e) {
					classLogger.error(Constants.STACKTRACE, e);
				}
			}
			this.graphTraversalSession = DseGraph.traversal(dseSession);
		} else {
			NounMetadata noun = new NounMetadata("Unable to establish connection", PixelDataType.CONST_STRING, PixelOperationType.ERROR);
			SemossPixelException exception = new SemossPixelException(noun);
			exception.setContinueThreadOfExecution(false);
			throw exception;
		}
	}
	
	public GraphTraversalSource getGraphTraversalSource() {
		return this.graphTraversalSession;
	}
	
	public Map getTypeMap() {
		return this.typeMap;
	}
	
	public Map getNameMap() {
		return this.nameMap;
	}
	
	public IQueryInterpreter getQueryInterpreter() {
		return new GremlinNoEdgeBindInterpreter(this.graphTraversalSession, this.typeMap, this.nameMap, this);
	}
	
	@Override
	public DATABASE_TYPE getDatabaseType() {
		return DATABASE_TYPE.DATASTAX_GRAPH;
	}
	
	
	////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////
	////////////////////////////////////////////////////////////////////////
	
	@Override
	public Object execQuery(String query) {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public void insertData(String query) {
		// TODO Auto-generated method stub
		
	}

	@Override
	public Vector