Correction du restore

sheaft-app · Oct 11, 2021 · 73c99d0 · 73c99d0
1 parent fe14857
commit 73c99d0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Sheaft.Mediatr/Consumer/Commands/RegisterConsumerCommand.cs b/Sheaft.Mediatr/Consumer/Commands/RegisterConsumerCommand.cs
@@ -72,7 +72,7 @@ public async Task<Result<Guid>> Handle(RegisterConsumerCommand request, Cancella
             {
                 if (consumer != null)
                 {
-                    consumer.Restore();
+                    _context.Restore(consumer);
                 }
                 else
                 {

diff --git a/Sheaft.Mediatr/Producer/Commands/RegisterProducerCommand.cs b/Sheaft.Mediatr/Producer/Commands/RegisterProducerCommand.cs
@@ -94,7 +94,7 @@ await _context.Producers.SingleOrDefaultAsync(
             {
                 if (producer != null)
                 {
-                    producer.Restore();
+                    _context.Restore(producer);
                 }
                 else
                 {

diff --git a/Sheaft.Mediatr/Store/Commands/RegisterStoreCommand.cs b/Sheaft.Mediatr/Store/Commands/RegisterStoreCommand.cs
@@ -95,7 +95,7 @@ public async Task<Result<Guid>> Handle(RegisterStoreCommand request, Cancellatio
 
                 if (store != null)
                 {
-                    store.Restore();
+                    _context.Restore(store);
                 }
                 else
                 {