Add UnitOfWorkManager to DI container

swisschain · Aug 12, 2020 · de5e6f8 · de5e6f8
1 parent e0efc1a
commit de5e6f8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/src/Swisschain.Extensions.Idempotency/IdempotencyConfigurationBuilder.cs b/src/Swisschain.Extensions.Idempotency/IdempotencyConfigurationBuilder.cs
@@ -2,7 +2,8 @@
 
 namespace Swisschain.Extensions.Idempotency
 {
-    public sealed class IdempotencyConfigurationBuilder
+    public sealed class IdempotencyConfigurationBuilder<TUnitOfWork>
+        where TUnitOfWork : UnitOfWorkBase
     {
         internal IdempotencyConfigurationBuilder(IServiceCollection services)
         {

diff --git a/src/Swisschain.Extensions.Idempotency/IdempotencyServiceCollectionExtensions.cs b/src/Swisschain.Extensions.Idempotency/IdempotencyServiceCollectionExtensions.cs
@@ -5,21 +5,22 @@ namespace Swisschain.Extensions.Idempotency
 {
     public static class IdempotencyServiceCollectionExtensions
     {
-        /// <summary>
-        /// 
-        /// </summary>
-        public static IServiceCollection AddIdempotency(this IServiceCollection services, Action<IdempotencyConfigurationBuilder> config)
+        public static IServiceCollection AddIdempotency<TUnitOfWork>(this IServiceCollection services, Action<IdempotencyConfigurationBuilder<TUnitOfWork>> config)
+            where TUnitOfWork : UnitOfWorkBase
         {
+            services.AddTransient<IUnitOfWorkManager<TUnitOfWork>, UnitOfWorkManager<TUnitOfWork>>();
+
             services.AddTransient<IOutboxManager, OutboxManager>();
             services.AddTransient<IIdGenerator, IdGenerator>();
 
             services.AddTransient<IOutboxDispatcher, DefaultOutboxDispatcher>();
+
             services.AddTransient<IOutboxReadRepository, DefaultOutboxReadRepository>();
             services.AddTransient<IIdGeneratorRepository, DefaultIdGeneratorRepository>();
 
             if (config != null)
             {
-                var configBuilder = new IdempotencyConfigurationBuilder(services);
+                var configBuilder = new IdempotencyConfigurationBuilder<TUnitOfWork>(services);
 
                 config.Invoke(configBuilder);
             }