Free cookie consent management tool by TermsFeed Policy Generator

source: branches/3.3-HiveMigration/sources/HeuristicLab.Hive/HeuristicLab.Hive.Server.LINQDataAccess/3.3/JobDao.cs @ 4333

Last change on this file since 4333 was 4333, checked in by cneumuel, 12 years ago

added authorizationManager which checks for permission to specific jobs (#1168)

File size: 7.8 KB
Line 
1using System;
2using System.Collections.Generic;
3using System.Linq;
4using System.Text;
5using HeuristicLab.Hive.Contracts.BusinessObjects;
6using System.Data.Linq;
7using HeuristicLab.Hive.Server.DataAccess;
8using System.IO;
9using System.Data.SqlClient;
10
11namespace HeuristicLab.Hive.Server.LINQDataAccess {
12  public class JobDao : BaseDao<JobDto, Job>, IJobDao {
13
14    #region IGenericDao<JobDto,Job> Members
15
16    public JobDto FindById(Guid id) {
17      return (from job in Context.Jobs
18              where job.JobId.Equals(id)
19              select EntityToDto(job, null)).SingleOrDefault();
20    }
21
22    public IEnumerable<JobDto> FindAll() {
23      return (from job in Context.Jobs
24              select EntityToDto(job, null)).ToList();
25    }
26
27    public IEnumerable<JobDto> FindWithLimitations(JobState jobState, int offset, int count) {
28      IQueryable<JobDto> query = null;
29      if (jobState == JobState.Finished) {
30        query = from job in Context.Jobs
31                where job.JobState == Enum.GetName(typeof(JobState), jobState)
32                orderby job.DateFinished
33                select EntityToDto(job, null);
34      } else if (jobState == JobState.Calculating || jobState == JobState.SnapshotRequested || jobState == JobState.SnapshotSent) {
35        query = from job in Context.Jobs
36                where job.JobState == Enum.GetName(typeof(JobState), jobState)
37                orderby job.DateCalculated
38                select EntityToDto(job, null);
39      } else {
40        query = from job in Context.Jobs
41                where job.JobState == Enum.GetName(typeof(JobState), jobState)
42                orderby job.DateCreated
43                select EntityToDto(job, null);
44      }
45
46      return query.Skip(offset).Take(count).ToList();
47    }
48
49
50    public byte[] GetBinaryJobFile(Guid jobId) {
51      return (from job in Context.Jobs
52              where job.JobId.Equals(jobId)
53              select job.SerializedJob).SingleOrDefault().ToArray();
54    }
55
56    public JobDto Insert(JobDto bObj) {
57      Job j = DtoToEntity(bObj, null);
58      Context.Jobs.InsertOnSubmit(j);
59      CommitChanges();
60      bObj.Id = j.JobId;
61      return bObj;
62    }
63
64    public void SetBinaryJobFile(Guid jobId, byte[] data) {
65      Job j = (from job in Context.Jobs
66               where job.JobId.Equals(jobId)
67               select job).SingleOrDefault();
68      j.SerializedJob = data;
69      CommitChanges();
70    }
71
72    public SerializedJob InsertWithAttachedJob(SerializedJob job) {
73      Job j = DtoToEntity(job.JobInfo, null);
74      j.SerializedJob = job.SerializedJobData;
75      foreach (Guid assignRessourceId in job.JobInfo.AssignedResourceIds)
76        j.AssignedResources.Add(new AssignedResource { ResourceId = assignRessourceId });
77      Context.Jobs.InsertOnSubmit(j);
78      CommitChanges();
79      job.JobInfo.Id = j.JobId;
80      return job;
81    }
82
83    public void Delete(JobDto bObj) {
84      Job job = Context.Jobs.SingleOrDefault(j => j.JobId.Equals(bObj.Id));
85      Context.Jobs.DeleteOnSubmit(job);
86      CommitChanges();
87    }
88
89    public void Update(JobDto bObj) {
90      Job job = Context.Jobs.SingleOrDefault(j => j.JobId.Equals(bObj.Id));
91      DtoToEntity(bObj, job);
92      CommitChanges();
93    }
94
95    public IEnumerable<JobDto> FindActiveJobsOfSlave(SlaveDto slave) {
96      return (from j in Context.Jobs
97              where (j.JobState == Enum.GetName(typeof(JobState), JobState.Calculating) ||
98                     j.JobState == Enum.GetName(typeof(JobState), JobState.Aborted) ||
99                     j.JobState == Enum.GetName(typeof(JobState), JobState.SnapshotRequested) ||
100                     j.JobState == Enum.GetName(typeof(JobState), JobState.SnapshotSent)) &&
101                    (j.ResourceId.Equals(slave.Id))
102              select EntityToDto(j, null)).ToList();
103    }
104
105    public IEnumerable<JobDto> FindFittingJobsForSlave(JobState state, int freeCores, int freeMemory, Guid slaveId) {
106      SlaveGroupDao cgd = new SlaveGroupDao();
107
108      List<Guid> idList = new List<Guid>(cgd.FindAllGroupAndParentGroupIdsForSlave(slaveId));
109      //Add myself too - enables jobs for one specific host!
110      idList.Add(slaveId);
111
112      var q = (from ar in Context.AssignedResources
113               where ar.Job.JobState == Enum.GetName(typeof(JobState), JobState.Offline) &&
114                     ar.Job.CoresNeeded <= freeCores &&
115                     ar.Job.MemoryNeeded <= freeMemory &&
116                     idList.Contains(ar.ResourceId)
117               orderby ar.Job.Priority descending
118               select EntityToDto(ar.Job, null));
119      return q.ToList();
120    }
121
122    public IEnumerable<JobDto> GetJobsByState(JobState state) {
123      return (from j in Context.Jobs
124              where (j.JobState == Enum.GetName(typeof(JobState), state))
125              select EntityToDto(j, null)).ToList();
126    }
127
128    public void AssignSlaveToJob(Guid slaveId, Guid jobId) {
129      Slave c = Context.Resources.OfType<Slave>().SingleOrDefault(slave => slave.ResourceId.Equals(slaveId));
130      Job j = Context.Jobs.SingleOrDefault(job => job.JobId.Equals(jobId));
131      c.Jobs.Add(j);
132      j.Slave = c;
133      CommitChanges();
134    }
135
136    public void SetJobOffline(JobDto job) {
137      Job j = Context.Jobs.SingleOrDefault(jq => jq.JobId.Equals(job.Id));
138      j.Slave = null;
139      j.JobState = Enum.GetName(typeof(JobState), JobState.Offline);
140      CommitChanges();
141    }
142
143    public Stream GetSerializedJobStream(Guid jobId) {
144      VarBinarySource source = new VarBinarySource((SqlConnection)Context.Connection, null, "Job", "SerializedJob", "JobId", jobId);
145      return new VarBinaryStream(source);
146    }
147
148    public IEnumerable<JobDto> FindJobsById(IEnumerable<Guid> jobIds) {
149      IQueryable<JobDto> jobs = from job in Context.Jobs
150                                where jobIds.Contains(job.JobId)
151                                select EntityToDto(job, null);
152      return jobs.ToList();
153    }
154
155    public bool IsUserAuthorizedForJobs(string userId, params Guid[] jobIds) {
156      var jobs = from job in Context.Jobs
157                 where jobIds.Contains(job.JobId)
158                 select job;
159      return jobs.All(job => job.UserId == userId);
160    }
161    #endregion
162
163    public override Job DtoToEntity(JobDto source, Job target) {
164      if (source == null)
165        return null;
166      if (target == null)
167        target = new Job();
168
169      target.CoresNeeded = source.CoresNeeded;
170      target.MemoryNeeded = source.MemoryNeeded;
171
172      target.DateCalculated = source.DateCalculated;
173      target.DateCreated = source.DateCreated;
174      target.DateFinished = source.DateFinished;
175      target.JobId = source.Id;
176
177      target.Percentage = source.Percentage;
178      target.Exception = source.Exception;
179
180      target.Priority = source.Priority;
181      target.JobState = Enum.GetName(typeof(JobState), source.State);
182      target.UserId = source.UserId;
183      return target;
184    }
185
186    //Assigned ressources are not used atm!
187    //Slave is not used ATM - not sure when we stopped using those...
188    public override JobDto EntityToDto(Job source, JobDto target) {
189      if (source == null)
190        return null;
191      if (target == null)
192        target = new JobDto();
193
194      target.CoresNeeded = source.CoresNeeded;
195      target.MemoryNeeded = source.MemoryNeeded;
196
197      target.DateCalculated = source.DateCalculated;
198      target.DateCreated = source.DateCreated;
199      target.DateFinished = source.DateFinished;
200      target.Id = source.JobId;
201     
202      target.Exception = source.Exception;
203      target.Percentage = source.Percentage;
204
205      target.Priority = source.Priority;
206      target.State = (JobState)Enum.Parse(typeof(JobState), source.JobState, true);
207      target.UserId = source.UserId;
208      return target;
209    }
210  }
211}
Note: See TracBrowser for help on using the repository browser.