Merge pull request #267 from crupest/backend

春节大换血 Spring festival big change.

1 files changed, 8 insertions, 9 deletions

diff --git a/BackEnd/Timeline/Program.cs b/BackEnd/Timeline/Program.cs
index 75bf6154..19fa6e37 100644
--- a/BackEnd/Timeline/Program.cs
+++ b/BackEnd/Timeline/Program.cs
@@ -4,8 +4,10 @@ using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;

 using Microsoft.Extensions.Hosting;

 using System.Resources;

+using System.Threading.Tasks;

 using Timeline.Entities;

 using Timeline.Services;

+using Timeline.Services.Migration;

 

 [assembly: NeutralResourcesLanguage("en")]

 

@@ -13,22 +15,19 @@ namespace Timeline
 {

     public static class Program

     {

-        public static void Main(string[] args)

+        public async static Task Main(string[] args)

         {

             var host = CreateWebHostBuilder(args).Build();

 

-            var env = host.Services.GetRequiredService<IWebHostEnvironment>();

-

             var databaseBackupService = host.Services.GetRequiredService<IDatabaseBackupService>();

             databaseBackupService.BackupNow();

 

-            if (env.IsProduction())

+            using (var scope = host.Services.CreateScope())

             {

-                using (var scope = host.Services.CreateScope())

-                {

-                    var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();

-                    databaseContext.Database.Migrate();

-                }

+                var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();

+                await databaseContext.Database.MigrateAsync();

+                var customMigrationManager = scope.ServiceProvider.GetRequiredService<ICustomMigrationManager>();

+                await customMigrationManager.Migrate();

             }

 

             host.Run();