Refactored OutboxExtensionAddOutbox (#302)

adimiko · Jun 16, 2024 · 2403a6c · 2403a6c
1 parent a059328
commit 2403a6c
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/source/TransactionalBox/Extensions/Outbox/OutboxExtensionAddOutbox.cs b/source/TransactionalBox/Extensions/Outbox/OutboxExtensionAddOutbox.cs
@@ -69,18 +69,7 @@ public static void AddOutbox(
                 assemblyConfiguraton(assemblyConfigurator);
             }
 
-            var assemblies = assemblyConfigurator.Assemblies;
-
-            var allTypes = assemblyConfigurator.Assemblies.SelectMany(x => x.GetTypes());
-
-            var outboxMessageDefinitions = allTypes.Where(x => x.BaseType != null && x.BaseType.IsGenericType && x.BaseType.GetGenericTypeDefinition() == typeof(OutboxDefinition<>)).ToList();
-
-            foreach (var outboxMessageDefinition in outboxMessageDefinitions)
-            {
-                var messageType = outboxMessageDefinition.BaseType.GetGenericArguments()[0];
-
-                services.AddKeyedSingleton(typeof(IOutboxDefinition), messageType, outboxMessageDefinition);
-            }
+            services.RegisterOutboxMessageDefinitions(assemblyConfigurator);
 
             services.AddSingleton<TransportEnvelopeFactory>();
 
@@ -112,5 +101,21 @@ public static void AddOutbox(
 
             services.AddHostedService<OutboxStartup>();
         }
+
+        private static void RegisterOutboxMessageDefinitions(this IServiceCollection services, AssemblyConfigurator assemblyConfigurator) 
+        {
+            var assemblies = assemblyConfigurator.Assemblies;
+
+            var allTypes = assemblyConfigurator.Assemblies.SelectMany(x => x.GetTypes());
+
+            var outboxMessageDefinitions = allTypes.Where(x => x.BaseType != null && x.BaseType.IsGenericType && x.BaseType.GetGenericTypeDefinition() == typeof(OutboxDefinition<>)).ToList();
+
+            foreach (var outboxMessageDefinition in outboxMessageDefinitions)
+            {
+                var messageType = outboxMessageDefinition.BaseType.GetGenericArguments()[0];
+
+                services.AddKeyedSingleton(typeof(IOutboxDefinition), messageType, outboxMessageDefinition);
+            }
+        }
     }
 }