aleksandr chegodaev 38f0e3fcd5 lab3
2024-04-13 02:17:11 +04:00

111 lines
3.5 KiB
C#

using LawFirmContracts.BindingModels;
using LawFirmContracts.SearchModels;
using LawFirmContracts.StoragesContracts;
using LawFirmContracts.ViewModels;
using LawFirmDatabaseImplement.Models;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace LawFirmDatabaseImplement.Implements
{
public class LawStorage : ILawStorage
{
public List<LawViewModel> GetFullList()
{
using var context = new LawFirmDataBase();
return context.Laws
.Include(x => x.Components)
.ThenInclude(x => x.Component)
.ToList()
.Select(x => x.GetViewModel)
.ToList();
}
public List<LawViewModel> GetFilteredList(LawSearchModel model)
{
if (string.IsNullOrEmpty(model.LawName))
{
return new();
}
using var context = new LawFirmDataBase();
return context.Laws.Include(x => x.Components)
.ThenInclude(x => x.Component)
.Where(x => x.LawName.Contains(model.LawName))
.ToList()
.Select(x => x.GetViewModel)
.ToList();
}
public LawViewModel? GetElement(LawSearchModel model)
{
if (string.IsNullOrEmpty(model.LawName) &&
!model.Id.HasValue)
{
return null;
}
using var context = new LawFirmDataBase();
return context.Laws
.Include(x => x.Components)
.ThenInclude(x => x.Component)
.FirstOrDefault(x => (!string.IsNullOrEmpty(model.LawName) &&
x.LawName == model.LawName) ||
(model.Id.HasValue && x.Id ==
model.Id))
?.GetViewModel;
}
public LawViewModel? Insert(LawBindingModel model)
{
using var context = new LawFirmDataBase();
var newLaw = Law.Create(context, model);
if (newLaw == null)
{
return null;
}
context.Laws.Add(newLaw);
context.SaveChanges();
return newLaw.GetViewModel;
}
public LawViewModel? Update(LawBindingModel model)
{
using var context = new LawFirmDataBase();
using var transaction = context.Database.BeginTransaction();
try
{
var Law = context.Laws.FirstOrDefault(rec =>
rec.Id == model.Id);
if (Law == null)
{
return null;
}
Law.Update(model);
context.SaveChanges();
Law.UpdateComponents(context, model);
transaction.Commit();
return Law.GetViewModel;
}
catch
{
transaction.Rollback();
throw;
}
}
public LawViewModel? Delete(LawBindingModel model)
{
using var context = new LawFirmDataBase();
var element = context.Laws
.Include(x => x.Components)
.FirstOrDefault(rec => rec.Id == model.Id);
if (element != null)
{
context.Laws.Remove(element);
context.SaveChanges();
return element.GetViewModel;
}
return null;
}
}
}