Inline check for already decorated type

khellang · May 23, 2022 · 5067e34 · 5067e34
1 parent e03e051
commit 5067e34
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/Scrutor/Decoration/Decorator.cs b/src/Scrutor/Decoration/Decorator.cs
@@ -52,8 +52,12 @@ private int DecorateServices(IServiceCollection services)
             {
                 var serviceDescriptor = services[i];
 
-                if (IsNotAlreadyDecorated(serviceDescriptor)
-                    && DecoratorStrategy.CanDecorate(serviceDescriptor.ServiceType))
+                if (serviceDescriptor.ServiceType is DecoratedType)
+                {
+                    continue; // Service has already been decorated.
+                }
+
+                if (DecoratorStrategy.CanDecorate(serviceDescriptor.ServiceType))
                 {
                     var decoratedType = new DecoratedType(serviceDescriptor.ServiceType);
 
@@ -70,7 +74,5 @@ private int DecorateServices(IServiceCollection services)
 
             return decorated;
         }
-
-        private static bool IsNotAlreadyDecorated(ServiceDescriptor serviceDescriptor) => serviceDescriptor.ServiceType is not DecoratedType;
     }
 }