added open generic support for decorator func

khellang · Jun 6, 2019 · 1dff356 · 1dff356
1 parent 5df6b40
commit 1dff356
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/Scrutor/ServiceCollectionExtensions.Decoration.cs b/src/Scrutor/ServiceCollectionExtensions.Decoration.cs
@@ -258,6 +258,16 @@ private static bool IsSameGenericType(Type t1, Type t2)
             return t1.IsGenericType && t2.IsGenericType && t1.GetGenericTypeDefinition() == t2.GetGenericTypeDefinition();
         }
 
+        private static IServiceCollection DecorateOpenGeneric(this IServiceCollection services, Type serviceType, Func<object, IServiceProvider, object> decorator)
+        {
+            if (services.TryDecorateOpenGeneric(serviceType, decorator))
+            {
+                return services;
+            }
+
+            throw new MissingTypeRegistrationException(serviceType);
+        }
+
         private static bool TryDecorateOpenGeneric(this IServiceCollection services, Type serviceType, Type decoratorType)
         {
             bool TryDecorate(Type[] typeArguments)