X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=runtime%2Forg.argeo.slc.core%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fslc%2Fcore%2Fruntime%2FDefaultAgent.java;h=73f5cfd342a5d2394fbb063b37c8b076574993fb;hb=a2ecbd2a3eae94576fe4db3d651bc0908245da90;hp=ef2c52382f073f6ba6b5358b866ccfb53f1437ef;hpb=1fdb1b4e7b1d2b0cabb6483238301b857a6392fa;p=gpl%2Fargeo-slc.git

diff --git a/runtime/org.argeo.slc.core/src/main/java/org/argeo/slc/core/runtime/DefaultAgent.java b/runtime/org.argeo.slc.core/src/main/java/org/argeo/slc/core/runtime/DefaultAgent.java
index ef2c52382..73f5cfd34 100644
--- a/runtime/org.argeo.slc.core/src/main/java/org/argeo/slc/core/runtime/DefaultAgent.java
+++ b/runtime/org.argeo.slc.core/src/main/java/org/argeo/slc/core/runtime/DefaultAgent.java
@@ -16,26 +16,100 @@
 
 package org.argeo.slc.core.runtime;
 
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.UUID;
 
+import org.argeo.slc.SlcException;
+import org.argeo.slc.core.execution.ProcessThread;
 import org.argeo.slc.execution.ExecutionModuleDescriptor;
 import org.argeo.slc.execution.ExecutionModulesManager;
+import org.argeo.slc.execution.ExecutionProcess;
+import org.argeo.slc.execution.ExecutionProcessNotifier;
+import org.argeo.slc.execution.ExecutionStep;
 import org.argeo.slc.process.SlcExecution;
 import org.argeo.slc.runtime.SlcAgent;
+import org.argeo.slc.runtime.SlcAgentDescriptor;
 
-public class DefaultAgent implements SlcAgent {
-	// private final static Log log = LogFactory.getLog(AbstractAgent.class);
-
+/** Implements the base methods of an SLC agent. */
+@SuppressWarnings("deprecation")
+public class DefaultAgent implements SlcAgent, ExecutionProcessNotifier {
+	private SlcAgentDescriptor agentDescriptor;
 	private ExecutionModulesManager modulesManager;
 
-	public void runSlcExecution(final SlcExecution slcExecution) {
-		modulesManager.process(slcExecution);
+	private ThreadGroup processesThreadGroup;
+	private Map<String, ProcessThread> runningProcesses = Collections
+			.synchronizedMap(new HashMap<String, ProcessThread>());
+
+	/*
+	 * LIFECYCLE
+	 */
+	/** Initialization */
+	public void init() {
+		try {
+			agentDescriptor = new SlcAgentDescriptor();
+			agentDescriptor.setUuid(initAgentUuid());
+			agentDescriptor.setHost(InetAddress.getLocalHost().getHostName());
+		} catch (UnknownHostException e) {
+			throw new SlcException("Unable to create agent descriptor.", e);
+		}
+		processesThreadGroup = new ThreadGroup("SLC Processes of Agent #"
+				+ agentDescriptor.getUuid());
+		modulesManager.registerProcessNotifier(this,
+				new HashMap<String, String>());
+	}
+
+	/** Clean up (needs to be called by overriding method) */
+	public void destroy() {
+		modulesManager.unregisterProcessNotifier(this,
+				new HashMap<String, String>());
+	}
+
+	/**
+	 * Called during initialization in order to determines the agent UUID. To be
+	 * overridden. By default creates a new one per instance.
+	 */
+	protected String initAgentUuid() {
+		return UUID.randomUUID().toString();
+	}
+
+	/*
+	 * SLC AGENT
+	 */
+	public void process(ExecutionProcess process) {
+		ProcessThread processThread = createProcessThread(processesThreadGroup,
+				modulesManager, process);
+		processThread.start();
+		runningProcesses.put(process.getUuid(), processThread);
+		// FIXME find a way to remove them from this register
+	}
+
+	public void kill(ExecutionProcess process) {
+		String processUuid = process.getUuid();
+		if (runningProcesses.containsKey(processUuid)) {
+			runningProcesses.get(processUuid).interrupt();
+		}
+	}
+
+	/** Creates the thread which will coordinate the execution for this agent. */
+	protected ProcessThread createProcessThread(
+			ThreadGroup processesThreadGroup,
+			ExecutionModulesManager modulesManager, ExecutionProcess process) {
+		if (!(process instanceof SlcExecution))
+			throw new SlcException("Unsupported process type "
+					+ process.getClass());
+		ProcessThread processThread = new ProcessThread(processesThreadGroup,
+				modulesManager, (SlcExecution) process);
+		return processThread;
 	}
 
 	public ExecutionModuleDescriptor getExecutionModuleDescriptor(
 			String moduleName, String version) {
-		return modulesManager.getExecutionModuleDescriptor(moduleName,
-				version);
+		return modulesManager.getExecutionModuleDescriptor(moduleName, version);
 	}
 
 	public List<ExecutionModuleDescriptor> listExecutionModuleDescriptors() {
@@ -46,12 +120,38 @@ public class DefaultAgent implements SlcAgent {
 		return true;
 	}
 
+	/*
+	 * PROCESS NOTIFIER
+	 */
+	public void updateStatus(ExecutionProcess process, String oldStatus,
+			String newStatus) {
+		if (newStatus.equals(ExecutionProcess.COMPLETED)
+				|| newStatus.equals(ExecutionProcess.ERROR)
+				|| newStatus.equals(ExecutionProcess.KILLED)) {
+			runningProcesses.remove(process.getUuid());
+		}
+	}
+
+	public void addSteps(ExecutionProcess process, List<ExecutionStep> steps) {
+	}
+
+	/*
+	 * BEAN
+	 */
 	public void setModulesManager(ExecutionModulesManager modulesManager) {
 		this.modulesManager = modulesManager;
 	}
 
-	public ExecutionModulesManager getModulesManager() {
-		return modulesManager;
+	protected SlcAgentDescriptor getAgentDescriptor() {
+		return agentDescriptor;
+	}
+
+	public String getAgentUuid() {
+		return agentDescriptor.getUuid();
 	}
 
+	@Override
+	public String toString() {
+		return agentDescriptor.toString();
+	}
 }