added SupportedDatabasesForDropDatabase extensions

DbUp · Jul 28, 2024 · 58837ad · 58837ad
1 parent 461bd38
commit 58837ad
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 0 deletions.
diff --git a/src/Tests/ApprovalFiles/NoPublicApiChanges.Run.DotNet.verified.cs b/src/Tests/ApprovalFiles/NoPublicApiChanges.Run.DotNet.verified.cs
@@ -14,6 +14,9 @@ public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase s
     public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, string collation) { }
     public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, int commandTimeout, string collation) { }
     public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, int timeout = -1, string collation = null) { }
+    public static void MySqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString) { }
+    public static void MySqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, int commandTimeout) { }
+    public static void MySqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, int timeout = -1) { }
 }
 namespace DbUp.MySql
 {

diff --git a/src/Tests/ApprovalFiles/NoPublicApiChanges.Run.Net.verified.cs b/src/Tests/ApprovalFiles/NoPublicApiChanges.Run.Net.verified.cs
@@ -14,6 +14,9 @@ public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase s
     public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, string collation) { }
     public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, int commandTimeout, string collation) { }
     public static void MySqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, int timeout = -1, string collation = null) { }
+    public static void MySqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString) { }
+    public static void MySqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, int commandTimeout) { }
+    public static void MySqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, int timeout = -1) { }
 }
 namespace DbUp.MySql
 {

diff --git a/src/dbup-mysql/MySqlExtensions.cs b/src/dbup-mysql/MySqlExtensions.cs
@@ -199,6 +199,69 @@ public static void MySqlDatabase(
         }
     }
 
+    /// <summary>
+    /// Drop the database specified in the connection string.
+    /// </summary>
+    /// <param name="supported">Fluent helper type.</param>
+    /// <param name="connectionString">The connection string.</param>
+    public static void MySqlDatabase(
+        this SupportedDatabasesForDropDatabase supported,
+        string connectionString)
+    {
+        MySqlDatabase(supported, connectionString, new ConsoleUpgradeLog());
+    }
+
+    /// <summary>
+    /// Drop the database specified in the connection string.
+    /// </summary>
+    /// <param name="supported">Fluent helper type.</param>
+    /// <param name="connectionString">The connection string.</param>
+    /// <param name="commandTimeout">Use this to set the command time out for dropping a database in case you're encountering a time out in this operation.</param>
+    public static void MySqlDatabase(
+        this SupportedDatabasesForDropDatabase supported,
+        string connectionString,
+        int commandTimeout)
+    {
+        MySqlDatabase(supported, connectionString, new ConsoleUpgradeLog(), commandTimeout);
+    }
+
+    /// <summary>
+    /// Drop the database specified in the connection string.
+    /// </summary>
+    /// <param name="supported">Fluent helper type.</param>
+    /// <param name="connectionString">The connection string.</param>
+    /// <param name="logger">The <see cref="DbUp.Engine.Output.IUpgradeLog"/> used to record actions.</param>
+    /// <param name="timeout">Use this to set the command time out for dropping a database in case you're encountering a time out in this operation.</param>
+    public static void MySqlDatabase(
+        this SupportedDatabasesForDropDatabase supported,
+        string connectionString,
+        IUpgradeLog logger,
+        int timeout = -1)
+    {
+        GetMysqlConnectionStringBuilder(connectionString, logger, out var masterConnectionString, out var databaseName);
+        using (var connection = new MySqlConnection(masterConnectionString))
+        {
+            connection.Open();
+            if (!DatabaseExists(connection, databaseName))
+                return;
+            var dropDatabaseCommand = new MySqlCommand($"DROP DATABASE @databaseName;", connection)
+            {
+                CommandType = CommandType.Text
+            };
+            dropDatabaseCommand.Parameters.AddWithValue("@databaseName", databaseName);
+            using (var command = dropDatabaseCommand)
+            {
+                if (timeout >= 0)
+                {
+                    command.CommandTimeout = timeout;
+                }
+
+                command.ExecuteNonQuery();
+            }
+            logger.WriteInformation(@"Dropped database {0}", databaseName);
+        }
+    }
+
     static bool DatabaseExists(MySqlConnection connection, string databaseName)
     {
         var sqlCommandText = string.Format