Show one of the possible ways to use UoW with repository pattern

Author: nemanjarogic

src/AdditionalPatterns/Repository/OrderManagement/Controllers/OrderController.cs

@@ -7,10 +7,12 @@ namespace OrderManagement.Controllers
     public class OrderController
     {
         private readonly IOrderRepository _orderRepository;
+        private readonly IUnitOfWork _unitOfWork;
 
-        public OrderController(IOrderRepository orderRepository)
+        public OrderController(IOrderRepository orderRepository, IUnitOfWork unitOfWork)
         {
             _orderRepository = orderRepository;
+            _unitOfWork = unitOfWork;
         }
 
         public Order Create(int id, string description, string deliveryAddress, decimal price)
@@ -25,7 +27,7 @@ public Order Create(int id, string description, string deliveryAddress, decimal
             };
 
             newOrder = _orderRepository.Add(newOrder);
-            _orderRepository.SaveChanges();
+            _unitOfWork.Commit();
 
             return newOrder;
         }

src/AdditionalPatterns/Repository/OrderManagement/Executor.cs

@@ -9,72 +9,80 @@ namespace OrderManagement
 {
     public class Executor : PatternExecutor
     {
+        private OrderManagementContext _context;
+        private CustomerRepository _customerRepository;
+        private OrderRepository _orderRepository;
+        private UnitOfWork _unitOfWork;
+
+        private CustomerController _customerController;
+        private OrderController _orderController;
+
         public override string Name => "Repository - Data Access Pattern";
 
         public override void Execute()
         {
+            ConfigureDependencies();
             InitializeDatabase();
 
-            using var context = new OrderManagementContext();
-            var customerRepository = new CustomerRepository(context);
-            var orderRepository = new OrderRepository(context);
-
-            var customerController = new CustomerController(customerRepository);
-            var orderController = new OrderController(orderRepository);
-
             ShowAllCustomers();
             ShowAllOrders();
             CreateNewOrder();
             ShowAllOrders();
+        }
 
-            void ShowAllCustomers()
-            {
-                Console.WriteLine("Showing all customers...");
-                foreach (var customer in customerController.GetAll())
-                {
-                    Console.WriteLine($"{customer.FirstName} {customer.LastName}");
-                }
-            }
-
-            void ShowAllOrders()
-            {
-                Console.WriteLine("\nShowing all orders...");
-                foreach (var order in orderController.GetAll())
-                {
-                    Console.WriteLine($"{order.Description} with price of {order.Price:C}");
-                }
-            }
+        private void ConfigureDependencies()
+        {
+            _context = new OrderManagementContext();
+            _customerRepository = new CustomerRepository(_context);
+            _orderRepository = new OrderRepository(_context);
+            _unitOfWork = new UnitOfWork(_context);
 
-            void CreateNewOrder()
-            {
-                Console.WriteLine("\nCreating new order...");
-                orderController.Create(3, "PlayStation 5", "Address 3", 600);
-            }
+            _customerController = new CustomerController(_customerRepository);
+            _orderController = new OrderController(_orderRepository, _unitOfWork);
         }
 
         private void InitializeDatabase()
         {
-            using (var context = new OrderManagementContext())
-            {
-                context.Database.EnsureDeleted();
-                context.Database.EnsureCreated();
+            _context.Database.EnsureDeleted();
+            _context.Database.EnsureCreated();
+
+            var emma = new Customer { Id = 1, FirstName = "Emma", LastName = "Johnson" };
+            var marc = new Customer { Id = 2, FirstName = "Marc", LastName = "Milston" };
 
-                var emma = new Customer { Id = 1, FirstName = "Emma", LastName = "Johnson" };
-                var marc = new Customer { Id = 2, FirstName = "Marc", LastName = "Milston" };
+            _customerRepository.Add(emma);
+            _customerRepository.Add(marc);
 
-                var camera = new Order { Id = 1, Description = "Camera", DeliveryAddress = "Address 1", Price = 550 };
-                var wallet = new Order { Id = 2, Description = "Wallet", DeliveryAddress = "Address 2", Price = 10 };
+            var camera = new Order { Id = 1, Description = "Camera", DeliveryAddress = "Address 1", Price = 550 };
+            var wallet = new Order { Id = 2, Description = "Wallet", DeliveryAddress = "Address 2", Price = 10 };
 
-                var customerRepository = new CustomerRepository(context);
-                customerRepository.Add(emma);
-                customerRepository.Add(marc);
-                customerRepository.SaveChanges();
+            _orderRepository.Add(camera);
+            _orderRepository.Add(wallet);
+            
+            _unitOfWork.Commit();
+        }
 
-                var orderRepository = new OrderRepository(context);
-                orderRepository.Add(camera);
-                orderRepository.Add(wallet);
-                orderRepository.SaveChanges();
+        private void ShowAllCustomers()
+        {
+            Console.WriteLine("Showing all customers...");
+            foreach (var customer in _customerController.GetAll())
+            {
+                Console.WriteLine($"{customer.FirstName} {customer.LastName}");
             }
         }
+
+        private void ShowAllOrders()
+        {
+            Console.WriteLine("\nShowing all orders...");
+            foreach (var order in _orderController.GetAll())
+            {
+                Console.WriteLine($"{order.Description} with price of {order.Price:C}");
+            }
+        }
+
+        private void CreateNewOrder()
+        {
+            Console.WriteLine("\nCreating new order...");
+            _orderController.Create(3, "PlayStation 5", "Address 3", 600);
+        }
     }
 }

src/AdditionalPatterns/Repository/OrderManagement/Infrastructure/Repositories/Contracts/IRepository.cs

@@ -17,7 +17,5 @@ public interface IRepository<T>
         T GetById(int id);
 
         void Delete(T entity);
-
-        void SaveChanges();
     }
 }

src/AdditionalPatterns/Repository/OrderManagement/Infrastructure/Repositories/Contracts/IUnitOfWork.cs

@@ -0,0 +1,7 @@
+namespace OrderManagement.Infrastructure.Repositories.Contracts
+{
+    public interface IUnitOfWork
+    {
+        void Commit();
+    }
+}

src/AdditionalPatterns/Repository/OrderManagement/Infrastructure/Repositories/Repository.cs

@@ -55,15 +55,5 @@ public virtual void Delete(T entity)
         {
             _context.Remove(entity);
         }
-
-        /// <summary>
-        /// Save all changes to the database.
-        /// Additionaly, we can implement UnitOfWork design pattern
-        /// to enhance this example and coordinate database changes from one place.
-        /// </summary>
-        public void SaveChanges()
-        {
-            _context.SaveChanges();
-        }
     }
 }

src/AdditionalPatterns/Repository/OrderManagement/Infrastructure/Repositories/UnitOfWork.cs

@@ -0,0 +1,19 @@
+using OrderManagement.Infrastructure.Repositories.Contracts;
+
+namespace OrderManagement.Infrastructure.Repositories
+{
+    public class UnitOfWork : IUnitOfWork
+    {
+        private readonly OrderManagementContext _context;
+
+        public UnitOfWork(OrderManagementContext context)
+        {
+            _context = context;
+        }
+
+        public void Commit()
+        {
+            _context.SaveChanges();
+        }
+    }
+}