Changeset 9381 for branches/HivePerformance/sources/HeuristicLab.Services.Hive/3.3/HiveDao.cs

Timestamp:

04/18/13 16:35:14 (12 years ago)

Author:

pfleck

Message:

#2030
Activated Delayed Loading for binary data.
Added HiveOperationContext to store HiveDataContext for whole ServiceOperation duration.
Added HiveDao methods to query database objects, not DTOs.
Changed HartbeatManager to use only database objects from new queries.

File:

• branches/HivePerformance/sources/HeuristicLab.Services.Hive/3.3/HiveDao.cs (modified) (7 diffs)

branches/HivePerformance/sources/HeuristicLab.Services.Hive/3.3/HiveDao.cs

@@ -43 +43 @@
     }
 
+    public Task GetTaskDA(Guid id) {
+      var db = HiveOperationContext.Current.DataContext;
+      return db.Tasks.SingleOrDefault(x => x.TaskId == id);
+    }
+
     public IEnumerable<DT.Task> GetTasks(Expression<Func<Task, bool>> predicate) {
       using (var db = CreateContext()) {
@@ -138 +143 @@
         db.SubmitChanges();
       }
+    }
+
+    public void UpdateTaskDA(Task task) {
+      var db = HiveOperationContext.Current.DataContext;
+      db.SubmitChanges();
     }
 
@@ -195 +205 @@
     }
 
+    public IEnumerable<TaskInfoForScheduler> GetWaitingTasksDA(Slave slave) {
+      var db = HiveOperationContext.Current.DataContext;
+      var parentResources = GetParentResourcesDA(slave.ResourceId);
+      var resourceIds = parentResources.Select(x => x.ResourceId);
+      //Originally we checked here if there are parent tasks which should be calculated (with GetParentTasks(resourceIds, count, false);).
+      //Because there is at the moment no case where this makes sense (there don't exist parent tasks which need to be calculated), 
+      //we skip this step because it's wasted runtime
+
+      var query = from ar in db.AssignedResources
+                  where resourceIds.Contains(ar.ResourceId)
+                     && !(ar.Task.IsParentTask && ar.Task.FinishWhenChildJobsFinished)
+                     && ar.Task.State == TaskState.Waiting
+                     && ar.Task.CoresNeeded <= slave.FreeCores
+                     && ar.Task.MemoryNeeded <= slave.FreeMemory
+                  select new TaskInfoForScheduler() { TaskId = ar.Task.TaskId, JobId = ar.Task.JobId, Priority = ar.Task.Priority };
+      var waitingTasks = query.ToArray();
+      return waitingTasks;
+    }
+
     public DT.Task UpdateTaskState(Guid taskId, TaskState taskState, Guid? slaveId, Guid? userId, string exception) {
       using (var db = CreateContext()) {
@@ -222 +251 @@
         return DT.Convert.ToDto(task);
       }
+    }
+
+    public Task UpdateTaskStateDA(Guid taskId, TaskState taskState, Guid? slaveId, Guid? userId, string exception) {
+      var db = HiveOperationContext.Current.DataContext;
+
+      db.StateLogs.InsertOnSubmit(new StateLog {
+        TaskId = taskId,
+        State = taskState,
+        SlaveId = slaveId,
+        UserId = userId,
+        Exception = exception,
+        DateTime = DateTime.Now
+      });
+
+      var task = db.Tasks.SingleOrDefault(x => x.TaskId == taskId);
+      task.State = taskState;
+
+      db.SubmitChanges();
+
+      return task;
     }
     #endregion
@@ -516 +565 @@
     }
 
+    public Slave GetSlaveDA(Guid id) {
+      var db = HiveOperationContext.Current.DataContext;
+      return db.Resources.OfType<Slave>().SingleOrDefault(x => x.ResourceId == id);
+    }
+
     public IEnumerable<DT.Slave> GetSlaves(Expression<Func<Slave, bool>> predicate) {
       using (var db = CreateContext()) {
@@ -538 +592 @@
         db.SubmitChanges();
       }
+    }
+
+    public void UpdateSlaveDA(Slave slave) {
+      var db = HiveOperationContext.Current.DataContext;
+      db.SubmitChanges();
     }
 
@@ -664 +723 @@
         return resources.Select(r => DT.Convert.ToDto(r)).ToArray();
       }
+    }
+
+    public IEnumerable<Resource> GetParentResourcesDA(Guid resourceId) {
+      var db = HiveOperationContext.Current.DataContext;
+      var resources = new List<Resource>();
+      CollectParentResources(resources, db.Resources.Where(r => r.ResourceId == resourceId).Single());
+      return resources;
     }