Merge pull request #838 from jbogard/publish-strategies

Adding notification publisher strategies

samples/MediatR.Examples.PublishStrategies/CustomMediator.cs

@@ -7,11 +7,11 @@ namespace MediatR.Examples.PublishStrategies;
 
 public class CustomMediator : Mediator
 {
-    private readonly Func<IEnumerable<Func<INotification, CancellationToken, Task>>, INotification, CancellationToken, Task> _publish;
+    private readonly Func<IEnumerable<NotificationHandlerExecutor>, INotification, CancellationToken, Task> _publish;
 
-    public CustomMediator(IServiceProvider serviceFactory, Func<IEnumerable<Func<INotification, CancellationToken, Task>>, INotification, CancellationToken, Task> publish) : base(serviceFactory) 
+    public CustomMediator(IServiceProvider serviceFactory, Func<IEnumerable<NotificationHandlerExecutor>, INotification, CancellationToken, Task> publish) : base(serviceFactory) 
         => _publish = publish;
 
-    protected override Task PublishCore(IEnumerable<Func<INotification, CancellationToken, Task>> allHandlers, INotification notification, CancellationToken cancellationToken) 
-        => _publish(allHandlers, notification, cancellationToken);
+    protected override Task PublishCore(IEnumerable<NotificationHandlerExecutor> handlerExecutors, INotification notification, CancellationToken cancellationToken) 
+        => _publish(handlerExecutors, notification, cancellationToken);
 }

samples/MediatR.Examples.PublishStrategies/Publisher.cs

@@ -50,41 +50,41 @@ public Task Publish<TNotification>(TNotification notification, PublishStrategy s
         return mediator.Publish(notification, cancellationToken);
     }
 
-    private Task ParallelWhenAll(IEnumerable<Func<INotification, CancellationToken, Task>> handlers, INotification notification, CancellationToken cancellationToken)
+    private Task ParallelWhenAll(IEnumerable<NotificationHandlerExecutor> handlers, INotification notification, CancellationToken cancellationToken)
     {
         var tasks = new List<Task>();
 
         foreach (var handler in handlers)
         {
-            tasks.Add(Task.Run(() => handler(notification, cancellationToken)));
+            tasks.Add(Task.Run(() => handler.HandlerCallback(notification, cancellationToken)));
         }
 
         return Task.WhenAll(tasks);
     }
 
-    private Task ParallelWhenAny(IEnumerable<Func<INotification, CancellationToken, Task>> handlers, INotification notification, CancellationToken cancellationToken)
+    private Task ParallelWhenAny(IEnumerable<NotificationHandlerExecutor> handlers, INotification notification, CancellationToken cancellationToken)
     {
         var tasks = new List<Task>();
 
         foreach (var handler in handlers)
         {
-            tasks.Add(Task.Run(() => handler(notification, cancellationToken)));
+            tasks.Add(Task.Run(() => handler.HandlerCallback(notification, cancellationToken)));
         }
 
         return Task.WhenAny(tasks);
     }
 
-    private Task ParallelNoWait(IEnumerable<Func<INotification, CancellationToken, Task>> handlers, INotification notification, CancellationToken cancellationToken)
+    private Task ParallelNoWait(IEnumerable<NotificationHandlerExecutor> handlers, INotification notification, CancellationToken cancellationToken)
     {
         foreach (var handler in handlers)
         {
-            Task.Run(() => handler(notification, cancellationToken));
+            Task.Run(() => handler.HandlerCallback(notification, cancellationToken));
         }
 
         return Task.CompletedTask;
     }
 
-    private async Task AsyncContinueOnException(IEnumerable<Func<INotification, CancellationToken, Task>> handlers, INotification notification, CancellationToken cancellationToken)
+    private async Task AsyncContinueOnException(IEnumerable<NotificationHandlerExecutor> handlers, INotification notification, CancellationToken cancellationToken)
     {
         var tasks = new List<Task>();
         var exceptions = new List<Exception>();
@@ -93,7 +93,7 @@ private async Task AsyncContinueOnException(IEnumerable<Func<INotification, Canc
         {
             try
             {
-                tasks.Add(handler(notification, cancellationToken));
+                tasks.Add(handler.HandlerCallback(notification, cancellationToken));
             }
             catch (Exception ex) when (!(ex is OutOfMemoryException || ex is StackOverflowException))
             {
@@ -120,23 +120,23 @@ private async Task AsyncContinueOnException(IEnumerable<Func<INotification, Canc
         }
     }
 
-    private async Task SyncStopOnException(IEnumerable<Func<INotification, CancellationToken, Task>> handlers, INotification notification, CancellationToken cancellationToken)
+    private async Task SyncStopOnException(IEnumerable<NotificationHandlerExecutor> handlers, INotification notification, CancellationToken cancellationToken)
     {
         foreach (var handler in handlers)
         {
-            await handler(notification, cancellationToken).ConfigureAwait(false);
+            await handler.HandlerCallback(notification, cancellationToken).ConfigureAwait(false);
         }
     }
 
-    private async Task SyncContinueOnException(IEnumerable<Func<INotification, CancellationToken, Task>> handlers, INotification notification, CancellationToken cancellationToken)
+    private async Task SyncContinueOnException(IEnumerable<NotificationHandlerExecutor> handlers, INotification notification, CancellationToken cancellationToken)
     {
         var exceptions = new List<Exception>();
 
         foreach (var handler in handlers)
         {
             try
             {
-                await handler(notification, cancellationToken).ConfigureAwait(false);
+                await handler.HandlerCallback(notification, cancellationToken).ConfigureAwait(false);
             }
             catch (AggregateException ex)
             {

src/MediatR/INotificationPublisher.cs

@@ -0,0 +1,11 @@
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using System.Threading;
+
+namespace MediatR;
+
+public interface INotificationPublisher
+{
+    Task Publish(IEnumerable<NotificationHandlerExecutor> handlerExecutors, INotification notification,
+        CancellationToken cancellationToken);
+}

src/MediatR/MediatR.csproj

@@ -27,11 +27,15 @@
   </ItemGroup>
 
   <ItemGroup>
+    <PackageReference Include="IsExternalInit" Version="1.0.3">
+      <PrivateAssets>all</PrivateAssets>
+      <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
+    </PackageReference>
     <PackageReference Include="MediatR.Contracts" Version="[2.0.0, 3.0.0)" />
     <PackageReference Include="Microsoft.Bcl.AsyncInterfaces" Version="7.0.0" />
     <PackageReference Include="Microsoft.Extensions.DependencyInjection.Abstractions" Version="7.0.0" />
     <PackageReference Include="Microsoft.SourceLink.GitHub" Version="1.1.1" PrivateAssets="All" />
-    <PackageReference Include="MinVer" Version="4.2.0" PrivateAssets="All" />
+    <PackageReference Include="MinVer" Version="4.3.0" PrivateAssets="All" />
   </ItemGroup>
 
 </Project>

src/MediatR/Mediator.cs

@@ -1,3 +1,5 @@
+using MediatR.NotificationPublishers;
+
 namespace MediatR;
 
 using System;
@@ -14,6 +16,7 @@ namespace MediatR;
 public class Mediator : IMediator
 {
     private readonly IServiceProvider _serviceProvider;
+    private readonly INotificationPublisher _publisher;
     private static readonly ConcurrentDictionary<Type, RequestHandlerBase> _requestHandlers = new();
     private static readonly ConcurrentDictionary<Type, NotificationHandlerWrapper> _notificationHandlers = new();
     private static readonly ConcurrentDictionary<Type, StreamRequestHandlerBase> _streamRequestHandlers = new();
@@ -23,7 +26,18 @@ public class Mediator : IMediator
     /// </summary>
     /// <param name="serviceProvider">Service provider. Can be a scoped or root provider</param>
     public Mediator(IServiceProvider serviceProvider) 
-        => _serviceProvider = serviceProvider;
+        : this(serviceProvider, new ForeachAwaitPublisher()) { }
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="Mediator"/> class.
+    /// </summary>
+    /// <param name="serviceProvider">Service provider. Can be a scoped or root provider</param>
+    /// <param name="publisher">Notification publisher. Defaults to <see cref="ForeachAwaitPublisher"/>.</param>
+    public Mediator(IServiceProvider serviceProvider, INotificationPublisher publisher)
+    {
+        _serviceProvider = serviceProvider;
+        _publisher = publisher;
+    }
 
     public Task<TResponse> Send<TResponse>(IRequest<TResponse> request, CancellationToken cancellationToken = default)
     {
@@ -124,19 +138,14 @@ notification switch
         };
 
     /// <summary>
-    /// Override in a derived class to control how the tasks are awaited. By default the implementation is a foreach and await of each handler
+    /// Override in a derived class to control how the tasks are awaited. By default the implementation calls the <see cref="INotificationPublisher"/>.
     /// </summary>
-    /// <param name="allHandlers">Enumerable of tasks representing invoking each notification handler</param>
+    /// <param name="handlerExecutors">Enumerable of tasks representing invoking each notification handler</param>
     /// <param name="notification">The notification being published</param>
     /// <param name="cancellationToken">The cancellation token</param>
     /// <returns>A task representing invoking all handlers</returns>
-    protected virtual async Task PublishCore(IEnumerable<Func<INotification, CancellationToken, Task>> allHandlers, INotification notification, CancellationToken cancellationToken)
-    {
-        foreach (var handler in allHandlers)
-        {
-            await handler(notification, cancellationToken).ConfigureAwait(false);
-        }
-    }
+    protected virtual Task PublishCore(IEnumerable<NotificationHandlerExecutor> handlerExecutors, INotification notification, CancellationToken cancellationToken) 
+        => _publisher.Publish(handlerExecutors, notification, cancellationToken);
 
     private Task PublishNotification(INotification notification, CancellationToken cancellationToken = default)
     {

src/MediatR/MicrosoftExtensionsDI/MediatrServiceConfiguration.cs

@@ -2,34 +2,83 @@
 using System.Collections.Generic;
 using System.Reflection;
 using MediatR;
+using MediatR.NotificationPublishers;
 
 namespace Microsoft.Extensions.DependencyInjection;
 
 public class MediatRServiceConfiguration
 {
+    /// <summary>
+    /// Optional filter for types to register. Default value is a function returning true.
+    /// </summary>
     public Func<Type, bool> TypeEvaluator { get; set; } = t => true;
+
+    /// <summary>
+    /// Mediator implementation type to register. Default is <see cref="Mediator"/>
+    /// </summary>
     public Type MediatorImplementationType { get; set; } = typeof(Mediator);
+
+    /// <summary>
+    /// Strategy for publishing notifications. Defaults to <see cref="ForeachAwaitPublisher"/>
+    /// </summary>
+    public INotificationPublisher NotificationPublisher { get; set; } = new ForeachAwaitPublisher();
+
+    /// <summary>
+    /// Type of notification publisher strategy to register. If set, overrides <see cref="NotificationPublisher"/>
+    /// </summary>
+    public Type? NotificationPublisherType { get; set; }
+
+    /// <summary>
+    /// Service lifetime to register services under. Default value is <see cref="ServiceLifetime.Transient"/>
+    /// </summary>
     public ServiceLifetime Lifetime { get; set; } = ServiceLifetime.Transient;
+
+    /// <summary>
+    /// Request exception action processor strategy. Default value is <see cref="DependencyInjection.RequestExceptionActionProcessorStrategy.ApplyForUnhandledExceptions"/>
+    /// </summary>
     public RequestExceptionActionProcessorStrategy RequestExceptionActionProcessorStrategy { get; set; }
         = RequestExceptionActionProcessorStrategy.ApplyForUnhandledExceptions;
 
     internal List<Assembly> AssembliesToRegister { get; } = new();
 
+    /// <summary>
+    /// List of behaviors to register in specific order
+    /// </summary>
     public List<ServiceDescriptor> BehaviorsToRegister { get; } = new();
 
+    /// <summary>
+    /// Register various handlers from assembly containing given type
+    /// </summary>
+    /// <typeparam name="T">Type from assembly to scan</typeparam>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration RegisterServicesFromAssemblyContaining<T>()
         => RegisterServicesFromAssemblyContaining(typeof(T));
 
+    /// <summary>
+    /// Register various handlers from assembly containing given type
+    /// </summary>
+    /// <param name="type">Type from assembly to scan</param>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration RegisterServicesFromAssemblyContaining(Type type)
         => RegisterServicesFromAssembly(type.Assembly);
 
+    /// <summary>
+    /// Register various handlers from assembly
+    /// </summary>
+    /// <param name="assembly">Assembly to scan</param>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration RegisterServicesFromAssembly(Assembly assembly)
     {
         AssembliesToRegister.Add(assembly);
 
         return this;
     }
 
+    /// <summary>
+    /// Register various handlers from assemblies
+    /// </summary>
+    /// <param name="assemblies">Assemblies to scan</param>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration RegisterServicesFromAssemblies(
         params Assembly[] assemblies)
     {
@@ -38,10 +87,24 @@ public MediatRServiceConfiguration RegisterServicesFromAssembly(Assembly assembl
         return this;
     }
 
+    /// <summary>
+    /// Register a closed behavior type
+    /// </summary>
+    /// <typeparam name="TServiceType">Closed behavior interface type</typeparam>
+    /// <typeparam name="TImplementationType">Closed behavior implementation type</typeparam>
+    /// <param name="serviceLifetime">Optional service lifetime, defaults to <see cref="ServiceLifetime.Transient"/>.</param>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration AddBehavior<TServiceType, TImplementationType>(
         ServiceLifetime serviceLifetime = ServiceLifetime.Transient) =>
         AddBehavior(typeof(TServiceType), typeof(TImplementationType), serviceLifetime);
 
+    /// <summary>
+    /// Register a closed behavior type
+    /// </summary>
+    /// <param name="serviceType">Closed behavior interface type</param>
+    /// <param name="implementationType">Closed behavior implementation type</param>
+    /// <param name="serviceLifetime">Optional service lifetime, defaults to <see cref="ServiceLifetime.Transient"/>.</param>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration AddBehavior(
         Type serviceType,
         Type implementationType,
@@ -52,6 +115,12 @@ public MediatRServiceConfiguration RegisterServicesFromAssembly(Assembly assembl
         return this;
     }
 
+    /// <summary>
+    /// Registers an open behavior type against the <see cref="IPipelineBehavior{TRequest,TResponse}"/> open generic interface type
+    /// </summary>
+    /// <param name="openBehaviorType">An open generic behavior type</param>
+    /// <param name="serviceLifetime">Optional service lifetime, defaults to <see cref="ServiceLifetime.Transient"/>.</param>
+    /// <returns>This</returns>
     public MediatRServiceConfiguration AddOpenBehavior(Type openBehaviorType, ServiceLifetime serviceLifetime = ServiceLifetime.Transient)
     {
         var serviceType = typeof(IPipelineBehavior<,>);

src/MediatR/NotificationHandlerExecutor.cs

@@ -0,0 +1,7 @@
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediatR;
+
+public record NotificationHandlerExecutor(object HandlerInstance, Func<INotification, CancellationToken, Task> HandlerCallback);

src/MediatR/NotificationPublishers/ForeachAwaitPublisher.cs

@@ -0,0 +1,24 @@
+using System.Collections.Generic;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediatR.NotificationPublishers;
+
+/// <summary>
+/// Awaits each notification handler in a single foreach loop:
+/// <code>
+/// foreach (var handler in handlers) {
+///     await handler(notification, cancellationToken);
+/// }
+/// </code>
+/// </summary>
+public class ForeachAwaitPublisher : INotificationPublisher
+{
+    public async Task Publish(IEnumerable<NotificationHandlerExecutor> handlerExecutors, INotification notification, CancellationToken cancellationToken)
+    {
+        foreach (var handler in handlerExecutors)
+        {
+            await handler.HandlerCallback(notification, cancellationToken).ConfigureAwait(false);
+        }
+    }
+}

src/MediatR/NotificationPublishers/TaskWhenAllPublisher.cs

@@ -0,0 +1,28 @@
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediatR.NotificationPublishers;
+
+/// <summary>
+/// Uses Task.WhenAll with the list of Handler tasks:
+/// <code>
+/// var tasks = handlers
+///                .Select(handler => handler.Handle(notification, cancellationToken))
+///                .ToList();
+/// 
+/// return Task.WhenAll(tasks);
+/// </code>
+/// </summary>
+public class TaskWhenAllPublisher : INotificationPublisher
+{
+    public Task Publish(IEnumerable<NotificationHandlerExecutor> handlerExecutors, INotification notification, CancellationToken cancellationToken)
+    {
+        var tasks = handlerExecutors
+            .Select(handler => handler.HandlerCallback(notification, cancellationToken))
+            .ToArray();
+
+        return Task.WhenAll(tasks);
+    }
+}

src/MediatR/Registration/ServiceRegistrar.cs

@@ -218,6 +218,11 @@ public static void AddRequiredServices(IServiceCollection services, MediatRServi
         services.TryAdd(new ServiceDescriptor(typeof(ISender), sp => sp.GetRequiredService<IMediator>(), serviceConfiguration.Lifetime));
         services.TryAdd(new ServiceDescriptor(typeof(IPublisher), sp => sp.GetRequiredService<IMediator>(), serviceConfiguration.Lifetime));
 
+        services.TryAdd(serviceConfiguration.NotificationPublisherType != null
+            ? new ServiceDescriptor(typeof(INotificationPublisher), serviceConfiguration.NotificationPublisherType,
+                serviceConfiguration.Lifetime)
+            : new ServiceDescriptor(typeof(INotificationPublisher), serviceConfiguration.NotificationPublisher));
+
         foreach (var serviceDescriptor in serviceConfiguration.BehaviorsToRegister)
         {
             services.Add(serviceDescriptor);