#region License Information
/* HeuristicLab
* Copyright (C) 2002-2012 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
*
* This file is part of HeuristicLab.
*
* HeuristicLab is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* HeuristicLab is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with HeuristicLab. If not, see .
*/
#endregion
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using DT = HeuristicLab.Services.Hive.DataTransfer;
using System.ServiceModel;
using HeuristicLab.Services.Hive.Interfaces;
using System.Data.Common;
namespace HeuristicLab.Services.Hive.DataAccess {
public class HiveDao : IHiveDao {
public HiveDataContext CreateContext(bool longRunning = false) {
//var context = new HiveDataContext(Settings.Default.HeuristicLab_Hive_LinqConnectionString);
var context = new HiveDataContext(provider.GetOpenConnection(Settings.Default.HiveConnectionStringName));
if (longRunning) context.CommandTimeout = (int)Settings.Default.LongRunningDatabaseCommandTimeout.TotalSeconds;
return context;
}
private IConnectionProvider provider;
public HiveDao(IConnectionProvider provider) { this.provider = provider; }
private delegate T ExecuteWithContextDelegate(HiveDataContext context);
private delegate void ExecuteWithContextDelegate(HiveDataContext context);
private void ExecuteWithContext(ExecuteWithContextDelegate call) {
DbConnection con = null;
try {
using (var db = CreateContext()) {
con = db.Connection;
call(db);
}
}
finally {
provider.ReleaseConnection(con);
}
}
private T ExecuteWithContext(ExecuteWithContextDelegate call) {
DbConnection con = null;
try {
using (var db = CreateContext()) {
con = db.Connection;
T result = call(db);
return result;
}
}
finally {
provider.ReleaseConnection(con);
}
}
#region Task Methods
public DT.Task GetTask(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Tasks.SingleOrDefault(x => x.TaskId == id));
});
}
public IEnumerable GetTasks(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Tasks.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddTask(DT.Task dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Tasks.InsertOnSubmit(entity);
db.SubmitChanges();
foreach (Guid pluginId in dto.PluginsNeededIds) {
db.RequiredPlugins.InsertOnSubmit(new RequiredPlugin() { TaskId = entity.TaskId, PluginId = pluginId });
}
db.SubmitChanges();
return entity.TaskId;
});
}
public void UpdateTask(DT.Task dto) {
ExecuteWithContext((db) => {
var entity = db.Tasks.FirstOrDefault(x => x.TaskId == dto.Id);
if (entity == null) db.Tasks.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
foreach (Guid pluginId in dto.PluginsNeededIds) {
if (db.RequiredPlugins.Count(p => p.PluginId == pluginId) == 0) {
db.RequiredPlugins.InsertOnSubmit(new RequiredPlugin() { TaskId = entity.TaskId, PluginId = pluginId });
}
}
db.SubmitChanges();
});
}
public void DeleteTask(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Tasks.FirstOrDefault(x => x.TaskId == id);
if (entity != null) db.Tasks.DeleteOnSubmit(entity);
db.SubmitChanges(); // taskData and child tasks are deleted by db-trigger
});
}
///
/// returns all parent tasks which are waiting for their child tasks to finish
///
/// list of resourceids which for which the task should be valid
/// maximum number of task to return
/// if true, all parent task which have FinishWhenChildJobsFinished=true are returned, otherwise only FinishWhenChildJobsFinished=false are returned
///
public IEnumerable GetParentTasks(IEnumerable resourceIds, int count, bool finished) {
return ExecuteWithContext>((db) => {
var query = from ar in db.AssignedResources
where resourceIds.Contains(ar.ResourceId)
&& ar.Task.State == TaskState.Waiting
&& ar.Task.IsParentTask
&& (finished ? ar.Task.FinishWhenChildJobsFinished : !ar.Task.FinishWhenChildJobsFinished)
&& (from child in db.Tasks
where child.ParentTaskId == ar.Task.TaskId
select child.State == TaskState.Finished
|| child.State == TaskState.Aborted
|| child.State == TaskState.Failed).All(x => x)
&& (from child in db.Tasks // avoid returning WaitForChildTasks task where no child-task exist (yet)
where child.ParentTaskId == ar.Task.TaskId
select child).Count() > 0
orderby ar.Task.Priority descending, db.Random()
select DT.Convert.ToDto(ar.Task);
return count == 0 ? query.ToArray() : query.Take(count).ToArray();
});
}
public IEnumerable GetWaitingTasks(DT.Slave slave, int count) {
return ExecuteWithContext>((db) => {
var resourceIds = GetParentResources(slave.Id).Select(r => r.Id);
//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
orderby ar.Task.Priority descending, db.Random() // take random task to avoid the race condition that occurs when this method is called concurrently (the same task would be returned)
select DT.Convert.ToDto(ar.Task);
var waitingTasks = (count == 0 ? query : query.Take(count)).ToArray();
return waitingTasks;
});
}
public DT.Task UpdateTaskState(Guid taskId, TaskState taskState, Guid? slaveId, Guid? userId, string exception) {
return ExecuteWithContext((db) => {
var job = db.Tasks.SingleOrDefault(x => x.TaskId == taskId);
job.State = taskState;
db.StateLogs.InsertOnSubmit(new StateLog {
TaskId = taskId,
State = taskState,
SlaveId = slaveId,
UserId = userId,
Exception = exception,
DateTime = DateTime.Now
});
db.SubmitChanges();
job = db.Tasks.SingleOrDefault(x => x.TaskId == taskId);
return DT.Convert.ToDto(job);
});
}
#endregion
#region TaskData Methods
public DT.TaskData GetTaskData(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.TaskDatas.SingleOrDefault(x => x.TaskId == id));
});
}
public IEnumerable GetTaskDatas(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.TaskDatas.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddTaskData(DT.TaskData dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.TaskDatas.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.TaskId;
});
}
public void UpdateTaskData(DT.TaskData dto) {
ExecuteWithContext((db) => {
var entity = db.TaskDatas.FirstOrDefault(x => x.TaskId == dto.TaskId);
if (entity == null) db.TaskDatas.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteTaskData(Guid id) {
ExecuteWithContext((db) => {
var entity = db.TaskDatas.FirstOrDefault(x => x.TaskId == id); // check if all the byte[] is loaded into memory here. otherwise work around to delete without loading it
if (entity != null) db.TaskDatas.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region StateLog Methods
public DT.StateLog GetStateLog(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.StateLogs.SingleOrDefault(x => x.StateLogId == id));
});
}
public IEnumerable GetStateLogs(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.StateLogs.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddStateLog(DT.StateLog dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.StateLogs.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.StateLogId;
});
}
public void UpdateStateLog(DT.StateLog dto) {
ExecuteWithContext((db) => {
var entity = db.StateLogs.FirstOrDefault(x => x.StateLogId == dto.Id);
if (entity == null) db.StateLogs.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteStateLog(Guid id) {
ExecuteWithContext((db) => {
var entity = db.StateLogs.FirstOrDefault(x => x.StateLogId == id);
if (entity != null) db.StateLogs.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region Job Methods
public DT.Job GetJob(Guid id) {
return ExecuteWithContext((db) => {
return AddStatsToJob(db, DT.Convert.ToDto(db.Jobs.SingleOrDefault(x => x.JobId == id)));
});
}
private DT.Job AddStatsToJob(HiveDataContext db, DT.Job exp) {
if (exp == null)
return null;
var jobs = db.Tasks.Where(j => j.JobId == exp.Id);
exp.JobCount = jobs.Count();
exp.CalculatingCount = jobs.Count(j => j.State == TaskState.Calculating);
exp.FinishedCount = jobs.Count(j => j.State == TaskState.Finished);
return exp;
}
public IEnumerable GetJobs(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Jobs.Where(predicate).Select(x => AddStatsToJob(db, DT.Convert.ToDto(x))).ToArray();
});
}
public Guid AddJob(DT.Job dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Jobs.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.JobId;
});
}
public void UpdateJob(DT.Job dto) {
ExecuteWithContext((db) => {
var entity = db.Jobs.FirstOrDefault(x => x.JobId == dto.Id);
if (entity == null) db.Jobs.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteJob(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Jobs.FirstOrDefault(x => x.JobId == id);
if (entity != null) db.Jobs.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region JobPermission Methods
public DT.JobPermission GetJobPermission(Guid jobId, Guid grantedUserId) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.JobPermissions.SingleOrDefault(x => x.JobId == jobId && x.GrantedUserId == grantedUserId));
});
}
public IEnumerable GetJobPermissions(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.JobPermissions.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public void AddJobPermission(DT.JobPermission dto) {
ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.JobPermissions.InsertOnSubmit(entity);
db.SubmitChanges();
});
}
public void UpdateJobPermission(DT.JobPermission dto) {
ExecuteWithContext((db) => {
var entity = db.JobPermissions.FirstOrDefault(x => x.JobId == dto.JobId && x.GrantedUserId == dto.GrantedUserId);
if (entity == null) db.JobPermissions.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteJobPermission(Guid jobId, Guid grantedUserId) {
ExecuteWithContext((db) => {
var entity = db.JobPermissions.FirstOrDefault(x => x.JobId == jobId && x.GrantedUserId == grantedUserId);
if (entity != null) db.JobPermissions.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
///
/// Sets the permissions for a experiment. makes sure that only one permission per user exists.
///
public void SetJobPermission(Guid jobId, Guid grantedByUserId, Guid grantedUserId, Permission permission) {
ExecuteWithContext((db) => {
JobPermission jobPermission = db.JobPermissions.SingleOrDefault(x => x.JobId == jobId && x.GrantedUserId == grantedUserId);
if (jobPermission != null) {
if (permission == Permission.NotAllowed) {
// not allowed, delete
db.JobPermissions.DeleteOnSubmit(jobPermission);
}
else {
// update
jobPermission.Permission = permission;
jobPermission.GrantedByUserId = grantedByUserId; // update grantedByUserId, always the last "granter" is stored
}
}
else {
// insert
if (permission != Permission.NotAllowed) {
jobPermission = new JobPermission() { JobId = jobId, GrantedByUserId = grantedByUserId, GrantedUserId = grantedUserId, Permission = permission };
db.JobPermissions.InsertOnSubmit(jobPermission);
}
}
db.SubmitChanges();
});
}
#endregion
#region Plugin Methods
public DT.Plugin GetPlugin(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Plugins.SingleOrDefault(x => x.PluginId == id));
});
}
public IEnumerable GetPlugins(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Plugins.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddPlugin(DT.Plugin dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Plugins.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.PluginId;
});
}
public void UpdatePlugin(DT.Plugin dto) {
ExecuteWithContext((db) => {
var entity = db.Plugins.FirstOrDefault(x => x.PluginId == dto.Id);
if (entity == null) db.Plugins.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeletePlugin(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Plugins.FirstOrDefault(x => x.PluginId == id);
if (entity != null) db.Plugins.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region PluginData Methods
public DT.PluginData GetPluginData(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.PluginDatas.SingleOrDefault(x => x.PluginDataId == id));
});
}
public IEnumerable GetPluginDatas(Expression> predicate) {
return ExecuteWithContext >((db) => {
return db.PluginDatas.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddPluginData(DT.PluginData dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.PluginDatas.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.PluginDataId;
});
}
public void UpdatePluginData(DT.PluginData dto) {
ExecuteWithContext((db) => {
var entity = db.PluginDatas.FirstOrDefault(x => x.PluginId == dto.PluginId);
if (entity == null) db.PluginDatas.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeletePluginData(Guid id) {
ExecuteWithContext((db) => {
var entity = db.PluginDatas.FirstOrDefault(x => x.PluginDataId == id);
if (entity != null) db.PluginDatas.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region Slave Methods
public DT.Slave GetSlave(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Resources.OfType().SingleOrDefault(x => x.ResourceId == id));
});
}
public IEnumerable GetSlaves(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Resources.OfType().Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddSlave(DT.Slave dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Resources.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.ResourceId;
});
}
public void UpdateSlave(DT.Slave dto) {
ExecuteWithContext((db) => {
var entity = db.Resources.OfType().FirstOrDefault(x => x.ResourceId == dto.Id);
if (entity == null) db.Resources.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteSlave(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Resources.OfType().FirstOrDefault(x => x.ResourceId == id);
if (entity != null) db.Resources.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region SlaveGroup Methods
public DT.SlaveGroup GetSlaveGroup(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Resources.OfType().SingleOrDefault(x => x.ResourceId == id));
});
}
public IEnumerable GetSlaveGroups(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Resources.OfType().Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddSlaveGroup(DT.SlaveGroup dto) {
return ExecuteWithContext((db) => {
if (dto.Id == Guid.Empty)
dto.Id = Guid.NewGuid();
var entity = DT.Convert.ToEntity(dto);
db.Resources.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.ResourceId;
});
}
public void UpdateSlaveGroup(DT.SlaveGroup dto) {
ExecuteWithContext((db) => {
var entity = db.Resources.OfType().FirstOrDefault(x => x.ResourceId == dto.Id);
if (entity == null) db.Resources.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteSlaveGroup(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Resources.OfType().FirstOrDefault(x => x.ResourceId == id);
if (entity != null) {
if (db.Resources.Where(r => r.ParentResourceId == id).Count() > 0) {
throw new InvalidOperationException("Cannot delete SlaveGroup as long as there are Slaves in the group");
}
db.Resources.DeleteOnSubmit(entity);
}
db.SubmitChanges();
});
}
#endregion
#region Resource Methods
public DT.Resource GetResource(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Resources.SingleOrDefault(x => x.ResourceId == id));
});
}
public IEnumerable GetResources(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Resources.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddResource(DT.Resource dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Resources.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.ResourceId;
});
}
public void UpdateResource(DT.Resource dto) {
ExecuteWithContext((db) => {
var entity = db.Resources.FirstOrDefault(x => x.ResourceId == dto.Id);
if (entity == null) db.Resources.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteResource(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Resources.FirstOrDefault(x => x.ResourceId == id);
if (entity != null) db.Resources.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
public void AssignJobToResource(Guid jobId, Guid resourceId) {
ExecuteWithContext((db) => {
var job = db.Tasks.Where(x => x.TaskId == jobId).Single();
job.AssignedResources.Add(new AssignedResource() { TaskId = jobId, ResourceId = resourceId });
db.SubmitChanges();
});
}
public IEnumerable GetAssignedResources(Guid jobId) {
return ExecuteWithContext >((db) => {
var job = db.Tasks.Where(x => x.TaskId == jobId).Single();
return job.AssignedResources.Select(x => DT.Convert.ToDto(x.Resource)).ToArray();
});
}
///
/// Returns all parent resources of a resource (the given resource is also added)
///
public IEnumerable GetParentResources(Guid resourceId) {
return ExecuteWithContext>((db) => {
var resources = new List();
CollectParentResources(resources, db.Resources.Where(r => r.ResourceId == resourceId).Single());
return resources.Select(r => DT.Convert.ToDto(r)).ToArray();
});
}
private void CollectParentResources(List resources, Resource resource) {
if (resource == null) return;
resources.Add(resource);
CollectParentResources(resources, resource.ParentResource);
}
///
/// Returns all child resources of a resource (without the given resource)
///
public IEnumerable GetChildResources(Guid resourceId) {
return ExecuteWithContext>((db) => {
var childs = new List();
foreach (var child in db.Resources.Where(x => x.ParentResourceId == resourceId)) {
childs.Add(DT.Convert.ToDto(child));
childs.AddRange(GetChildResources(child.ResourceId));
}
return childs;
});
}
public IEnumerable GetJobsByResourceId(Guid resourceId) {
return ExecuteWithContext>((db) => {
var resources = GetChildResources(resourceId).Select(x => x.Id).ToList();
resources.Add(resourceId);
var jobs = db.Tasks.Where(j =>
j.State == TaskState.Calculating &&
j.StateLogs.OrderByDescending(x => x.DateTime).First().SlaveId.HasValue &&
resources.Contains(j.StateLogs.OrderByDescending(x => x.DateTime).First().SlaveId.Value));
return jobs.Select(j => DT.Convert.ToDto(j)).ToArray();
});
}
#endregion
#region ResourcePermission Methods
public DT.ResourcePermission GetResourcePermission(Guid resourceId, Guid grantedUserId) {
return ExecuteWithContext ((db) => {
return DT.Convert.ToDto(db.ResourcePermissions.SingleOrDefault(x => x.ResourceId == resourceId && x.GrantedUserId == grantedUserId));
});
}
public IEnumerable GetResourcePermissions(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.ResourcePermissions.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public void AddResourcePermission(DT.ResourcePermission dto) {
ExecuteWithContext((db) => {
var entity = db.ResourcePermissions.SingleOrDefault(x => x.ResourceId == dto.ResourceId && x.GrantedUserId == dto.GrantedUserId);
if (entity == null) { db.ResourcePermissions.InsertOnSubmit(DT.Convert.ToEntity(dto)); db.SubmitChanges(); }
});
}
public void UpdateResourcePermission(DT.ResourcePermission dto) {
ExecuteWithContext((db) => {
var entity = db.ResourcePermissions.FirstOrDefault(x => x.ResourceId == dto.ResourceId && x.GrantedUserId == dto.GrantedUserId);
if (entity == null) db.ResourcePermissions.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteResourcePermission(Guid resourceId, Guid grantedUserId) {
ExecuteWithContext((db) => {
var entity = db.ResourcePermissions.FirstOrDefault(x => x.ResourceId == resourceId && x.GrantedUserId == grantedUserId);
if (entity != null) db.ResourcePermissions.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region Authorization Methods
public Permission GetPermissionForTask(Guid taskId, Guid userId) {
return ExecuteWithContext((db) => {
return GetPermissionForJob(GetJobForTask(taskId), userId);
});
}
public Permission GetPermissionForJob(Guid jobId, Guid userId) {
return ExecuteWithContext((db) => {
Job job = db.Jobs.SingleOrDefault(x => x.JobId == jobId);
if (job == null) return Permission.NotAllowed;
if (job.OwnerUserId == userId) return Permission.Full;
JobPermission permission = db.JobPermissions.SingleOrDefault(p => p.JobId == jobId && p.GrantedUserId == userId);
return permission != null ? permission.Permission : Permission.NotAllowed;
});
}
public Guid GetJobForTask(Guid taskId) {
return ExecuteWithContext((db) => {
return db.Tasks.Single(j => j.TaskId == taskId).JobId;
});
}
#endregion
#region Lifecycle Methods
public DateTime GetLastCleanup() {
return ExecuteWithContext((db) => {
var entity = db.Lifecycles.SingleOrDefault();
return entity != null ? entity.LastCleanup : DateTime.MinValue;
});
}
public void SetLastCleanup(DateTime datetime) {
ExecuteWithContext((db) => {
var entity = db.Lifecycles.SingleOrDefault();
if (entity != null) {
entity.LastCleanup = datetime;
}
else {
entity = new Lifecycle();
entity.LifecycleId = 0; // always only one entry with ID:0
entity.LastCleanup = datetime;
db.Lifecycles.InsertOnSubmit(entity);
}
db.SubmitChanges();
});
}
#endregion
#region Downtime Methods
public DT.Downtime GetDowntime(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Downtimes.SingleOrDefault(x => x.DowntimeId == id));
});
}
public IEnumerable GetDowntimes(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Downtimes.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddDowntime(DT.Downtime dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Downtimes.InsertOnSubmit(entity);
db.SubmitChanges();
return entity.DowntimeId;
});
}
public void UpdateDowntime(DT.Downtime dto) {
ExecuteWithContext((db) => {
var entity = db.Downtimes.FirstOrDefault(x => x.DowntimeId == dto.Id);
if (entity == null) db.Downtimes.InsertOnSubmit(DT.Convert.ToEntity(dto));
else DT.Convert.ToEntity(dto, entity);
db.SubmitChanges();
});
}
public void DeleteDowntime(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Downtimes.FirstOrDefault(x => x.DowntimeId == id);
if (entity != null) db.Downtimes.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
#endregion
#region Statistics Methods
public DT.Statistics GetStatistic(Guid id) {
return ExecuteWithContext((db) => {
return DT.Convert.ToDto(db.Statistics.SingleOrDefault(x => x.StatisticsId == id));
});
}
public IEnumerable GetStatistics(Expression> predicate) {
return ExecuteWithContext>((db) => {
return db.Statistics.Where(predicate).Select(x => DT.Convert.ToDto(x)).ToArray();
});
}
public Guid AddStatistics(DT.Statistics dto) {
return ExecuteWithContext((db) => {
var entity = DT.Convert.ToEntity(dto);
db.Statistics.InsertOnSubmit(entity);
db.SubmitChanges();
foreach (var slaveStat in dto.SlaveStatistics) {
slaveStat.Id = entity.StatisticsId;
db.SlaveStatistics.InsertOnSubmit(DT.Convert.ToEntity(slaveStat));
}
if (dto.UserStatistics != null) {
foreach (var userStat in dto.UserStatistics) {
userStat.Id = entity.StatisticsId;
db.UserStatistics.InsertOnSubmit(DT.Convert.ToEntity(userStat));
}
}
db.SubmitChanges();
return entity.StatisticsId;
});
}
public void DeleteStatistics(Guid id) {
ExecuteWithContext((db) => {
var entity = db.Statistics.FirstOrDefault(x => x.StatisticsId == id);
if (entity != null) db.Statistics.DeleteOnSubmit(entity);
db.SubmitChanges();
});
}
public List GetUserStatistics() {
return ExecuteWithContext>((db) => {
var userStats = new Dictionary();
var usedCoresByUser = from job in db.Tasks
where job.State == TaskState.Calculating
group job by job.Job.OwnerUserId into g
select new { UserId = g.Key, UsedCores = g.Count() };
foreach (var item in usedCoresByUser) {
if (!userStats.ContainsKey(item.UserId)) {
userStats.Add(item.UserId, new DT.UserStatistics() { UserId = item.UserId });
}
userStats[item.UserId].UsedCores += item.UsedCores;
}
var executionTimesByUser = from task in db.Tasks
group task by task.Job.OwnerUserId into g
select new { UserId = g.Key, ExecutionTime = TimeSpan.FromMilliseconds(g.Select(x => x.ExecutionTimeMs).Sum()) };
foreach (var item in executionTimesByUser) {
if (!userStats.ContainsKey(item.UserId)) {
userStats.Add(item.UserId, new DT.UserStatistics() { UserId = item.UserId });
}
userStats[item.UserId].ExecutionTime += item.ExecutionTime;
}
// execution times only of finished task - necessary to compute efficieny
var executionTimesFinishedJobs = from job in db.Tasks
where job.State == TaskState.Finished
group job by job.Job.OwnerUserId into g
select new { UserId = g.Key, ExecutionTimeFinishedJobs = TimeSpan.FromMilliseconds(g.Select(x => x.ExecutionTimeMs).Sum()) };
foreach (var item in executionTimesFinishedJobs) {
if (!userStats.ContainsKey(item.UserId)) {
userStats.Add(item.UserId, new DT.UserStatistics() { UserId = item.UserId });
}
userStats[item.UserId].ExecutionTimeFinishedJobs += item.ExecutionTimeFinishedJobs;
}
// start to end times only of finished task - necessary to compute efficiency
var startToEndTimesFinishedJobs = from job in db.Tasks
where job.State == TaskState.Finished
group job by job.Job.OwnerUserId into g
select new {
UserId = g.Key,
StartToEndTime = new TimeSpan(g.Select(x => x.StateLogs.OrderByDescending(sl => sl.DateTime).First().DateTime - x.StateLogs.OrderBy(sl => sl.DateTime).First().DateTime).Sum(ts => ts.Ticks))
};
foreach (var item in startToEndTimesFinishedJobs) {
if (!userStats.ContainsKey(item.UserId)) {
userStats.Add(item.UserId, new DT.UserStatistics() { UserId = item.UserId });
}
userStats[item.UserId].StartToEndTime += item.StartToEndTime;
}
// also consider executiontimes of DeletedJobStats
var deletedJobsExecutionTimesByUsers = from del in db.DeletedJobStatistics
group del by del.UserId into g
select new {
UserId = g.Key,
ExecutionTime = TimeSpan.FromSeconds(g.Select(x => x.ExecutionTimeS).Sum()),
ExecutionTimeFinishedJobs = TimeSpan.FromSeconds(g.Select(x => x.ExecutionTimeSFinishedJobs).Sum()),
StartToEndTime = TimeSpan.FromSeconds(g.Select(x => x.StartToEndTimeS).Sum())
};
foreach (var item in deletedJobsExecutionTimesByUsers) {
if (!userStats.ContainsKey(item.UserId)) {
userStats.Add(item.UserId, new DT.UserStatistics() { UserId = item.UserId });
}
userStats[item.UserId].ExecutionTime += item.ExecutionTime;
userStats[item.UserId].ExecutionTimeFinishedJobs += item.ExecutionTimeFinishedJobs;
userStats[item.UserId].StartToEndTime += item.StartToEndTime;
}
return userStats.Values.ToList();
});
}
#endregion
#region Helpers
private void CollectChildTasks(HiveDataContext db, Guid parentTaskId, List collection) {
var tasks = db.Tasks.Where(j => j.ParentTaskId == parentTaskId);
foreach (var task in tasks) {
collection.Add(task);
if (task.IsParentTask)
CollectChildTasks(db, task.TaskId, collection);
}
}
#endregion
}
}