Update database handling code (#247)

2026-10-14 17:54:42 +00:00 · 2026-08-25 13:42:51 -04:00
parent 03ca966708
commit 8a047e8d9f
37 changed files with 386 additions and 405 deletions
--- a/web/backend/Database/Database.Migrations/DbMigrator.cs
+++ b/web/backend/Database/Database.Migrations/DbMigrator.cs
@@ -0,0 +1,113 @@
+using System.Reflection;
+using Dapper;
+using Database.Connection;
+using DbUp;
+using DbUp.Engine;
+using Microsoft.Data.SqlClient;
+
+namespace Database.Migrations;
+
+/// <summary>
+/// Owns the configuration and state for a single database migration run,
+/// and performs the drop/create/deploy steps against SQL Server.
+/// </summary>
+internal sealed class DbMigrator(
+    ISqlConnectionFactory standardConnectionStandardConnectionFactory,
+    ISqlConnectionFactory masterConnectionConnectionFactory,
+    bool clearDatabase)
+{
+    /// <summary>
+    /// Runs the full migration workflow: optional drop, create-if-not-exists,
+    /// then deploy DbUp scripts. Throws if the upgrade fails.
+    /// </summary>
+    public async Task RunAsync(CancellationToken cancellationToken = default)
+    {
+        if (clearDatabase)
+        {
+            await DropDatabaseIfExistsAsync(cancellationToken);
+        }
+
+        await CreateDatabaseIfNotExistsAsync(cancellationToken);
+
+        DatabaseUpgradeResult result = DeployMigrations();
+
+        if (!result.Successful)
+        {
+            throw result.Error;
+        }
+    }
+
+    private async Task DropDatabaseIfExistsAsync(CancellationToken cancellationToken = default)
+    {
+        string databaseName = ResolveDatabaseName();
+
+        await using SqlConnection connection = masterConnectionConnectionFactory.CreateConnection();
+        await connection.OpenAsync(cancellationToken);
+
+        await connection.ExecuteAsync(
+            new CommandDefinition(
+                """
+                DECLARE @statement nvarchar(max) =
+                    N'ALTER DATABASE ' + QUOTENAME(@databaseName) + N' SET SINGLE_USER WITH ROLLBACK IMMEDIATE;'
+                    + N' DROP DATABASE ' + QUOTENAME(@databaseName) + N';';
+
+                IF EXISTS (SELECT 1 FROM sys.databases WHERE name = @databaseName)
+                    EXEC sys.sp_executesql @statement;
+                """,
+                new { databaseName },
+                cancellationToken: cancellationToken
+            )
+        );
+
+        Console.WriteLine($"Database '{databaseName}' dropped, or did not exist.");
+    }
+
+    private async Task CreateDatabaseIfNotExistsAsync(CancellationToken cancellationToken = default)
+    {
+        string databaseName = ResolveDatabaseName();
+
+        await using SqlConnection connection = masterConnectionConnectionFactory.CreateConnection();
+        await connection.OpenAsync(cancellationToken);
+
+        await connection.ExecuteAsync(
+            new CommandDefinition(
+                """
+                DECLARE @statement nvarchar(max);
+                IF NOT EXISTS (SELECT 1 FROM sys.databases WHERE name = @databaseName)
+                BEGIN
+                    SET @statement = N'CREATE DATABASE ' + QUOTENAME(@databaseName) + N';';
+                    EXEC sys.sp_executesql @statement;
+                END
+                """,
+                new { databaseName },
+                cancellationToken: cancellationToken
+            )
+        );
+
+        Console.WriteLine($"Database '{databaseName}' is present.");
+    }
+
+    private DatabaseUpgradeResult DeployMigrations()
+    {
+        UpgradeEngine upgrader = DeployChanges
+            .To.SqlDatabase(standardConnectionStandardConnectionFactory.ConnectionString)
+            .WithScriptsEmbeddedInAssembly(Assembly.GetExecutingAssembly())
+            .WithTransactionPerScript()
+            .LogToConsole()
+            .Build();
+
+        return upgrader.PerformUpgrade();
+    }
+
+    private string ResolveDatabaseName()
+    {
+        return !SqlConnectionStringHelper.TryGetDatabaseNameFromConnectionString(
+            standardConnectionStandardConnectionFactory.ConnectionString,
+            out var databaseName
+        )
+            ? throw new InvalidOperationException(
+                "Unable to resolve a database name from the connection string."
+            )
+            : databaseName;
+    }
+}