From e0bf7d8f16f7bcd4176b144d09409019104b78ee Mon Sep 17 00:00:00 2001
From: maliming <malimings@gmail.com>
Date: Mon, 18 Apr 2022 17:01:28 +0800
Subject: [PATCH] Update AbpDesignTimeDbContextBase.cs

---
 .../Design/AbpDesignTimeDbContextBase.cs           | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/framework/src/Volo.Abp.EntityFrameworkCore/Volo/Abp/EntityFrameworkCore/Design/AbpDesignTimeDbContextBase.cs b/framework/src/Volo.Abp.EntityFrameworkCore/Volo/Abp/EntityFrameworkCore/Design/AbpDesignTimeDbContextBase.cs
index c8fce37399..c3746f620d 100644
--- a/framework/src/Volo.Abp.EntityFrameworkCore/Volo/Abp/EntityFrameworkCore/Design/AbpDesignTimeDbContextBase.cs
+++ b/framework/src/Volo.Abp.EntityFrameworkCore/Volo/Abp/EntityFrameworkCore/Design/AbpDesignTimeDbContextBase.cs
@@ -19,11 +19,21 @@ public abstract class AbpDesignTimeDbContextBase<TModule, TContext> : IDesignTim
 
     protected virtual async Task<TContext> CreateDbContextAsync(string[] args)
     {
-        var application = await AbpApplicationFactory.CreateAsync<TModule>();
-        application.Services.ReplaceConfiguration(BuildConfiguration());
+        var application = await AbpApplicationFactory.CreateAsync<TModule>(options =>
+        {
+            options.Services.ReplaceConfiguration(BuildConfiguration());
+            ConfigureServices(options.Services);
+        });
+
         await application.InitializeAsync();
+
         return application.ServiceProvider.GetRequiredService<TContext>();
     }
 
+    protected virtual void ConfigureServices(IServiceCollection services)
+    {
+
+    }
+
     protected abstract IConfigurationRoot BuildConfiguration();
 }