Request will now publish an event IF result implements IEvent

shinyorg · Sep 28, 2024 · 1991434 · 1991434
1 parent 08c547e
commit 1991434
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/Shiny.Mediator/Impl/Mediator.cs b/src/Shiny.Mediator/Impl/Mediator.cs
@@ -8,8 +8,17 @@ public class Mediator(
     IEventPublisher eventPublisher
 ) : IMediator
 {
-    public Task<TResult> Request<TResult>(IRequest<TResult> request, CancellationToken cancellationToken = default)
-        => requestSender.Request(request, cancellationToken);
+    public async Task<TResult> Request<TResult>(IRequest<TResult> request, CancellationToken cancellationToken = default)
+    {
+        var result = await requestSender
+            .Request(request, cancellationToken)
+            .ConfigureAwait(false);
+
+        if (result is IEvent @event)
+            await this.Publish(@event, cancellationToken).ConfigureAwait(false);
+
+        return result;
+    }
 
     public Task Send(IRequest request, CancellationToken cancellationToken = default)
         => requestSender.Send(request, cancellationToken);