From 83de31a999132b052f1ccec75eb664c9e4a18e87 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Fri, 12 Feb 2021 22:10:56 +0800
Subject: feat: Add databse custom migration service.

---
 BackEnd/Timeline/Program.cs | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

(limited to 'BackEnd/Timeline/Program.cs')

diff --git a/BackEnd/Timeline/Program.cs b/BackEnd/Timeline/Program.cs
index 75bf6154..0f75908f 100644
--- a/BackEnd/Timeline/Program.cs
+++ b/BackEnd/Timeline/Program.cs
@@ -6,6 +6,7 @@ using Microsoft.Extensions.Hosting;
 using System.Resources;
 using Timeline.Entities;
 using Timeline.Services;
+using Timeline.Services.Migration;
 
 [assembly: NeutralResourcesLanguage("en")]
 
@@ -17,18 +18,16 @@ namespace Timeline
         {
             var host = CreateWebHostBuilder(args).Build();
 
-            var env = host.Services.GetRequiredService<IWebHostEnvironment>();
+            using (var scope = host.Services.CreateScope())
+            {
+                var databaseBackupService = scope.ServiceProvider.GetRequiredService<IDatabaseBackupService>();
+                databaseBackupService.BackupNow();
 
-            var databaseBackupService = host.Services.GetRequiredService<IDatabaseBackupService>();
-            databaseBackupService.BackupNow();
+                var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();
+                databaseContext.Database.Migrate();
 
-            if (env.IsProduction())
-            {
-                using (var scope = host.Services.CreateScope())
-                {
-                    var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();
-                    databaseContext.Database.Migrate();
-                }
+                var customMigrationManager = scope.ServiceProvider.GetRequiredService<ICustomMigrationManager>();
+                customMigrationManager.Migrate();
             }
 
             host.Run();
-- 
cgit v1.2.3