big day behind me

HanyadikHetVan/Services/PauseService.cs

@@ -1,5 +1,7 @@
-using HanyadikHetVan.Data;
+using AutoMapper;
+using HanyadikHetVan.Data;
 using HanyadikHetVan.Data.Entities;
+using HanyadikHetVan.DTO;
 using Microsoft.EntityFrameworkCore;
 using System;
 using System.Collections.Generic;
@@ -11,31 +13,41 @@ namespace HanyadikHetVan.Services
     public class PauseService
     {
         private readonly ApplicationDbContext _dbContext;
+        private readonly IMapper _mapper;
 
-        public PauseService(ApplicationDbContext dbContext)
+        public PauseService(ApplicationDbContext dbContext, IMapper mapper)
         {
             _dbContext = dbContext ?? throw new ArgumentNullException(nameof(dbContext));
+            _mapper = mapper ?? throw new ArgumentNullException(nameof(mapper));
         }
 
-        public IEnumerable<Pause> GetAllPauses()
+        public async Task<List<PauseDTO>> GetAllPauses()
         {
-            return _dbContext.Pauses.Include(x => x.WeeklyTimeSpan).ToList();
+            var pauses = await _dbContext.Pauses.Include(x => x.WeeklyTimeSpan).ToListAsync();
+            return _mapper.Map<List<Pause>, List<PauseDTO>>(pauses);
         }
 
-        public async Task<Pause> AddPause(Pause pause)
+        public async Task<PauseDTO> AddPause(PauseDTO pause)
         {
-            var obj = await _dbContext.Pauses.AddAsync(pause);
-            _dbContext.SaveChanges();
-            return obj.Entity;
+            var pauseEntity = _mapper.Map<PauseDTO, Pause>(pause);
+            var obj = _dbContext.Pauses.Add(pauseEntity);
+            await _dbContext.SaveChangesAsync();
+            return _mapper.Map<Pause, PauseDTO>(obj.Entity);
         }
 
-        public bool DeletePause(int pauseId)
+        public async Task<PauseDTO> GetPause(int pauseId)
+        {
+            var pause = await _dbContext.Pauses.Where(x => x.Id == pauseId).FirstOrDefaultAsync();
+            return _mapper.Map<Pause, PauseDTO>(pause);
+        }
+
+        public async Task<bool> DeletePause(int pauseId)
         {
             try
             {
-                var item = _dbContext.Pauses.Where(x => x.Id == pauseId).FirstOrDefault();
+                var item = await _dbContext.Pauses.Where(x => x.Id == pauseId).FirstOrDefaultAsync();
                 _dbContext.Pauses.Remove(item);
-                _dbContext.SaveChanges();
+                await _dbContext.SaveChangesAsync();
                 return true;
             }
             catch (Exception)
@@ -44,22 +56,14 @@ namespace HanyadikHetVan.Services
             }
         }
 
-        public bool UpdatePause(Pause pause)
+        public async Task<PauseDTO> UpdatePause(PauseDTO pause)
         {
-            try
-            {
-                var DataList = _dbContext.Pauses.ToList();
-                foreach (var item in DataList)
-                {
-                    _dbContext.Pauses.Update(item);
-                    _dbContext.SaveChanges();
-                }
-                return true;
-            }
-            catch (Exception)
-            {
-                return true;
-            }
+            var pauseEntity = _mapper.Map<PauseDTO, Pause>(pause);
+            pauseEntity.Id = pause.Id;
+            _dbContext.Pauses.Update(pauseEntity);
+            await _dbContext.SaveChangesAsync();
+            var newPause = await _dbContext.Pauses.Where(x => x.Id == pause.Id).FirstOrDefaultAsync();
+            return _mapper.Map<Pause, PauseDTO>(newPause);
         }
     }
 }