being able to register handler type without specifying events

jacqueskang · Nov 21, 2018 · d816465 · d816465
1 parent e410c20
commit d816465
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/JKang.EventBus.Core/DependencyInjection/EventBusBuilder.cs b/src/JKang.EventBus.Core/DependencyInjection/EventBusBuilder.cs
@@ -1,5 +1,8 @@
 using JKang.EventBus.MultiChannel;
 using Microsoft.Extensions.DependencyInjection;
+using System;
+using System.Collections.Generic;
+using System.Linq;
 
 namespace JKang.EventBus.DependencyInjection
 {
@@ -38,5 +41,22 @@ public IEventBusBuilder AddEventHandler<TEvent, TEventHandler>()
             Services.AddScoped<IEventHandler<TEvent>, TEventHandler>();
             return this;
         }
+
+        public IEventBusBuilder AddEventHandler<TEventHandler>()
+            where TEventHandler : class
+        {
+            Type implementationType = typeof(TEventHandler);
+            IEnumerable<Type> serviceTypes = implementationType
+                .GetInterfaces()
+                .Where(i => i.IsGenericType)
+                .Where(i => i.GetGenericTypeDefinition() == typeof(IEventHandler<>));
+
+            foreach (Type serviceType in serviceTypes)
+            {
+                Services.AddScoped(serviceType, implementationType);
+            }
+
+            return this;
+        }
     }
 }
diff --git a/src/Samples.EventBus.AspNetCore/Startup.cs b/src/Samples.EventBus.AspNetCore/Startup.cs
@@ -30,7 +30,7 @@ public void ConfigureServices(IServiceCollection services)
                     builder
                         .AddInMemoryEventBus()
                         .AddAmazonSnsEventPublisher(x => x.Region = "eu-west-3")
-                        .AddEventHandler<MessageSent, MessageSentEventHandler>()
+                        .AddEventHandler<MessageSentEventHandler>()
                         ;
                 });
         }