From ce89faebf9270c306bfa7ce92953a34de8e8dc9b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Halil=20=C4=B0brahim=20Kalkan?= <hi_kalkan@yahoo.com>
Date: Thu, 6 May 2021 23:26:22 +0300
Subject: [PATCH] Allow to globally disable class interceptors.

---
 .../AbpRegistrationBuilderExtensions.cs       | 22 ++++++++++++++-----
 ...eCollectionRegistrationActionExtensions.cs | 10 +++++++++
 .../ServiceRegistrationActionList.cs          |  2 +-
 3 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/framework/src/Volo.Abp.Autofac/Autofac/Builder/AbpRegistrationBuilderExtensions.cs b/framework/src/Volo.Abp.Autofac/Autofac/Builder/AbpRegistrationBuilderExtensions.cs
index 96caf956b1..776a8b9a7a 100644
--- a/framework/src/Volo.Abp.Autofac/Autofac/Builder/AbpRegistrationBuilderExtensions.cs
+++ b/framework/src/Volo.Abp.Autofac/Autofac/Builder/AbpRegistrationBuilderExtensions.cs
@@ -35,7 +35,11 @@ namespace Autofac.Builder
             return registrationBuilder;
         }
 
-        private static IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> InvokeRegistrationActions<TLimit, TActivatorData, TRegistrationStyle>(this IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> registrationBuilder, ServiceRegistrationActionList registrationActionList, Type serviceType, Type implementationType) 
+        private static IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> InvokeRegistrationActions<TLimit, TActivatorData, TRegistrationStyle>(
+            this IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> registrationBuilder, 
+            ServiceRegistrationActionList registrationActionList, 
+            Type serviceType,
+            Type implementationType) 
             where TActivatorData : ReflectionActivatorData
         {
             var serviceRegistredArgs = new OnServiceRegistredContext(serviceType, implementationType);
@@ -48,6 +52,7 @@ namespace Autofac.Builder
             if (serviceRegistredArgs.Interceptors.Any())
             {
                 registrationBuilder = registrationBuilder.AddInterceptors(
+                    registrationActionList,
                     serviceType,
                     serviceRegistredArgs.Interceptors
                 );
@@ -71,10 +76,12 @@ namespace Autofac.Builder
             return registrationBuilder;
         }
 
-        private static IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> AddInterceptors<TLimit, TActivatorData, TRegistrationStyle>(
-            this IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> registrationBuilder, 
-            Type serviceType,
-            IEnumerable<Type> interceptors)
+        private static IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle>
+            AddInterceptors<TLimit, TActivatorData, TRegistrationStyle>(
+                this IRegistrationBuilder<TLimit, TActivatorData, TRegistrationStyle> registrationBuilder,
+                ServiceRegistrationActionList serviceRegistrationActionList,
+                Type serviceType,
+                IEnumerable<Type> interceptors)
             where TActivatorData : ReflectionActivatorData
         {
             if (serviceType.IsInterface)
@@ -83,6 +90,11 @@ namespace Autofac.Builder
             }
             else
             {
+                if (serviceRegistrationActionList.IsClassInterceptorsDisabled)
+                {
+                    return registrationBuilder;
+                }
+                
                 (registrationBuilder as IRegistrationBuilder<TLimit, ConcreteReflectionActivatorData, TRegistrationStyle>)?.EnableClassInterceptors();
             }
 
diff --git a/framework/src/Volo.Abp.Core/Microsoft/Extensions/DependencyInjection/ServiceCollectionRegistrationActionExtensions.cs b/framework/src/Volo.Abp.Core/Microsoft/Extensions/DependencyInjection/ServiceCollectionRegistrationActionExtensions.cs
index 59cbdec860..41676f00d5 100644
--- a/framework/src/Volo.Abp.Core/Microsoft/Extensions/DependencyInjection/ServiceCollectionRegistrationActionExtensions.cs
+++ b/framework/src/Volo.Abp.Core/Microsoft/Extensions/DependencyInjection/ServiceCollectionRegistrationActionExtensions.cs
@@ -28,6 +28,16 @@ namespace Microsoft.Extensions.DependencyInjection
 
             return actionList;
         }
+        
+        public static void DisableAbpClassInterceptors(this IServiceCollection services)
+        {
+            GetOrCreateRegistrationActionList(services).IsClassInterceptorsDisabled = true;
+        }
+        
+        public static bool IsAbpClassInterceptorsDisabled(this IServiceCollection services)
+        {
+            return GetOrCreateRegistrationActionList(services).IsClassInterceptorsDisabled;
+        }
 
         // OnExposing
 
diff --git a/framework/src/Volo.Abp.Core/Volo/Abp/DependencyInjection/ServiceRegistrationActionList.cs b/framework/src/Volo.Abp.Core/Volo/Abp/DependencyInjection/ServiceRegistrationActionList.cs
index bb72d28929..19ab6cdba2 100644
--- a/framework/src/Volo.Abp.Core/Volo/Abp/DependencyInjection/ServiceRegistrationActionList.cs
+++ b/framework/src/Volo.Abp.Core/Volo/Abp/DependencyInjection/ServiceRegistrationActionList.cs
@@ -5,6 +5,6 @@ namespace Volo.Abp.DependencyInjection
 {
     public class ServiceRegistrationActionList : List<Action<IOnServiceRegistredContext>>
     {
-        
+        public bool IsClassInterceptorsDisabled { get; set; }
     }
 }
\ No newline at end of file