Changeset 16622 for trunk/HeuristicLab.Clients.Hive.Slave/3.3/Core.cs

Timestamp:

02/27/19 12:44:11 (5 years ago)

Author:

jkarder

Message:

#2839: worked on hive project management

• Core.cs
  • locked checks for amount of available resources (cores/memory)
  • task state of aborted tasks is now updated accordingly
• JobDao.cs
  • improved readability of GetJobsReadyForDeletionQuery
• HiveStatisticsGenerator.cs
  • DimJobs, FactTasks and Jobs (DeletionPending) are now updated in one transaction

File:

• trunk/HeuristicLab.Clients.Hive.Slave/3.3/Core.cs (modified) (3 diffs)

trunk/HeuristicLab.Clients.Hive.Slave/3.3/Core.cs

@@ -38 +38 @@
   /// </summary>
   public class Core : MarshalByRefObject {
+    private static readonly object locker = new object();
+
     private static HeartbeatManager heartbeatManager;
     public static HeartbeatManager HeartbeatManager {
@@ -240 +242 @@
         task = wcfService.GetTask(taskId);
         if (task == null) throw new TaskNotFoundException(taskId);
-        if (ConfigManager.Instance.GetFreeCores() < task.CoresNeeded) throw new OutOfCoresException();
-        if (ConfigManager.Instance.GetFreeMemory() < task.MemoryNeeded) throw new OutOfMemoryException();
-        SlaveStatusInfo.IncrementUsedCores(task.CoresNeeded); usedCores = task.CoresNeeded;
+        lock (locker) {
+          // the amount of used cores/memory could exceed the amount of available cores/memory
+          // if HandleCalculateTask is called simultaneously from within two different tasks
+          if (ConfigManager.Instance.GetFreeCores() < task.CoresNeeded) throw new OutOfCoresException();
+          if (ConfigManager.Instance.GetFreeMemory() < task.MemoryNeeded) throw new OutOfMemoryException();
+          SlaveStatusInfo.IncrementUsedCores(task.CoresNeeded); usedCores = task.CoresNeeded;
+        }
         TaskData taskData = wcfService.GetTaskData(taskId);
         if (taskData == null) throw new TaskDataNotFoundException(taskId);
@@ -400 +406 @@
 
     private void taskManager_TaskAborted(object sender, EventArgs<SlaveTask> e) {
+      var slaveTask = e.Value;
+      var task = wcfService.GetTask(slaveTask.TaskId);
+      wcfService.UpdateJobState(task.Id, TaskState.Aborted, null);
       SlaveStatusInfo.DecrementUsedCores(e.Value.CoresNeeded);
     }