Cross data source queries

Author: tarelli

src/main/java/org/geppetto/datasources/ADataSourceService.java

@@ -35,13 +35,19 @@
 import java.util.ArrayList;
 import java.util.List;
 
+import org.geppetto.core.datasources.GeppettoDataSourceException;
 import org.geppetto.core.datasources.IDataSourceService;
+import org.geppetto.core.datasources.IQueryListener;
 import org.geppetto.core.datasources.QueryChecker;
 import org.geppetto.core.model.GeppettoModelAccess;
 import org.geppetto.core.services.AService;
 import org.geppetto.model.datasources.DataSource;
 import org.geppetto.model.datasources.Query;
+import org.geppetto.model.datasources.QueryResults;
+import org.geppetto.model.util.GeppettoModelTraversal;
+import org.geppetto.model.util.GeppettoVisitingException;
 import org.geppetto.model.variables.Variable;
+import org.geppetto.model.variables.VariablesFactory;
 
 /**
  * @author matteocantarelli
@@ -50,11 +56,17 @@
 public abstract class ADataSourceService extends AService implements IDataSourceService
 {
 
+	public abstract ConnectionType getConnectionType();
+
+	public abstract IQueryResponseProcessor getQueryResponseProcessor();
+	
 	public enum ConnectionType
 	{
 		GET, POST
 	}
 
+	protected IQueryResponseProcessor queryResponseProcessor;
+
 	private DataSource configuration;
 
 	private String dataSourceTemplate;
@@ -91,17 +103,7 @@ protected DataSource getConfiguration()
 		return configuration;
 	}
 
-	/**
-	 * @param url
-	 * @param queryString
-	 * @return
-	 */
-	protected String getQueryURL(String url, String queryString)
-	{
-		String queryURL = null;
-		// TODO Do we need a template? Plain concatenation?
-		return queryURL;
-	}
+
 
 	/*
 	 * (non-Javadoc)
@@ -122,6 +124,84 @@ public List<Query> getAvailableQueries(Variable variable)
 		return availableQueries;
 	}
 
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.core.model.QueryProvider#execute(org.geppetto.core.model.Query, org.geppetto.model.variables.Variable, org.geppetto.core.model.QueryListener)
+	 */
+	@Override
+	public QueryResults execute(Query query, Variable variable, IQueryListener listener) throws GeppettoDataSourceException
+	{
+		ExecuteQueryVisitor runQueryVisitor = new ExecuteQueryVisitor(variable, getGeppettoModelAccess());
+		runQueryVisitor.doSwitch(query);
+		return runQueryVisitor.getResults();
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.core.model.QueryProvider#getNumberOfResults(org.geppetto.core.model.Query, org.geppetto.model.variables.Variable, org.geppetto.core.model.QueryResults)
+	 */
+	@Override
+	public int getNumberOfResults(Query query, Variable variable, QueryResults results) throws GeppettoDataSourceException
+	{
+		// TODO Auto-generated method stub
+		return 0;
+	}
+
+	
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.core.model.QueryProvider#getNumberOfResults(org.geppetto.core.model.Query, org.geppetto.model.variables.Variable)
+	 */
+	@Override
+	public int getNumberOfResults(Query query, Variable variable) throws GeppettoDataSourceException
+	{
+		Query fetchVariableQuery = getConfiguration().getFetchVariableQuery();
+		ExecuteQueryVisitor runQueryVisitor = new ExecuteQueryVisitor(variable, getGeppettoModelAccess());
+		try
+		{
+			GeppettoModelTraversal.applyDirectChildren(fetchVariableQuery, runQueryVisitor);
+
+		}
+		catch(GeppettoVisitingException e)
+		{
+			throw new GeppettoDataSourceException(e);
+		}
+
+		return runQueryVisitor.getCount();
+	}
+	
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.core.model.QueryProvider#execute(org.geppetto.core.model.Query, org.geppetto.model.variables.Variable, org.geppetto.core.model.QueryResults,
+	 * org.geppetto.core.model.QueryListener)
+	 */
+	@Override
+	public QueryResults execute(Query query, Variable variable, QueryResults results, IQueryListener listener) throws GeppettoDataSourceException
+	{
+		// TODO Auto-generated method stub
+		return null;
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.core.model.IDataSource#fetchVariable(java.lang.String)
+	 */
+	@Override
+	public void fetchVariable(String variableId) throws GeppettoDataSourceException
+	{
+		Variable fetchedVariable = VariablesFactory.eINSTANCE.createVariable();
+		fetchedVariable.setId(variableId);
+		getGeppettoModelAccess().addVariable(fetchedVariable);
+		Query fetchVariableQuery = getConfiguration().getFetchVariableQuery();
+		ExecuteQueryVisitor runQueryVisitor = new ExecuteQueryVisitor(fetchedVariable, getGeppettoModelAccess());
+		runQueryVisitor.doSwitch(fetchVariableQuery);
+	}
+
 	/*
 	 * (non-Javadoc)
 	 * 
@@ -133,4 +213,6 @@ public void registerGeppettoService() throws Exception
 		// Nothing to do here
 	}
 
+
+
 }

src/main/java/org/geppetto/datasources/AQueryProcessor.java

@@ -41,7 +41,9 @@ public void addFeature(IFeature feature)
 
 	}
 
-	/* (non-Javadoc)
+	/*
+	 * (non-Javadoc)
+	 * 
 	 * @see org.geppetto.core.datasources.IQueryProcessor#getProcessingOutputMap()
 	 */
 	@Override

src/main/java/org/geppetto/datasources/DummyDataSourceService.java

@@ -35,6 +35,7 @@
 import org.geppetto.core.datasources.GeppettoDataSourceException;
 import org.geppetto.core.datasources.IDataSourceService;
 import org.geppetto.core.datasources.IQueryListener;
+import org.geppetto.datasources.neo4j.Neo4jResponseProcessor;
 import org.geppetto.model.GeppettoLibrary;
 import org.geppetto.model.datasources.DataSource;
 import org.geppetto.model.datasources.DataSourceLibraryConfiguration;
@@ -118,22 +119,23 @@ public void fetchVariable(String variableId) throws GeppettoDataSourceException
 		fetchedVariable.setId(variableId);
 		getGeppettoModelAccess().addVariable(fetchedVariable);
 		ImportType importType = TypesFactory.eINSTANCE.createImportType();
-		importType.setId("Type"+variableId); //an SWC for instance
-		importType.setUrl(""); //an SWC for instance
+		importType.setId("Type" + variableId); // an SWC for instance
+		importType.setUrl(""); // an SWC for instance
 		fetchedVariable.getTypes().add(importType);
 		importType.setModelInterpreterId("swcModelInterpreter");
-		getGeppettoModelAccess().addTypeToLibrary(importType, getLibraryFor(getConfiguration(),"swc"));
-		
+		getGeppettoModelAccess().addTypeToLibrary(importType, getLibraryFor(getConfiguration(), "swc"));
+
 	}
-	
+
 	/**
 	 * @param dataSource
 	 * @param format
 	 * @return
 	 */
 	private GeppettoLibrary getLibraryFor(DataSource dataSource, String format)
 	{
-		for(DataSourceLibraryConfiguration lc: dataSource.getLibraryConfigurations()){
+		for(DataSourceLibraryConfiguration lc : dataSource.getLibraryConfigurations())
+		{
 			if(lc.getFormat().equals(format))
 			{
 				return lc.getLibrary();
@@ -142,4 +144,20 @@ private GeppettoLibrary getLibraryFor(DataSource dataSource, String format)
 		return null;
 	}
 
+	@Override
+	public ConnectionType getConnectionType()
+	{
+		return ConnectionType.POST;
+	}
+
+	@Override
+	public IQueryResponseProcessor getQueryResponseProcessor()
+	{
+		if(queryResponseProcessor == null)
+		{
+			queryResponseProcessor = new Neo4jResponseProcessor();
+		}
+		return queryResponseProcessor;
+	}
+
 }