NBLAST: added new data source for NBLAST

Author: ddelpiano

src/main/java/META-INF/spring/osgi-config.xml

@@ -32,6 +32,12 @@
 	<aop:scoped-proxy proxy-target-class="false"/>
 	</bean>
 
+	<osgi:service id="nblastDataSourceExporter" ref="nblastDataSource"
+		interface="org.geppetto.core.datasources.IDataSourceService">
+	</osgi:service>
+	<bean id="nblastDataSource" scope="prototype" class="org.geppetto.datasources.nblast.NBLASTDataSourceService">
+	<aop:scoped-proxy proxy-target-class="false"/>
+	</bean>
 
 	<bean id="eventListenerBean" class="org.geppetto.core.services.registry.ApplicationListenerBean" />
 

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

@@ -216,6 +216,12 @@ public Object caseSimpleQuery(SimpleQuery query)
 	private void processResponse(String response, ADataSourceService dataSourceService) throws GeppettoDataSourceException
 	{
 		try{
+			String customJson = "";
+			if(response.startsWith("[")) {
+				customJson = "{\"response\":"+response+"}";
+				response = customJson;
+			}
+			
 			if(count)
 			{
 				Map<String, Object> responseMap = JSONUtility.getAsMap(response);

src/main/java/org/geppetto/datasources/nblast/NBLASTDataSourceService.java

@@ -0,0 +1,58 @@
+
+package org.geppetto.datasources.nblast;
+
+import org.geppetto.core.datasources.GeppettoDataSourceException;
+import org.geppetto.core.datasources.IQueryListener;
+import org.geppetto.datasources.ADataSourceService;
+import org.geppetto.datasources.ExecuteQueryVisitor;
+import org.geppetto.datasources.IQueryResponseProcessor;
+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 dariodelpiano
+ *
+ */
+
+public class NBLASTDataSourceService extends ADataSourceService
+{
+
+	public NBLASTDataSourceService()
+	{
+		super("/templates/NBLAST/queryTemplate.vm");
+	}
+
+
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.datasources.ADataSourceService#getConnectionType()
+	 */
+	@Override
+	public ConnectionType getConnectionType()
+	{
+		return ConnectionType.POST;
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.datasources.ADataSourceService#getQueryResponseProcessor()
+	 */
+	@Override
+	public IQueryResponseProcessor getQueryResponseProcessor()
+	{
+		if(queryResponseProcessor == null)
+		{
+			queryResponseProcessor = new NBLASTResponseProcessor();
+		}
+		return queryResponseProcessor;
+	}
+
+
+
+}

src/main/java/org/geppetto/datasources/nblast/NBLASTResponseProcessor.java

@@ -0,0 +1,66 @@
+
+package org.geppetto.datasources.nblast;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Arrays;
+
+
+import org.geppetto.datasources.IQueryResponseProcessor;
+import org.geppetto.model.datasources.DatasourcesFactory;
+import org.geppetto.model.datasources.QueryResult;
+import org.geppetto.model.datasources.QueryResults;
+
+/**
+ * @author dariodelpiano
+ *
+ */
+public class NBLASTResponseProcessor implements IQueryResponseProcessor
+{
+
+	/*
+	 * (non-Javadoc)
+	 * 
+	 * @see org.geppetto.datasources.IQueryResponseProcessor#processResponse(java.lang.String)
+	 */
+	@Override
+	public QueryResults processResponse(Map<String, Object> response)
+	{		
+		QueryResults results = DatasourcesFactory.eINSTANCE.createQueryResults();
+		List<String> headers = new ArrayList<String>();
+		List<Map<String, Object>> data = (List<Map<String, Object>>) response.get("response");
+
+		for(Map<String, Object> rowObject : data)
+		{
+			for(String column : rowObject.keySet())
+			{
+				if(!headers.contains(column))
+				{
+					headers.add(column);
+				}
+			}
+		}
+
+		results.getHeader().addAll(headers);
+
+		for(Map<String, Object> rowObject : data)
+		{
+			QueryResult resultRow = DatasourcesFactory.eINSTANCE.createQueryResult();
+			for(String column : headers)
+			{
+				if(rowObject.containsKey(column))
+				{
+					resultRow.getValues().add((rowObject.get(column)));
+				}
+				else
+				{
+					resultRow.getValues().add(null);
+				}
+			}
+			results.getResults().add(resultRow);
+		}
+
+		return results;
+	}
+}

src/main/resources/templates/NBLAST/queryTemplate.vm

@@ -0,0 +1,3 @@
+{
+	"query":"$QUERY"
+}