X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=runtime%2Forg.argeo.slc.support.activemq%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fslc%2Fjms%2FJmsAgent.java;h=4b031f9472801fdbb92fe244a52ff66a400de33a;hb=ec9fe3774f157092d748b0844dbbb55e29a67481;hp=8678c1705b03c484911ccc7950cc79d7bc1b8800;hpb=04322c762f1704f0510f972ecd81c53b632899b3;p=gpl%2Fargeo-slc.git

diff --git a/runtime/org.argeo.slc.support.activemq/src/main/java/org/argeo/slc/jms/JmsAgent.java b/runtime/org.argeo.slc.support.activemq/src/main/java/org/argeo/slc/jms/JmsAgent.java
index 8678c1705..4b031f947 100644
--- a/runtime/org.argeo.slc.support.activemq/src/main/java/org/argeo/slc/jms/JmsAgent.java
+++ b/runtime/org.argeo.slc.support.activemq/src/main/java/org/argeo/slc/jms/JmsAgent.java
@@ -2,34 +2,47 @@ package org.argeo.slc.jms;
 
 import java.net.InetAddress;
 import java.net.UnknownHostException;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.UUID;
 
-import javax.jms.ConnectionFactory;
 import javax.jms.Destination;
+import javax.jms.JMSException;
+import javax.jms.Message;
+import javax.jms.MessageProducer;
+import javax.jms.Session;
+import javax.jms.TextMessage;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.argeo.slc.SlcException;
 import org.argeo.slc.core.runtime.AbstractAgent;
+import org.argeo.slc.execution.ExecutionModule;
+import org.argeo.slc.execution.ExecutionModuleDescriptor;
+import org.argeo.slc.process.SlcExecution;
 import org.argeo.slc.runtime.SlcAgent;
 import org.argeo.slc.runtime.SlcAgentDescriptor;
 import org.springframework.beans.factory.DisposableBean;
 import org.springframework.beans.factory.InitializingBean;
 import org.springframework.jms.core.JmsTemplate;
-import org.springframework.jms.support.converter.MessageConverter;
+import org.springframework.jms.listener.SessionAwareMessageListener;
+import org.springframework.jms.support.converter.MessageConversionException;
 
 /** JMS based implementation of SLC Agent. */
 public class JmsAgent extends AbstractAgent implements SlcAgent,
-		InitializingBean, DisposableBean {
+		InitializingBean, DisposableBean, SessionAwareMessageListener {
 	private final static Log log = LogFactory.getLog(JmsAgent.class);
 
 	private final SlcAgentDescriptor agentDescriptor;
-	private ConnectionFactory connectionFactory;
+	// private ConnectionFactory connectionFactory;
 	private JmsTemplate jmsTemplate;
 	private Destination agentRegister;
 	private Destination agentUnregister;
 
-	private MessageConverter messageConverter;
+	// private Destination requestDestination;
+	private Destination responseDestination;
+
+	// private MessageConverter messageConverter;
 
 	public JmsAgent() {
 		try {
@@ -43,8 +56,8 @@ public class JmsAgent extends AbstractAgent implements SlcAgent,
 
 	public void afterPropertiesSet() throws Exception {
 		// Initialize JMS Template
-		jmsTemplate = new JmsTemplate(connectionFactory);
-		jmsTemplate.setMessageConverter(messageConverter);
+		// jmsTemplate = new JmsTemplate(connectionFactory);
+		// jmsTemplate.setMessageConverter(messageConverter);
 
 		jmsTemplate.convertAndSend(agentRegister, agentDescriptor);
 		log.info("Agent #" + agentDescriptor.getUuid() + " registered to "
@@ -81,18 +94,85 @@ public class JmsAgent extends AbstractAgent implements SlcAgent,
 	public String getMessageSelector() {
 		String messageSelector = "slc_agentId='" + agentDescriptor.getUuid()
 				+ "'";
-		// String messageSelector = "slc-agentId LIKE '%'";
-		if (log.isDebugEnabled())
-			log.debug("Message selector: " + messageSelector);
+		// if (log.isDebugEnabled())
+		// log.debug("Message selector: " + messageSelector);
 		return messageSelector;
 	}
 
-	public void setMessageConverter(MessageConverter messageConverter) {
-		this.messageConverter = messageConverter;
+	public void onMessage(Message message, Session session) throws JMSException {
+		MessageProducer producer = session.createProducer(responseDestination);
+		String query = message.getStringProperty("query");
+		String correlationId = message.getJMSCorrelationID();
+		if (log.isDebugEnabled())
+			log.debug("Received query " + query + " with correlationId "
+					+ correlationId);
+
+		Message responseMsg = null;
+		if ("getExecutionModuleDescriptor".equals(query)) {
+			String moduleName = message.getStringProperty("moduleName");
+			String version = message.getStringProperty("version");
+			ExecutionModuleDescriptor emd = getExecutionModuleDescriptor(
+					moduleName, version);
+			responseMsg = jmsTemplate.getMessageConverter().toMessage(emd,
+					session);
+		} else if ("listExecutionModuleDescriptors".equals(query)) {
+
+			List<ExecutionModuleDescriptor> lst = listExecutionModuleDescriptors();
+			SlcAgentDescriptor agentDescriptorToSend = new SlcAgentDescriptor(
+					agentDescriptor);
+			agentDescriptorToSend.setModuleDescriptors(lst);
+			responseMsg = jmsTemplate.getMessageConverter().toMessage(
+					agentDescriptorToSend, session);
+		} else {
+			try {
+				//FIXME: generalize
+				SlcExecution slcExecution = (SlcExecution) jmsTemplate
+						.getMessageConverter().fromMessage(message);
+				runSlcExecution(slcExecution);
+			} catch (MessageConversionException e) {
+				if (log.isDebugEnabled())
+					log.debug("Unsupported query " + query, e);
+			}
+			return;
+		}
+
+		if (responseMsg != null) {
+			responseMsg.setJMSCorrelationID(correlationId);
+			producer.send(responseMsg);
+			if (log.isDebugEnabled())
+				log.debug("Sent response to query " + query
+						+ " with correlationId " + correlationId + ": "
+						+ responseMsg);
+		}
+
+	}
+
+	public ExecutionModuleDescriptor getExecutionModuleDescriptor(
+			String moduleName, String version) {
+		return getModulesManager().getExecutionModuleDescriptor(moduleName,
+				version);
+	}
+
+	public List<ExecutionModuleDescriptor> listExecutionModuleDescriptors() {
+		List<ExecutionModule> modules = getModulesManager()
+				.listExecutionModules();
+
+		List<ExecutionModuleDescriptor> descriptors = new ArrayList<ExecutionModuleDescriptor>();
+		for (ExecutionModule module : modules) {
+			ExecutionModuleDescriptor md = new ExecutionModuleDescriptor();
+			md.setName(module.getName());
+			md.setVersion(module.getVersion());
+			descriptors.add(md);
+		}
+		return descriptors;
+	}
+
+	public void setResponseDestination(Destination responseDestination) {
+		this.responseDestination = responseDestination;
 	}
 
-	public void setConnectionFactory(ConnectionFactory connectionFactory) {
-		this.connectionFactory = connectionFactory;
+	public void setJmsTemplate(JmsTemplate jmsTemplate) {
+		this.jmsTemplate = jmsTemplate;
 	}
 
 }