From ad8aec9756aee51f52af0b405a0dc0db822ab91f Mon Sep 17 00:00:00 2001
From: rkac-bw <148072202+rkac-bw@users.noreply.github.com>
Date: Wed, 9 Oct 2024 08:48:19 -0600
Subject: [PATCH] Add variable for production migration transaction level
 (#4702)

* Addd variable for production migration transaction level

* Added variable for production migration transaction level with default value

* Clean up comments

* Removed uneeded directive

* Changed time format for timeout on migration

* white space formatting

* white space formatting again

* white space formatting once again

* white space formatting once again

* clean up

* CHnaged to builder.WithoutTransaction()

* Changed to optyion flag from n to nt for notransaction

* Changed to optyion flag from n to no-transaction for  without transaction

* Change desription of option

---------

Co-authored-by: Matt Bishop <mbishop@bitwarden.com>
(cherry picked from commit b38b537ed1554fd568f3c4b50d5b1246ee449e32)
---
 util/Migrator/DbMigrator.cs          | 28 +++++++++++++++++++++-------
 util/MsSqlMigratorUtility/Program.cs | 10 ++++++----
 2 files changed, 27 insertions(+), 11 deletions(-)

diff --git a/util/Migrator/DbMigrator.cs b/util/Migrator/DbMigrator.cs
index 11b80fac788d..b9584326e90d 100644
--- a/util/Migrator/DbMigrator.cs
+++ b/util/Migrator/DbMigrator.cs
@@ -14,13 +14,15 @@ public class DbMigrator
     private readonly string _connectionString;
     private readonly ILogger<DbMigrator> _logger;
     private readonly bool _skipDatabasePreparation;
+    private readonly bool _noTransactionMigration;
 
     public DbMigrator(string connectionString, ILogger<DbMigrator> logger = null,
-        bool skipDatabasePreparation = false)
+       bool skipDatabasePreparation = false, bool noTransactionMigration = false)
     {
         _connectionString = connectionString;
         _logger = logger ?? CreateLogger();
         _skipDatabasePreparation = skipDatabasePreparation;
+        _noTransactionMigration = noTransactionMigration;
     }
 
     public bool MigrateMsSqlDatabaseWithRetries(bool enableLogging = true,
@@ -30,6 +32,7 @@ public bool MigrateMsSqlDatabaseWithRetries(bool enableLogging = true,
         CancellationToken cancellationToken = default)
     {
         var attempt = 1;
+
         while (attempt < 10)
         {
             try
@@ -69,6 +72,7 @@ private void PrepareDatabase(CancellationToken cancellationToken = default)
         using (var connection = new SqlConnection(masterConnectionString))
         {
             var databaseName = new SqlConnectionStringBuilder(_connectionString).InitialCatalog;
+
             if (string.IsNullOrWhiteSpace(databaseName))
             {
                 databaseName = "vault";
@@ -105,10 +109,10 @@ private void PrepareDatabase(CancellationToken cancellationToken = default)
     }
 
     private bool MigrateDatabase(bool enableLogging = true,
-        bool repeatable = false,
-        string folderName = MigratorConstants.DefaultMigrationsFolderName,
-        bool dryRun = false,
-        CancellationToken cancellationToken = default)
+    bool repeatable = false,
+    string folderName = MigratorConstants.DefaultMigrationsFolderName,
+    bool dryRun = false,
+    CancellationToken cancellationToken = default)
     {
         if (enableLogging)
         {
@@ -121,8 +125,17 @@ private bool MigrateDatabase(bool enableLogging = true,
             .SqlDatabase(_connectionString)
             .WithScriptsAndCodeEmbeddedInAssembly(Assembly.GetExecutingAssembly(),
                 s => s.Contains($".{folderName}.") && !s.Contains(".Archive."))
-            .WithTransaction()
-            .WithExecutionTimeout(new TimeSpan(0, 5, 0));
+            .WithExecutionTimeout(TimeSpan.FromMinutes(5));
+
+        if (_noTransactionMigration)
+        {
+            builder = builder.WithoutTransaction()
+                .WithExecutionTimeout(TimeSpan.FromMinutes(60));
+        }
+        else
+        {
+            builder = builder.WithTransaction();
+        }
 
         if (repeatable)
         {
@@ -144,6 +157,7 @@ private bool MigrateDatabase(bool enableLogging = true,
         {
             var scriptsToExec = upgrader.GetScriptsToExecute();
             var stringBuilder = new StringBuilder("Scripts that will be applied:");
+
             foreach (var script in scriptsToExec)
             {
                 stringBuilder.AppendLine(script.Name);
diff --git a/util/MsSqlMigratorUtility/Program.cs b/util/MsSqlMigratorUtility/Program.cs
index 03e4716e06e0..056cb696f840 100644
--- a/util/MsSqlMigratorUtility/Program.cs
+++ b/util/MsSqlMigratorUtility/Program.cs
@@ -17,13 +17,15 @@ public void Execute(
         [Option('f', "folder", Description = "Folder name of database scripts")]
         string folderName = MigratorConstants.DefaultMigrationsFolderName,
         [Option('d', "dry-run", Description = "Print the scripts that will be applied without actually executing them")]
-        bool dryRun = false
-        ) => MigrateDatabase(databaseConnectionString, repeatable, folderName, dryRun);
+        bool dryRun = false,
+        [Option("no-transaction", Description = "Run without adding transaction per script or all scripts")]
+        bool noTransactionMigration = false
+        ) => MigrateDatabase(databaseConnectionString, repeatable, folderName, dryRun, noTransactionMigration);
 
     private static bool MigrateDatabase(string databaseConnectionString,
-        bool repeatable = false, string folderName = "", bool dryRun = false)
+        bool repeatable = false, string folderName = "", bool dryRun = false, bool noTransactionMigration = false)
     {
-        var migrator = new DbMigrator(databaseConnectionString);
+        var migrator = new DbMigrator(databaseConnectionString, noTransactionMigration: noTransactionMigration);
         bool success;
         if (!string.IsNullOrWhiteSpace(folderName))
         {