[12468] | 1 | #region License Information
|
---|
| 2 | /* HeuristicLab
|
---|
[15584] | 3 | * Copyright (C) 2002-2018 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
|
---|
[12468] | 4 | *
|
---|
| 5 | * This file is part of HeuristicLab.
|
---|
| 6 | *
|
---|
| 7 | * HeuristicLab is free software: you can redistribute it and/or modify
|
---|
| 8 | * it under the terms of the GNU General Public License as published by
|
---|
| 9 | * the Free Software Foundation, either version 3 of the License, or
|
---|
| 10 | * (at your option) any later version.
|
---|
| 11 | *
|
---|
| 12 | * HeuristicLab is distributed in the hope that it will be useful,
|
---|
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
| 15 | * GNU General Public License for more details.
|
---|
| 16 | *
|
---|
| 17 | * You should have received a copy of the GNU General Public License
|
---|
| 18 | * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
|
---|
| 19 | */
|
---|
| 20 | #endregion
|
---|
| 21 |
|
---|
| 22 | using System;
|
---|
| 23 | using System.Collections.Generic;
|
---|
| 24 | using System.Data.Linq;
|
---|
| 25 | using System.Linq;
|
---|
| 26 |
|
---|
| 27 | namespace HeuristicLab.Services.Hive.DataAccess.Daos.HiveStatistics {
|
---|
| 28 | public class FactTaskDao : GenericDao<Guid, FactTask> {
|
---|
[12551] | 29 |
|
---|
| 30 | private static readonly TaskState[] CompletedStates = { TaskState.Finished, TaskState.Aborted, TaskState.Failed };
|
---|
| 31 |
|
---|
| 32 | private Table<DimClient> DimClientTable {
|
---|
| 33 | get { return DataContext.GetTable<DimClient>(); }
|
---|
| 34 | }
|
---|
| 35 | private Table<DimJob> DimJobTable {
|
---|
| 36 | get { return DataContext.GetTable<DimJob>(); }
|
---|
| 37 | }
|
---|
| 38 |
|
---|
[12468] | 39 | public FactTaskDao(DataContext dataContext) : base(dataContext) { }
|
---|
| 40 |
|
---|
| 41 | public override FactTask GetById(Guid id) {
|
---|
[12691] | 42 | return GetByIdQuery(DataContext, id);
|
---|
[12468] | 43 | }
|
---|
| 44 |
|
---|
| 45 | public IQueryable<FactTask> GetNotFinishedTasks() {
|
---|
| 46 | return Table.Where(x => x.EndTime == null);
|
---|
| 47 | }
|
---|
| 48 |
|
---|
[12477] | 49 | public IQueryable<FactTask> GetByJobId(Guid id) {
|
---|
| 50 | return Table.Where(x => x.JobId == id);
|
---|
| 51 | }
|
---|
| 52 |
|
---|
[12516] | 53 | public DateTime? GetLastCompletedTaskFromJob(Guid id) {
|
---|
[12691] | 54 | return GetLastCompletedTaskFromJobQuery(DataContext, id);
|
---|
[12516] | 55 | }
|
---|
| 56 |
|
---|
| 57 | public IQueryable<FactTask> GetByClientId(Guid id) {
|
---|
| 58 | return Table.Where(x => x.LastClientId == id);
|
---|
| 59 | }
|
---|
| 60 |
|
---|
[12551] | 61 | public IQueryable<FactTask> GetTasksWithException() {
|
---|
| 62 | return Table.Where(x => x.Exception.Trim() != string.Empty);
|
---|
| 63 | }
|
---|
| 64 |
|
---|
| 65 | public IQueryable<FactTask> GetByGroupId(Guid id) {
|
---|
| 66 | return from factTask in Table
|
---|
| 67 | join client in DimClientTable on factTask.LastClientId equals client.Id
|
---|
| 68 | where client.ResourceGroupId == id
|
---|
| 69 | select factTask;
|
---|
| 70 | }
|
---|
| 71 |
|
---|
| 72 | public IQueryable<FactTask> GetByUserId(Guid id) {
|
---|
| 73 | return from factTask in Table
|
---|
| 74 | join job in DimJobTable on factTask.JobId equals job.JobId
|
---|
| 75 | where job.UserId == id
|
---|
| 76 | select factTask;
|
---|
| 77 | }
|
---|
| 78 |
|
---|
| 79 | public IQueryable<FactTask> GetCompletedTasks() {
|
---|
| 80 | return Table.Where(x => CompletedStates.Contains(x.TaskState));
|
---|
| 81 | }
|
---|
| 82 |
|
---|
[12468] | 83 | public override void Delete(IEnumerable<Guid> ids) {
|
---|
| 84 | string paramIds = string.Join(",", ids.Select(x => string.Format("'{0}'", x)));
|
---|
| 85 | if (!string.IsNullOrWhiteSpace(paramIds)) {
|
---|
| 86 | string query = string.Format(BatchDeleteQuery, paramIds);
|
---|
| 87 | DataContext.ExecuteCommand(query);
|
---|
| 88 | }
|
---|
| 89 | }
|
---|
| 90 |
|
---|
[12761] | 91 | public void DeleteByJobId(Guid jobId) {
|
---|
| 92 | DataContext.ExecuteCommand(DeleteByJobIdQuery, jobId);
|
---|
| 93 | }
|
---|
| 94 |
|
---|
[12691] | 95 | #region Compiled queries
|
---|
| 96 | private static readonly Func<DataContext, Guid, FactTask> GetByIdQuery =
|
---|
| 97 | CompiledQuery.Compile((DataContext db, Guid id) =>
|
---|
| 98 | (from factTask in db.GetTable<FactTask>()
|
---|
| 99 | where factTask.TaskId == id
|
---|
| 100 | select factTask).SingleOrDefault());
|
---|
| 101 |
|
---|
| 102 |
|
---|
| 103 | private static readonly Func<DataContext, Guid, DateTime?> GetLastCompletedTaskFromJobQuery =
|
---|
| 104 | CompiledQuery.Compile((DataContext db, Guid id) =>
|
---|
| 105 | (from factTask in db.GetTable<FactTask>()
|
---|
| 106 | where factTask.JobId == id && factTask.EndTime != null
|
---|
| 107 | select factTask.EndTime).Max());
|
---|
| 108 | #endregion
|
---|
| 109 |
|
---|
| 110 | #region String queries
|
---|
| 111 | private const string BatchDeleteQuery =
|
---|
| 112 | @"DELETE FROM [statistics].[FactTask]
|
---|
| 113 | WHERE TaskId IN ({0});";
|
---|
[12761] | 114 |
|
---|
| 115 | private const string DeleteByJobIdQuery =
|
---|
| 116 | @"DELETE FROM [statistics].[FactTask]
|
---|
| 117 | WHERE JobId = {0};";
|
---|
[12691] | 118 | #endregion
|
---|
[12468] | 119 | }
|
---|
| 120 | }
|
---|