diff --git a/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs b/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs index 92c6fca..b2bba17 100644 --- a/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs +++ b/src/KSFramework/KSMessaging/Extensions/RegisterMediatorServices.cs @@ -20,6 +20,8 @@ public static IServiceCollection AddKSMediator(this IServiceCollection services, { services.AddScoped(); services.AddScoped(sp => sp.GetRequiredService()); + + services.RegisterAllImplementationsOf(assemblies); services.Scan(scan => scan .FromAssemblies(assemblies) @@ -72,8 +74,6 @@ public static IServiceCollection AddKSMediator(this IServiceCollection services, // If you have default behaviors (e.g., logging), register them here services.AddScoped(typeof(IStreamPipelineBehavior<,>), typeof(StreamLoggingBehavior<,>)); - - services.RegisterAllImplementationsOf(assemblies); return services; }