Resolved #983: Added Configuration.ReplaceService method to override …

…built-in services.
aspnetboilerplate · May 5, 2016 · 223e4f8 · 223e4f8
1 parent 4958c9d
commit 223e4f8
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/src/Abp/AbpKernelModule.cs b/src/Abp/AbpKernelModule.cs
@@ -8,6 +8,7 @@
 using Abp.Authorization.Interceptors;
 using Abp.BackgroundJobs;
 using Abp.Configuration;
+using Abp.Configuration.Startup;
 using Abp.Dependency;
 using Abp.Domain.Uow;
 using Abp.Events.Bus;
@@ -74,7 +75,10 @@ public override void PreInitialize()
 
         public override void Initialize()
         {
-            base.Initialize();
+            foreach (var replaceAction in ((AbpStartupConfiguration)Configuration).ServiceReplaceActions.Values)
+            {
+                replaceAction();
+            }
 
             IocManager.IocContainer.Install(new EventBusInstaller(IocManager));
 

diff --git a/src/Abp/Configuration/Startup/AbpStartupConfiguration.cs b/src/Abp/Configuration/Startup/AbpStartupConfiguration.cs
@@ -1,4 +1,6 @@
-using Abp.Application.Features;
+using System;
+using System.Collections.Generic;
+using Abp.Application.Features;
 using Abp.Auditing;
 using Abp.BackgroundJobs;
 using Abp.Dependency;
@@ -14,6 +16,9 @@ namespace Abp.Configuration.Startup
     /// </summary>
     internal class AbpStartupConfiguration : DictionaryBasedConfig, IAbpStartupConfiguration
     {
+        /// <summary>
+        /// Reference to the IocManager.
+        /// </summary>
         public IIocManager IocManager { get; private set; }
 
         /// <summary>
@@ -85,6 +90,8 @@ internal class AbpStartupConfiguration : DictionaryBasedConfig, IAbpStartupConfi
         /// </summary>
         public IMultiTenancyConfig MultiTenancy { get; private set; }
 
+        public Dictionary<Type, Action> ServiceReplaceActions { get; private set; }
+
         /// <summary>
         /// Private constructor for singleton pattern.
         /// </summary>
@@ -108,6 +115,12 @@ public void Initialize()
             Caching = IocManager.Resolve<ICachingConfiguration>();
             BackgroundJobs = IocManager.Resolve<IBackgroundJobConfiguration>();
             Notifications = IocManager.Resolve<INotificationConfiguration>();
+            ServiceReplaceActions = new Dictionary<Type, Action>();
+        }
+
+        public void ReplaceService(Type type, Action replaceAction)
+        {
+            ServiceReplaceActions[type] = replaceAction;
         }
     }
 }
diff --git a/src/Abp/Configuration/Startup/IAbpStartupConfiguration.cs b/src/Abp/Configuration/Startup/IAbpStartupConfiguration.cs
@@ -1,4 +1,5 @@
-using Abp.Application.Features;
+using System;
+using Abp.Application.Features;
 using Abp.Auditing;
 using Abp.BackgroundJobs;
 using Abp.Dependency;
@@ -90,5 +91,7 @@ public interface IAbpStartupConfiguration : IDictionaryBasedConfig
         /// Used to configure notification system.
         /// </summary>
         INotificationConfiguration Notifications { get; }
+
+        void ReplaceService(Type type, Action replaceAction);
     }
 }