FIxed side effect handler registration (#91)

osstotalsoft · Aug 5, 2020 · 57c40db · 57c40db
1 parent 4077c4c
commit 57c40db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Core/NBB.Core.Effects/DependencyInjectionExtensions.cs b/src/Core/NBB.Core.Effects/DependencyInjectionExtensions.cs
@@ -23,7 +23,7 @@ public static IServiceCollection AddSideEffectHandler<TOutput>(this IServiceColl
         public static IServiceCollection AddSideEffectHandler<TSideEffect, TOutput>(this IServiceCollection services, Func<TSideEffect, CancellationToken, Task<TOutput>> handlerFn)
             where TSideEffect : ISideEffect<TOutput>
         {
-            services.AddSingleton(new GenericSideEffectHandler<TSideEffect, TOutput>(handlerFn));
+            services.AddSingleton<ISideEffectHandler<TSideEffect, TOutput>>(new GenericSideEffectHandler<TSideEffect, TOutput>(handlerFn));
             return services;
         }
 
@@ -35,7 +35,7 @@ Task<TOutput> HandlerFnAsync(TSideEffect sideEffect, CancellationToken _cancella
                 return Task.FromResult(handlerFn(sideEffect));
             }
 
-            services.AddSingleton(new GenericSideEffectHandler<TSideEffect, TOutput>(HandlerFnAsync));
+            services.AddSingleton<ISideEffectHandler<TSideEffect, TOutput>>(new GenericSideEffectHandler<TSideEffect, TOutput>(HandlerFnAsync));
             return services;
         }
     }