improve SlimBus registration

Author: baoduy

src/DKNet.FW.sln.DotSettings.user

@@ -579,7 +579,7 @@
 	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=8275c8a4_002D4ae3_002D406a_002D9cd0_002D6dc520603cf1/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" Name="All tests from &amp;lt;EfCore&amp;gt;\&amp;lt;EfCore.Specifications.Tests&amp;gt;" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
   &lt;Project Location="/Users/steven/_CODE/DRUNK/DKNet/src/EfCore/EfCore.Specifications.Tests" Presentation="&amp;lt;EfCore&amp;gt;\&amp;lt;EfCore.Specifications.Tests&amp;gt;" /&gt;
 &lt;/SessionState&gt;</s:String>
-	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=bd8ac265_002D8b5d_002D4892_002D9e95_002Dccb01cb363e1/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" IsActive="True" Name="All tests from &amp;lt;EfCore&amp;gt;\&amp;lt;EfCore.Specifications.Tests&amp;gt;" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
+	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=bd8ac265_002D8b5d_002D4892_002D9e95_002Dccb01cb363e1/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" Name="All tests from &amp;lt;EfCore&amp;gt;\&amp;lt;EfCore.Specifications.Tests&amp;gt;" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
   &lt;Project Location="/Users/steven/_CODE/DRUNK/DKNet/src" Presentation="&amp;lt;EfCore&amp;gt;" /&gt;
 &lt;/SessionState&gt;</s:String>
 

src/SlimBus/DKNet.SlimBus.Extensions/Behaviors/EfAutoSavePostProcessor.cs

@@ -1,12 +1,30 @@
 using DKNet.EfCore.Extensions.Extensions;
 using FluentResults;
 using Microsoft.EntityFrameworkCore;
-using Microsoft.Extensions.DependencyInjection;
 using SlimMessageBus;
 using SlimMessageBus.Host.Interceptor;
 
 namespace DKNet.SlimBus.Extensions.Behaviors;
 
+internal static class EfAutoSavePostProcessorRegistration
+{
+    #region Properties
+
+    public static HashSet<Type> DbContextTypes { get; } = [];
+
+    #endregion
+
+    #region Methods
+
+    public static void RegisterDbContextType<TDbContext>()
+        where TDbContext : DbContext
+    {
+        DbContextTypes.Add(typeof(TDbContext));
+    }
+
+    #endregion
+}
+
 internal sealed class EfAutoSavePostProcessor<TRequest, TResponse>(IServiceProvider serviceProvider)
     : IRequestHandlerInterceptor<TRequest, TResponse>, IInterceptorWithOrder
 {
@@ -32,7 +50,9 @@ request is Fluents.Queries.IWitPageResponse<TResponse>
             || request is Fluents.EventsConsumers.IHandler<IRequest>)
             return response;
 
-        var dbContexts = serviceProvider.GetServices<DbContext>();
+        var dbContexts = EfAutoSavePostProcessorRegistration.DbContextTypes
+            .Select(serviceProvider.GetService).OfType<DbContext>().ToArray();
+
         foreach (var db in dbContexts.Where(db => db.ChangeTracker.HasChanges()))
         {
             await db.AddNewEntitiesFromNavigations(context.CancellationToken);

src/SlimBus/DKNet.SlimBus.Extensions/SlimBusEfCoreSetup.cs

@@ -7,7 +7,6 @@
 using DKNet.SlimBus.Extensions.Behaviors;
 using DKNet.SlimBus.Extensions.Handlers;
 using Microsoft.EntityFrameworkCore;
-using SlimMessageBus.Host;
 using SlimMessageBus.Host.Interceptor;
 
 // ReSharper disable once CheckNamespace
@@ -42,13 +41,18 @@ public IServiceCollection AddSlimBusEventPublisher<TDbContext>()
         ///     request handlers run. Registers the <see cref="EfAutoSavePostProcessor{TRequest,TResponse}" />
         ///     as an <see cref="IRequestHandlerInterceptor{TRequest,TResponse}" /> to be executed by SlimMessageBus.
         /// </summary>
-        /// <param name="configure">A callback to configure the SlimMessageBus <see cref="MessageBusBuilder" />.</param>
         /// <returns>The same <see cref="IServiceCollection" /> instance for chaining.</returns>
-        public IServiceCollection AddSlimBusForEfCore(Action<MessageBusBuilder> configure)
+        public IServiceCollection AddSlimBusEfCoreInterceptor<TDbContext>()
+            where TDbContext : DbContext
         {
+            EfAutoSavePostProcessorRegistration.RegisterDbContextType<TDbContext>();
+
+            if (serviceCollection.Any(serviceDescriptor =>
+                    serviceDescriptor.ServiceType == typeof(IRequestHandlerInterceptor<,>)))
+                return serviceCollection;
+
             serviceCollection
-                .AddScoped(typeof(IRequestHandlerInterceptor<,>), typeof(EfAutoSavePostProcessor<,>))
-                .AddSlimMessageBus(configure);
+                .AddScoped(typeof(IRequestHandlerInterceptor<,>), typeof(EfAutoSavePostProcessor<,>));
             return serviceCollection;
         }
     }

src/SlimBus/SlimBus.Extensions.Tests/Fixtures/Fixture.cs

@@ -17,7 +17,8 @@ public Fixture()
             .AddScoped<IMapper, ServiceMapper>()
             .AddDbContext<TestDbContext>(b => b.UseInMemoryDatabase(nameof(TestDbContext)))
             .AddScoped<DbContext>(p => p.GetRequiredService<TestDbContext>())
-            .AddSlimBusForEfCore(mmb =>
+            .AddSlimBusEfCoreInterceptor<TestDbContext>()
+            .AddSlimMessageBus(mmb =>
             {
                 //This is a global config for all the child busses
                 mmb.AddJsonSerializer()