Changeset 5782 for branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave/3.4/Executor.cs

Timestamp:

03/21/11 21:50:39 (14 years ago)

Author:

ascheibe

Message:

#1233

• fixed job pause bug... again
• general Executor improvements

File:

• branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave/3.4/Executor.cs (modified) (8 diffs)

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave/3.4/Executor.cs

@@ -23 +23 @@
 using System.Collections.Generic;
 using System.Linq;
+using System.Threading;
 using HeuristicLab.Common;
 using HeuristicLab.Core;
@@ -35 +36 @@
     private bool wasJobAborted = false;
     public Core Core { get; set; }
+    private Semaphore pauseStopSem = new Semaphore(0, 1);
 
     private Exception currentException;
@@ -84 +86 @@
 
     public void Pause() {
-      Job.Pause();
-
+      if (Job == null) {
+        throw new InvalidStateException("Job is null");
+      }
+
+      if (Job.ExecutionState == HeuristicLab.Core.ExecutionState.Started) {
+        try {
+          Job.Pause();
+          //we need to block the pause...
+          pauseStopSem.WaitOne();
+        }
+        catch (Exception ex) {
+          SlaveClientCom.Instance.ClientCom.LogMessage("Error pausing job:" + ex.ToString());
+        }
+      }
     }
 
     public void Stop() {
+      if (Job == null) {
+        throw new InvalidStateException("Job is null");
+      }
       wasJobAborted = true;
+
       if ((ExecutionState == ExecutionState.Started) || (ExecutionState == ExecutionState.Paused)) {
-        Job.Stop();
-      } else {
-        Job_JobStopped(this, EventArgs.Empty);
+        try {
+          Job.Stop();
+          pauseStopSem.WaitOne();
+        }
+        catch (Exception ex) {
+          SlaveClientCom.Instance.ClientCom.LogMessage("Error stopping job:" + ex.ToString());
+        }
       }
     }
@@ -103 +125 @@
       Job.WaitForChildJobs += new EventHandler(Job_WaitForChildJobs);
       Job.DeleteChildJobs += new EventHandler(Job_DeleteChildJobs);
+      Job.JobPaused += new EventHandler(Job_JobPaused);
     }
 
@@ -111 +134 @@
       Job.WaitForChildJobs -= new EventHandler(Job_WaitForChildJobs);
       Job.DeleteChildJobs -= new EventHandler(Job_DeleteChildJobs);
+      Job.JobPaused -= new EventHandler(Job_JobPaused);
     }
 
@@ -159 +183 @@
     private void Job_JobStopped(object sender, EventArgs e) {
       if (wasJobAborted) {
-        Core.EnqueueExecutorMessage(Core.KillAppDomain, JobId);
+        pauseStopSem.Release();
       } else {
+        //it's a clean and finished job, so send it
         Core.EnqueueExecutorMessage(Core.SendFinishedJob, JobId);
       }
@@ -171 +196 @@
 
       if (Job.ExecutionState == HeuristicLab.Core.ExecutionState.Started) {
-        try { Job.Stop(); }
-        catch { }
+        try {
+          Job.Stop();
+          wasJobAborted = true;
+          pauseStopSem.WaitOne();
+        }
+        catch (Exception ex) {
+          SlaveClientCom.Instance.ClientCom.LogMessage("Error stopping job:" + ex.ToString());
+        }
       }
 
@@ -178 +209 @@
     }
 
+
     public JobData GetPausedJob() {
-      if (Job == null) {
-        throw new InvalidStateException("Job is null");
-      }
-
-      if (Job.ExecutionState == HeuristicLab.Core.ExecutionState.Started) {
-        try { Job.Pause(); }
-        catch { }
-      }
-
+      if (Job.ExecutionState != HeuristicLab.Core.ExecutionState.Paused) {
+        throw new Exception("Executor: Job has to be paused before fetching results.");
+      }
       return GetJob();
+    }
+
+    private void Job_JobPaused(object sender, EventArgs e) {
+      pauseStopSem.Release();
     }