Update migration applying code

Author: mitch-b

src/MaintenanceLog/DatabaseUtility.cs

@@ -1,11 +1,13 @@
-using MaintenanceLog.Data;
+using System.Threading;
+using MaintenanceLog.Data;
 using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Migrations;
 
 namespace MaintenanceLog;
 
 public static class DatabaseUtility
 {
-    public static async Task EnsureDbCreatedAndSeedWithDefaults(DbContextOptions<ApplicationDbContext> options)
+    public static async Task EnsureDbCreatedAndSeedWithDefaults(DbContextOptions<ApplicationDbContext> options, IMigrator migrator)
     {
         var factory = new LoggerFactory();
         var builder = new DbContextOptionsBuilder<ApplicationDbContext>(options)
@@ -20,11 +22,23 @@ public static async Task EnsureDbCreatedAndSeedWithDefaults(DbContextOptions<App
 
         try
         {
-            if ((await context.Database.GetPendingMigrationsAsync()).Any())
+            // not good with distributed UI... but for now...
+            Console.WriteLine("Applying migrations...");
+            var migrations = await context.Database.GetPendingMigrationsAsync();
+            foreach (var migration in migrations)
             {
-                // not good with distributed UI... but for now...
-                Console.WriteLine("Applying migrations...");
-                await context.Database.MigrateAsync();
+                // Execute all migrations in one single transaction
+                using var tran = await context.Database.BeginTransactionAsync();
+                try
+                {
+                    await migrator.MigrateAsync(migration);
+                    await tran.CommitAsync();
+                }
+                catch (Exception exc)
+                {
+                    await tran.RollbackAsync();
+                    throw new Exception($"Error while applying db migration '{migration}'.", exc);
+                }
             }
         }
         catch (Exception ex)

src/MaintenanceLog/Program.cs

@@ -13,6 +13,7 @@
 using Microsoft.Extensions.Options;
 using System.Text.Json;
 using Blazored.LocalStorage;
+using Microsoft.EntityFrameworkCore.Migrations;
 
 var builder = WebApplication.CreateBuilder(args);
 
@@ -93,7 +94,8 @@
 // handled this way in production. 
 await using var scope = app.Services.GetRequiredService<IServiceScopeFactory>().CreateAsyncScope();
 var options = scope.ServiceProvider.GetRequiredService<DbContextOptions<ApplicationDbContext>>();
-await MaintenanceLog.DatabaseUtility.EnsureDbCreatedAndSeedWithDefaults(options);
+var migrator = scope.ServiceProvider.GetService<IMigrator>();
+await MaintenanceLog.DatabaseUtility.EnsureDbCreatedAndSeedWithDefaults(options, migrator);
 
 app.MapControllers();