From ae42ffeb94bc351e56d2b276b7f273fefeac22d6 Mon Sep 17 00:00:00 2001 From: Heinrichs <Heinrichs@itc.rwth-aachen.de> Date: Fri, 26 Jun 2020 13:23:08 +0200 Subject: [PATCH] Add a method to create a returnobject for projectquota --- src/Database/Models/ProjectQuotasModel.cs | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/src/Database/Models/ProjectQuotasModel.cs b/src/Database/Models/ProjectQuotasModel.cs index f8d21b8..5452a3c 100644 --- a/src/Database/Models/ProjectQuotasModel.cs +++ b/src/Database/Models/ProjectQuotasModel.cs @@ -1,12 +1,34 @@ using Coscine.Database.DataModel; +using Coscine.Database.ReturnObjects; using LinqToDB; using System; +using System.Collections.Generic; using System.Linq.Expressions; namespace Coscine.Database.Models { public class ProjectQuotaModel : DatabaseModel<ProjectQuota> { + public ProjectQuotaObject CreateReturnObjectFromDatabaseObject(ProjectQuota projectQuota) + { + if (projectQuota.ResourceType == null) + { + ResourceTypeModel resourceTypeModel = new ResourceTypeModel(); + projectQuota.ResourceType = resourceTypeModel.GetById(projectQuota.ResourceTypeId); + } + + return new ProjectQuotaObject( + projectQuota.RelationId, + projectQuota.ProjectId, + new ResourceTypeObject( + projectQuota.ResourceType.Id, + projectQuota.ResourceType.DisplayName, + projectQuota.ResourceType.Enabled + ), + projectQuota.Quota + ); + } + public override Expression<Func<ProjectQuota, Guid>> GetIdFromObject() { return (projectQuota) => projectQuota.RelationId; -- GitLab