1 files changed, 45 insertions, 0 deletions
diff --git a/src/marlin.web/Services/DatabaseUpgrader.cs b/src/marlin.web/Services/DatabaseUpgrader.cs
new file mode 100644
index 0000000..6bb8bac
--- /dev/null
+++ b/src/marlin.web/Services/DatabaseUpgrader.cs
@@ -0,0 +1,45 @@
+namespace marlin.web.Services;
+
+using Data;
+using Microsoft.EntityFrameworkCore;
+
+public class DatabaseUpgrader : IHostedService
+{
+    private readonly ILogger<DatabaseUpgrader> _logger;
+    private readonly IServiceProvider _serviceProvider;
+    
+    public DatabaseUpgrader(ILogger<DatabaseUpgrader> logger, IServiceProvider serviceProvider)
+    {
+        this._logger = logger;
+        this._serviceProvider = serviceProvider;
+    }
+
+    public async Task StartAsync(CancellationToken ct)
+    {
+        await using AsyncServiceScope scope = this._serviceProvider.CreateAsyncScope();
+        DbContext dbContext = scope.ServiceProvider.GetRequiredService<ApplicationDbContext>();
+
+        var migrations = await dbContext.Database.GetPendingMigrationsAsync(ct);
+        int pendingCount = migrations.Count();
+
+        if (pendingCount == 0)
+        {
+            this._logger.LogInformation("Database is up to date.");
+            return;
+        }
+
+        this._logger.LogInformation("Upgrading database {count} versions.", pendingCount);
+
+        try
+        {
+            await dbContext.Database.MigrateAsync(ct);
+            this._logger.LogInformation("Database upgraded.");
+        }
+        catch (Exception ex)
+        {
+            this._logger.LogError(ex, "Database upgrade failed, {count} behind.", pendingCount);
+        }
+    }
+
+    public Task StopAsync(CancellationToken ct) => Task.CompletedTask;
+}
+\ No newline at end of file