using MongoDB.Driver; using MongoDB.Bson; using RouteGuideContracts.BindingModels; using RouteGuideContracts.SearchModels; using RouteGuideContracts.StoragesContracts; using RouteGuideContracts.ViewModels; using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; using RouteGuideMongoDBImplement.Models; namespace RouteGuideMongoDBImplement.Implements { /// /// Хранилище для сущности "Водитель" /// public class DriverStorage : IDriverStorage { /// /// Получение полного списка /// /// public List GetFullList() { using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Empty; return context.Drivers .Find(filter) .ToList() .Select(x => x.GetViewModel) .ToList(); } /// /// Получение фильтрованного списка /// /// /// public List GetFilteredList(DriverSearchModel model) { if (string.IsNullOrEmpty(model.Phone)) { return new(); } using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Eq(x => x.Phone, model.Phone); return context.Drivers .Find(filter) .ToList() .Select(x => x.GetViewModel) .ToList(); } /// /// Получить список из заданного количества элементов /// /// /// public List GetList(int count) { if (count <= 0) { return new(); } using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Empty; return context.Drivers .Find(filter) .Limit(count) .ToList() .Select(x => x.GetViewModel) .ToList(); } /// /// Получение элемента /// /// /// public DriverViewModel? GetElement(DriverSearchModel model) { if (string.IsNullOrEmpty(model.Id?.ToString()) && string.IsNullOrEmpty(model.Phone)) { return null; } using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Empty; if (!string.IsNullOrEmpty(model.Phone)) { filter &= filterBuilder.Eq(x => x.Phone, model.Phone); } if (!string.IsNullOrEmpty(model.Id?.ToString())) { filter &= filterBuilder.Eq(x => x.Id, model.Id); } return context.Drivers .Find(filter) .FirstOrDefault() ?.GetViewModel; } /// /// Добавление элемента /// /// /// public DriverViewModel? Insert(DriverBindingModel model) { var newDriver = Driver.Create(model); if (newDriver == null) { return null; } using var context = new RouteGuideDatabase(); context.Drivers.InsertOne(newDriver); return newDriver.GetViewModel; } /// /// Редактирование элемента /// /// /// public DriverViewModel? Update(DriverBindingModel model) { using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Eq(x => x.Id, model.Id); var driver = context.Drivers.Find(filter).FirstOrDefault(); if (driver == null) { return null; } driver.Update(model); context.Drivers.ReplaceOne(filter, driver); return driver.GetViewModel; } /// /// Удаление элемента /// /// /// public DriverViewModel? Delete(DriverBindingModel model) { using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Eq(x => x.Id, model.Id); var driver = context.Drivers.Find(filter).FirstOrDefault(); if (driver == null) { return null; } context.Drivers.FindOneAndDelete(filter); return driver.GetViewModel; } /// /// Удаление записи /// /// public DriverViewModel? Delete() { if (!GetFullList().Any()) { return null; } using var context = new RouteGuideDatabase(); var filterBuilder = Builders.Filter; var filter = filterBuilder.Empty; var element = context.Drivers.FindOneAndDelete(filter); return element.GetViewModel; } /// /// Удаление всех элементов /// /// public int Clear() { using var context = new RouteGuideDatabase(); int count = GetFullList().Count; if (count <= 0) { return 0; } var filterBuilder = Builders.Filter; var filter = filterBuilder.Empty; context.Drivers.DeleteMany(filter); return count; } } }