From e7fd5014f909c49d3d2acf2d6174d82fc3434dbb Mon Sep 17 00:00:00 2001
From: Galip Tolga Erdem <galiperdem@gmail.com>
Date: Tue, 11 Feb 2020 15:36:56 +0300
Subject: [PATCH] Updated MigrationService with tenant migrations.

---
 .../Data/MyProjectNameDbMigrationService.cs   | 43 ++++++++++++++++---
 ...meworkCoreMyProjectNameDbSchemaMigrator.cs | 16 +++++--
 2 files changed, 50 insertions(+), 9 deletions(-)

diff --git a/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.Domain/Data/MyProjectNameDbMigrationService.cs b/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.Domain/Data/MyProjectNameDbMigrationService.cs
index f6fd008e66..5645e10672 100644
--- a/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.Domain/Data/MyProjectNameDbMigrationService.cs
+++ b/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.Domain/Data/MyProjectNameDbMigrationService.cs
@@ -1,8 +1,12 @@
-using System.Threading.Tasks;
+using System;
+using System.Linq;
+using System.Threading.Tasks;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Logging.Abstractions;
 using Volo.Abp.Data;
 using Volo.Abp.DependencyInjection;
+using Volo.Abp.MultiTenancy;
+using Volo.Abp.TenantManagement;
 
 namespace MyCompanyName.MyProjectName.Data
 {
@@ -12,13 +16,19 @@ namespace MyCompanyName.MyProjectName.Data
 
         private readonly IDataSeeder _dataSeeder;
         private readonly IMyProjectNameDbSchemaMigrator _dbSchemaMigrator;
+        private readonly ITenantRepository _tenantRepository;
+        private readonly ICurrentTenant _currentTenant;
 
         public MyProjectNameDbMigrationService(
             IDataSeeder dataSeeder,
-            IMyProjectNameDbSchemaMigrator dbSchemaMigrator)
+            IMyProjectNameDbSchemaMigrator dbSchemaMigrator,
+            ITenantRepository tenantRepository,
+            ICurrentTenant currentTenant)
         {
             _dataSeeder = dataSeeder;
             _dbSchemaMigrator = dbSchemaMigrator;
+            _tenantRepository = tenantRepository;
+            _currentTenant = currentTenant;
 
             Logger = NullLogger<MyProjectNameDbMigrationService>.Instance;
         }
@@ -27,13 +37,36 @@ namespace MyCompanyName.MyProjectName.Data
         {
             Logger.LogInformation("Started database migrations...");
 
-            Logger.LogInformation("Migrating database schema...");
+            await MigrateHostDatabaseAsync();
+
+            foreach (var tenant in await _tenantRepository.GetListAsync())
+            {
+                using (_currentTenant.Change(tenant.Id))
+                {
+
+                    await MigrateTenantDatabasesAsync(tenant);
+                }
+            }
+            Logger.LogInformation("Successfully completed database migrations.");
+        }
+        private async Task MigrateHostDatabaseAsync()
+        {
+            Logger.LogInformation("Migrating host database schema...");
             await _dbSchemaMigrator.MigrateAsync();
 
-            Logger.LogInformation("Executing database seed...");
+            Logger.LogInformation("Executing host database seed...");
             await _dataSeeder.SeedAsync();
 
-            Logger.LogInformation("Successfully completed database migrations.");
+            Logger.LogInformation("Successfully completed host database migrations.");
+        }
+        private async Task MigrateTenantDatabasesAsync(Tenant tenant)
+        {
+            Logger.LogInformation($"Migrating {tenant.Name} database schema...");
+            await _dbSchemaMigrator.MigrateAsync();
+
+            Logger.LogInformation($"Executing {tenant.Name} database seed...");
+            await _dataSeeder.SeedAsync(tenant.Id);
+            Logger.LogInformation($"Successfully completed {tenant.Name} database migrations.");
         }
     }
 }
\ No newline at end of file
diff --git a/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.EntityFrameworkCore.DbMigrations/EntityFrameworkCore/EntityFrameworkCoreMyProjectNameDbSchemaMigrator.cs b/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.EntityFrameworkCore.DbMigrations/EntityFrameworkCore/EntityFrameworkCoreMyProjectNameDbSchemaMigrator.cs
index 7064697d1a..db119bb082 100644
--- a/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.EntityFrameworkCore.DbMigrations/EntityFrameworkCore/EntityFrameworkCoreMyProjectNameDbSchemaMigrator.cs
+++ b/templates/app/aspnet-core/src/MyCompanyName.MyProjectName.EntityFrameworkCore.DbMigrations/EntityFrameworkCore/EntityFrameworkCoreMyProjectNameDbSchemaMigrator.cs
@@ -1,5 +1,7 @@
-using System.Threading.Tasks;
+using System;
+using System.Threading.Tasks;
 using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.DependencyInjection;
 using MyCompanyName.MyProjectName.Data;
 using Volo.Abp.DependencyInjection;
 
@@ -10,15 +12,21 @@ namespace MyCompanyName.MyProjectName.EntityFrameworkCore
         : IMyProjectNameDbSchemaMigrator, ITransientDependency
     {
         private readonly MyProjectNameMigrationsDbContext _dbContext;
+        private readonly IServiceProvider _serviceProvider;
 
-        public EntityFrameworkCoreMyProjectNameDbSchemaMigrator(MyProjectNameMigrationsDbContext dbContext)
+        public EntityFrameworkCoreMyProjectNameDbSchemaMigrator(
+            IServiceProvider serviceProvider
+            )
         {
-            _dbContext = dbContext;
+            _serviceProvider = serviceProvider;
         }
 
         public async Task MigrateAsync()
         {
-            await _dbContext.Database.MigrateAsync();
+            await _serviceProvider
+                .GetRequiredService<MyProjectNameMigrationsDbContext>()
+                .Database
+                .MigrateAsync();
         }
     }
 }
\ No newline at end of file