+/*
+ * Copyright (C) 2010 Mathieu Baudier <mbaudier@argeo.org>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
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.Destination;
import javax.jms.JMSException;
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.core.runtime.DefaultAgent;
import org.argeo.slc.execution.ExecutionModuleDescriptor;
import org.argeo.slc.msg.ExecutionAnswer;
+import org.argeo.slc.msg.MsgConstants;
import org.argeo.slc.msg.ReferenceList;
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.MessagePostProcessor;
/** JMS based implementation of SLC Agent. */
-public class JmsAgent extends AbstractAgent implements SlcAgent,
- InitializingBean, DisposableBean, MessageListener {
+public class JmsAgent extends DefaultAgent implements InitializingBean,
+ DisposableBean, MessageListener {
public final static String PROPERTY_QUERY = "query";
- public final static String PROPERTY_SLC_AGENT_ID = "slc_agentId";
-
public final static String QUERY_PING_ALL = "pingAll";
private final static Log log = LogFactory.getLog(JmsAgent.class);
- private final SlcAgentDescriptor agentDescriptor;
private JmsTemplate jmsTemplate;
private Destination agentRegister;
private Destination agentUnregister;
private Destination responseDestination;
- public JmsAgent() {
- try {
- agentDescriptor = new SlcAgentDescriptor();
- agentDescriptor.setUuid(UUID.randomUUID().toString());
- agentDescriptor.setHost(InetAddress.getLocalHost().getHostName());
- } catch (UnknownHostException e) {
- throw new SlcException("Unable to create agent descriptor.", e);
- }
- }
-
public void afterPropertiesSet() throws Exception {
try {
- jmsTemplate.convertAndSend(agentRegister, agentDescriptor);
- log.info("Agent #" + agentDescriptor.getUuid() + " registered to "
+ jmsTemplate.convertAndSend(agentRegister, getAgentDescriptor());
+ log.info("Agent #" + getAgentUuid() + " registered to "
+ agentRegister);
} catch (JmsException e) {
log
.warn("Could not register agent "
- + agentDescriptor.getUuid()
+ + getAgentDescriptor().getUuid()
+ " to server: "
+ e.getMessage()
+ ". The agent will stay offline but will keep listening for a ping all sent by server.");
public void destroy() throws Exception {
try {
- jmsTemplate.convertAndSend(agentUnregister, agentDescriptor);
- log.info("Agent #" + agentDescriptor.getUuid()
- + " unregistered from " + agentUnregister);
+ jmsTemplate.convertAndSend(agentUnregister, getAgentDescriptor());
+ log.info("Agent #" + getAgentUuid() + " unregistered from "
+ + agentUnregister);
} catch (JmsException e) {
- log.warn("Could not unregister agent " + agentDescriptor.getUuid()
- + ": " + e.getMessage());
+ log.warn("Could not unregister agent " + getAgentUuid() + ": "
+ + e.getMessage());
if (log.isTraceEnabled())
log.debug("Original error.", e);
}
}
public String getMessageSelector() {
- String messageSelector = "slc_agentId='" + agentDescriptor.getUuid()
- + "'";
+ String messageSelector = "slc_agentId='" + getAgentUuid() + "'";
// if (log.isDebugEnabled())
// log.debug("Message selector: " + messageSelector);
return messageSelector;
}
- 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 boolean ping() {
- return true;
- }
-
public void onMessage(final Message message) {
final String query;
final String correlationId;
final Destination destinationSend;
if (QUERY_PING_ALL.equals(query)) {
ReferenceList refList = (ReferenceList) convertFrom(message);
- if (!refList.getReferences().contains(agentDescriptor.getUuid())) {
- response = agentDescriptor;
+ if (!refList.getReferences().contains(getAgentUuid())) {
+ response = getAgentDescriptor();
destinationSend = agentRegister;
- log.info("Agent #" + agentDescriptor.getUuid()
- + " registering to " + agentRegister
- + " in reply to a " + QUERY_PING_ALL + " query");
+ log.info("Agent #" + getAgentUuid() + " registering to "
+ + agentRegister + " in reply to a " + QUERY_PING_ALL
+ + " query");
} else {
return;
}
}
// Send response
+ if (log.isTraceEnabled())
+ log.trace("About to send response " + response.getClass());
jmsTemplate.convertAndSend(destinationSend, response,
new MessagePostProcessor() {
public Message postProcessMessage(Message messageToSend)
throws JMSException {
messageToSend.setStringProperty(PROPERTY_QUERY, query);
- messageToSend.setStringProperty(PROPERTY_SLC_AGENT_ID,
- agentDescriptor.getUuid());
+ messageToSend.setStringProperty(
+ MsgConstants.PROPERTY_SLC_AGENT_ID,
+ getAgentUuid());
messageToSend.setJMSCorrelationID(correlationId);
return messageToSend;
}
List<ExecutionModuleDescriptor> lst = listExecutionModuleDescriptors();
SlcAgentDescriptor agentDescriptorToSend = new SlcAgentDescriptor(
- agentDescriptor);
+ getAgentDescriptor());
agentDescriptorToSend.setModuleDescriptors(lst);
return agentDescriptorToSend;
} else if ("runSlcExecution".equals(query)) {
}
}.start();
return ExecutionAnswer.ok("Execution started on agent "
- + agentDescriptor.getUuid());
+ + getAgentUuid());
} else if ("ping".equals(query)) {
- return ExecutionAnswer.ok("Agent " + agentDescriptor.getUuid()
+ return ExecutionAnswer.ok("Agent " + getAgentUuid()
+ " is alive.");
} else {
throw new SlcException("Unsupported query " + query);
projects / gpl / argeo-slc.git / blobdiff