rebuilding head

[mir.git] / source / mir / producer / ExecuteProgramProducerNode.java

diff --git a/source/mir/producer/ExecuteProgramProducerNode.java b/source/mir/producer/ExecuteProgramProducerNode.java
new file mode 100755 (executable)
index 0000000..0c57a15
--- /dev/null
+++ b/source/mir/producer/ExecuteProgramProducerNode.java
@@ -0,0 +1,128 @@
+/*\r
+ * Copyright (C) 2001, 2002 The Mir-coders group\r
+ *\r
+ * This file is part of Mir.\r
+ *\r
+ * Mir is free software; you can redistribute it and/or modify\r
+ * it under the terms of the GNU General Public License as published by\r
+ * the Free Software Foundation; either version 2 of the License, or\r
+ * (at your option) any later version.\r
+ *\r
+ * Mir is distributed in the hope that it will be useful,\r
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
+ * GNU General Public License for more details.\r
+ *\r
+ * You should have received a copy of the GNU General Public License\r
+ * along with Mir; if not, write to the Free Software\r
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\r
+ *\r
+ * In addition, as a special exception, The Mir-coders gives permission to link\r
+ * the code of this program with  any library licensed under the Apache Software License,\r
+ * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library\r
+ * (or with modified versions of the above that use the same license as the above),\r
+ * and distribute linked combinations including the two.  You must obey the\r
+ * GNU General Public License in all respects for all of the code used other than\r
+ * the above mentioned libraries.  If you modify this file, you may extend this\r
+ * exception to your version of the file, but you are not obligated to do so.\r
+ * If you do not wish to do so, delete this exception statement from your version.\r
+ */\r
+package mir.producer;\r
+\r
+import java.util.Map;\r
+\r
+import mir.log.LoggerWrapper;\r
+import mir.util.ParameterExpander;\r
+\r
+public class ExecuteProgramProducerNode implements ProducerNode  {\r
+  private String scriptExpression;\r
+  private String maxDurationExpression;\r
+  private String outputVariableExpression;\r
+  private String returnValueVariableExpression;\r
+\r
+  public ExecuteProgramProducerNode(String aScriptExpression, String aMaxDurationExpression, String anOutputVariableExpression,\r
+                                    String aReturnValueVariableExpression) {\r
+    scriptExpression = aScriptExpression;\r
+    maxDurationExpression = aMaxDurationExpression;\r
+    outputVariableExpression = anOutputVariableExpression;\r
+    returnValueVariableExpression = aReturnValueVariableExpression;\r
+  }\r
+\r
+  public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerFailure {\r
+    String script;\r
+    long maxDuration;\r
+\r
+    try {\r
+      script = ParameterExpander.expandExpression(aValueMap, scriptExpression);\r
+      maxDuration = ParameterExpander.evaluateIntegerExpressionWithDefault(aValueMap, maxDurationExpression, 0);\r
+\r
+      ProcessRunner runner = new ProcessRunner(aLogger, script);\r
+      runner.start();\r
+\r
+      synchronized (runner) {\r
+        runner.wait(maxDuration);\r
+      }\r
+      runner.interrupt();\r
+\r
+      if (runner.getFinished()) {\r
+        aLogger.info(script + " terminated successfully, return value = " + runner.getReturnValue() + ".");\r
+\r
+        if (returnValueVariableExpression != null) {\r
+          ParameterExpander.setValueForKey(aValueMap,\r
+              ParameterExpander.expandExpression(aValueMap, returnValueVariableExpression),\r
+              new Integer(runner.getReturnValue()));\r
+        }\r
+\r
+      }\r
+      else {\r
+        aLogger.info(script + " interrupted prematurely after " + maxDuration + "ms.");\r
+      }\r
+    }\r
+    catch (Throwable e) {\r
+      aLogger.error("Error while executing " + scriptExpression + " : " + e.toString());\r
+    }\r
+  }\r
+\r
+  private static class ProcessRunner extends Thread {\r
+    private String script;\r
+    private boolean finished = false;\r
+    private int returnValue = 0;\r
+    private LoggerWrapper logger;\r
+\r
+    public ProcessRunner(LoggerWrapper aLogger, String aScript) {\r
+      script = aScript;\r
+      logger = aLogger;\r
+    }\r
+\r
+    public boolean getFinished() {\r
+      return finished;\r
+    }\r
+\r
+    public int getReturnValue() {\r
+      return returnValue;\r
+    }\r
+\r
+    public void run() {\r
+      Process process = null;\r
+      try {\r
+        process = Runtime.getRuntime().exec(script);\r
+        returnValue = process.waitFor();\r
+\r
+        finished = true;\r
+\r
+        synchronized (this) {\r
+          this.notify();\r
+        }\r
+      }\r
+      catch (InterruptedException e) {\r
+        if (process!=null) {\r
+          process.destroy();\r
+        }\r
+      }\r
+      catch (Exception e) {\r
+        logger.error(script + " failed to execute: " + e.getMessage());\r
+      }\r
+    }\r
+  }\r
+\r
+}\r