Merge pull request #1033 from LeeDumond/enhance/cancellation-token-support

provide support for cancellation tokens in command handlers

Author: mvelosop

src/Services/Ordering/Ordering.API/Application/Commands/CancelOrderCommandHandler.cs

@@ -33,7 +33,7 @@ public async Task<bool> Handle(CancelOrderCommand command, CancellationToken can
             }
 
             orderToUpdate.SetCancelledStatus();
-            return await _orderRepository.UnitOfWork.SaveEntitiesAsync();
+            return await _orderRepository.UnitOfWork.SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/Commands/CreateOrderCommandHandler.cs

@@ -58,7 +58,7 @@ public async Task<bool> Handle(CreateOrderCommand message, CancellationToken can
             _orderRepository.Add(order);
 
             return await _orderRepository.UnitOfWork
-                .SaveEntitiesAsync();
+                .SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/Commands/IdentifiedCommandHandler.cs

@@ -96,7 +96,7 @@ public async Task<R> Handle(IdentifiedCommand<T, R> message, CancellationToken c
                         command);
 
                     // Send the embeded business command to mediator so it runs its related CommandHandler 
-                    var result = await _mediator.Send(command);
+                    var result = await _mediator.Send(command, cancellationToken);
 
                     _logger.LogInformation(
                         "----- Command result: {@Result} - {CommandName} - {IdProperty}: {CommandId} ({@Command})",

src/Services/Ordering/Ordering.API/Application/Commands/SetAwaitingValidationOrderStatusCommandHandler.cs

@@ -33,7 +33,7 @@ public async Task<bool> Handle(SetAwaitingValidationOrderStatusCommand command,
             }
 
             orderToUpdate.SetAwaitingValidationStatus();
-            return await _orderRepository.UnitOfWork.SaveEntitiesAsync();
+            return await _orderRepository.UnitOfWork.SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/Commands/SetPaidOrderStatusCommandHandler.cs

@@ -27,7 +27,7 @@ public SetPaidOrderStatusCommandHandler(IOrderRepository orderRepository)
         public async Task<bool> Handle(SetPaidOrderStatusCommand command, CancellationToken cancellationToken)
         {
             // Simulate a work time for validating the payment
-            await Task.Delay(10000);
+            await Task.Delay(10000, cancellationToken);
 
             var orderToUpdate = await _orderRepository.GetAsync(command.OrderNumber);
             if(orderToUpdate == null)
@@ -36,7 +36,7 @@ public async Task<bool> Handle(SetPaidOrderStatusCommand command, CancellationTo
             }
 
             orderToUpdate.SetPaidStatus();
-            return await _orderRepository.UnitOfWork.SaveEntitiesAsync();
+            return await _orderRepository.UnitOfWork.SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/Commands/SetStockConfirmedOrderStatusCommandHandler.cs

@@ -27,7 +27,7 @@ public SetStockConfirmedOrderStatusCommandHandler(IOrderRepository orderReposito
         public async Task<bool> Handle(SetStockConfirmedOrderStatusCommand command, CancellationToken cancellationToken)
         {
             // Simulate a work time for confirming the stock
-            await Task.Delay(10000);
+            await Task.Delay(10000, cancellationToken);
 
             var orderToUpdate = await _orderRepository.GetAsync(command.OrderNumber);
             if(orderToUpdate == null)
@@ -36,7 +36,7 @@ public async Task<bool> Handle(SetStockConfirmedOrderStatusCommand command, Canc
             }
 
             orderToUpdate.SetStockConfirmedStatus();
-            return await _orderRepository.UnitOfWork.SaveEntitiesAsync();
+            return await _orderRepository.UnitOfWork.SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/Commands/SetStockRejectedOrderStatusCommandHandler.cs

@@ -27,7 +27,7 @@ public SetStockRejectedOrderStatusCommandHandler(IOrderRepository orderRepositor
         public async Task<bool> Handle(SetStockRejectedOrderStatusCommand command, CancellationToken cancellationToken)
         {
             // Simulate a work time for rejecting the stock
-            await Task.Delay(10000);
+            await Task.Delay(10000, cancellationToken);
 
             var orderToUpdate = await _orderRepository.GetAsync(command.OrderNumber);
             if(orderToUpdate == null)
@@ -37,7 +37,7 @@ public async Task<bool> Handle(SetStockRejectedOrderStatusCommand command, Cance
 
             orderToUpdate.SetCancelledStatusWhenStockIsRejected(command.OrderStockItems);
 
-            return await _orderRepository.UnitOfWork.SaveEntitiesAsync();
+            return await _orderRepository.UnitOfWork.SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/Commands/ShipOrderCommandHandler.cs

@@ -33,7 +33,7 @@ public async Task<bool> Handle(ShipOrderCommand command, CancellationToken cance
             }
 
             orderToUpdate.SetShippedStatus();
-            return await _orderRepository.UnitOfWork.SaveEntitiesAsync();
+            return await _orderRepository.UnitOfWork.SaveEntitiesAsync(cancellationToken);
         }
     }
 

src/Services/Ordering/Ordering.API/Application/DomainEventHandlers/OrderStartedEvent/ValidateOrAddBuyerAggregateWhenOrderStartedDomainEventHandler.cs

@@ -56,7 +56,7 @@ public async Task Handle(OrderStartedDomainEvent orderStartedEvent, Cancellation
                 _buyerRepository.Add(buyer);
 
             await _buyerRepository.UnitOfWork
-                .SaveEntitiesAsync();
+                .SaveEntitiesAsync(cancellationToken);
 
             var orderStatusChangedTosubmittedIntegrationEvent = new OrderStatusChangedToSubmittedIntegrationEvent(orderStartedEvent.Order.Id, orderStartedEvent.Order.OrderStatus.Name, buyer.Name);
             await _orderingIntegrationEventService.AddAndSaveEventAsync(orderStatusChangedTosubmittedIntegrationEvent);

src/Services/Ordering/Ordering.Infrastructure/OrderingContext.cs

@@ -64,7 +64,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
 
             // After executing this line all the changes (from the Command Handler and Domain Event Handlers) 
             // performed through the DbContext will be committed
-            var result = await base.SaveChangesAsync();
+            var result = await base.SaveChangesAsync(cancellationToken);
 
             return true;
         }