Changeset 5314 for branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave/3.4/Core.cs

Timestamp:

01/17/11 20:03:46 (14 years ago)

Author:

ascheibe

Message:

#1233

• added ItemView and Item for the Slave
• added a Tray Icon App for data visualization and control of the slave windows service
• added control methods to SlaveCommunication for controlling the slave core
• fixed typo in namespace

File:

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

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

@@ -26 +26 @@
 using System.ServiceModel;
 using System.Threading;
-using HeuristicLab.Clients.Hive.Slave;
 using HeuristicLab.Clients.Hive.Slave.ServiceContracts;
 using HeuristicLab.Common;
@@ -34 +33 @@
 
 
-namespace HeuristicLab.Clients.Hive.Salve {
+namespace HeuristicLab.Clients.Hive.Slave {
   /// <summary>
   /// The core component of the Hive Client
@@ -160 +159 @@
             ShutdownCore();
             break;
+          case MessageContainer.MessageType.HardPause:
+            doHardPause();
+            break;
+          case MessageContainer.MessageType.SoftPause:
+            doSoftPause();
+            break;
+          case MessageContainer.MessageType.Restart:
+            doRestart();
+            break;
         }
       } else {
         ClientCom.LogMessage("Unknown MessageContainer: " + container);
       }
+    }
+
+    /// <summary>
+    /// reinitializes everything and continues operation, 
+    /// can be called after SoftPause() or HardPause()
+    /// </summary>
+    public void Restart() {
+      MessageContainer mc = new MessageContainer(MessageContainer.MessageType.Restart);
+      MessageQueue.GetInstance().AddMessage(mc);
+    }
+
+    private void doRestart() {
+      ClientCom.LogMessage("Restart received");
+      StartHeartbeats();
+      ClientCom.LogMessage("Restart done");
+    }
+
+    /// <summary>
+    /// wait for jobs to finish, then pause client
+    /// </summary>
+    public void SoftPause() {
+      MessageContainer mc = new MessageContainer(MessageContainer.MessageType.SoftPause);
+      MessageQueue.GetInstance().AddMessage(mc);
+    }
+
+    private void doSoftPause() {
+      ClientCom.LogMessage("Soft pause received");
+
+      //TODO: jobs get removed from Jobs map, is this a problem?
+      foreach (Job job in Jobs.Values) {
+        engines[job.Id].Pause();
+        JobData sJob = engines[job.Id].GetFinishedJob();
+        job.Exception = engines[job.Id].CurrentException;
+        job.ExecutionTime = engines[job.Id].ExecutionTime;
+
+        try {
+          ClientCom.LogMessage("Sending the paused job with id: " + job.Id);
+          wcfService.UpdateJob(job, sJob);
+          SlaveStatusInfo.JobsProcessed++;    //TODO: count or not count, thats the question
+        }
+        catch (Exception e) {
+          ClientCom.LogMessage("Transmitting to server failed. Storing the paused job with id: " + job.Id + " to hdd (" + e.ToString() + ")");
+        }
+        finally {
+          KillAppDomain(job.Id); // kill app-domain in every case          
+        }
+      }
+
+      heartbeatManager.StopHeartBeat();
+      WcfService.Instance.Disconnect();
+      ClientCom.LogMessage("Soft pause done");
+    }
+
+    /// <summary>
+    /// pause slave immediately
+    /// </summary>
+    public void HardPause() {
+      MessageContainer mc = new MessageContainer(MessageContainer.MessageType.HardPause);
+      MessageQueue.GetInstance().AddMessage(mc);
+    }
+
+    private void doHardPause() {
+      ClientCom.LogMessage("Hard pause received");
+      heartbeatManager.StopHeartBeat();
+
+      lock (engines) {
+        ClientCom.LogMessage("engines locked");
+        foreach (KeyValuePair<Guid, AppDomain> kvp in appDomains) {
+          ClientCom.LogMessage("Shutting down Appdomain for " + kvp.Key);
+          appDomains[kvp.Key].UnhandledException -= new UnhandledExceptionEventHandler(appDomain_UnhandledException);
+          AppDomain.Unload(kvp.Value);
+        }
+      }
+      WcfService.Instance.Disconnect();
+      ClientCom.LogMessage("Hard pause done");
     }
 
@@ -172 +255 @@
     }
 
+    /// <summary>
+    /// hard shutdown, should be called before the the application is exited
+    /// </summary>
     private void ShutdownCore() {
       ClientCom.LogMessage("Shutdown Signal received");