Added function to create postgres database with owner

Author: Aashik Nagadikeri Harish

src/Tests/ApprovalFiles/NoPublicApiChanges.Run.approved.cs

@@ -1,4 +1,6 @@
-[assembly: System.CLSCompliantAttribute(false)]
+using DbUp.Postgresql;
+
+[assembly: System.CLSCompliantAttribute(false)]
 [assembly: System.Runtime.InteropServices.ComVisibleAttribute(false)]
 [assembly: System.Runtime.InteropServices.GuidAttribute("5ddc04cc-0bd3-421e-9ae4-9fd0e4f4ef04")]
 
@@ -16,6 +18,7 @@ public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatab
     public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString) { }
     public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, System.Security.Cryptography.X509Certificates.X509Certificate2 certificate) { }
     public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Postgresql.PostgresqlConnectionOptions connectionOptions) { }
+    public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Postgresql.PostgresqlConnectionOptions connectionOptions, string owner) { }
     public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger) { }
     public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, System.Security.Cryptography.X509Certificates.X509Certificate2 certificate) { }
     public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, DbUp.Postgresql.PostgresqlConnectionOptions connectionOptions) { }

src/dbup-postgresql/PostgresqlExtensions.cs

@@ -16,7 +16,7 @@
 /// </summary>
 public static class PostgresqlExtensions
 {
-    private static readonly string pattern= @"(?i)Search\s?Path=([^;]+)";
+    private static readonly string pattern = @"(?i)Search\s?Path=([^;]+)";
     /// <summary>
     /// Creates an upgrader for PostgreSQL databases.
     /// </summary>
@@ -165,21 +165,55 @@ public static void PostgresqlDatabase(this SupportedDatabasesForEnsureDatabase s
         PostgresqlDatabase(supported, connectionString, logger, new PostgresqlConnectionOptions());
     }
 
+    /// <summary>
+    /// Ensures that the database specified in the connection string exists.
+    /// </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="certificate">Certificate for securing connection.</param>
+    /// <returns></returns>
     public static void PostgresqlDatabase(this SupportedDatabasesForEnsureDatabase supported, string connectionString, IUpgradeLog logger, X509Certificate2 certificate)
     {
         var options = new PostgresqlConnectionOptions
-        { 
+        {
             ClientCertificate = certificate
         };
         PostgresqlDatabase(supported, connectionString, logger, options);
     }
 
+    /// <summary>
+    /// Ensures that the database specified in the connection string exists.
+    /// </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="connectionOptions">Connection SSL to customize SSL behaviour</param>
     public static void PostgresqlDatabase(
-        this SupportedDatabasesForEnsureDatabase supported, 
-        string connectionString, 
-        IUpgradeLog logger, 
+        this SupportedDatabasesForEnsureDatabase supported,
+        string connectionString,
+        IUpgradeLog logger,
         PostgresqlConnectionOptions connectionOptions
     )
+    {
+        PostgresqlDatabase(supported, connectionString, logger, connectionOptions, null);
+    }
+
+    /// <summary>
+    /// Ensures that the database specified in the connection string exists, assigning an owner at creation time.
+    /// </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="connectionOptions">Connection SSL to customize SSL behaviour</param>
+    /// <param name="owner">Role to own the new database during creation (adds 'WITH OWNER = "role"').</param>
+    public static void PostgresqlDatabase(
+        this SupportedDatabasesForEnsureDatabase supported,
+        string connectionString,
+        IUpgradeLog logger,
+        PostgresqlConnectionOptions connectionOptions,
+        string owner
+    )
     {
         if (supported == null) throw new ArgumentNullException("supported");
 
@@ -191,14 +225,16 @@ PostgresqlConnectionOptions connectionOptions
         if (logger == null) throw new ArgumentNullException("logger");
 
         var masterConnectionStringBuilder = new NpgsqlConnectionStringBuilder(connectionString);
-        
+
         var databaseName = masterConnectionStringBuilder.Database;
 
         if (string.IsNullOrEmpty(databaseName) || databaseName.Trim() == string.Empty)
         {
             throw new InvalidOperationException("The connection string does not specify a database name.");
         }
 
+        owner = string.IsNullOrWhiteSpace(owner) ? masterConnectionStringBuilder.Username : owner;
+
         masterConnectionStringBuilder.Database = connectionOptions.MasterDatabaseName;
 
         var logMasterConnectionStringBuilder = new NpgsqlConnectionStringBuilder(masterConnectionStringBuilder.ConnectionString);
@@ -218,9 +254,9 @@ PostgresqlConnectionOptions connectionOptions
 
         // check to see if the database already exists..
         using (var command = new NpgsqlCommand(sqlCommandText, connection)
-               {
-                   CommandType = CommandType.Text
-               })
+        {
+            CommandType = CommandType.Text
+        })
         {
             var results = Convert.ToInt32(command.ExecuteScalar());
 
@@ -231,13 +267,29 @@ PostgresqlConnectionOptions connectionOptions
             }
         }
 
-        sqlCommandText = $"create database \"{databaseName}\";";
+        sqlCommandText = $"select 1 from pg_roles where rolname = \'{owner}\' limit 1;";
+        // check to see if the owner exists..
+        using (var command = new NpgsqlCommand(sqlCommandText, connection)
+        {
+            CommandType = CommandType.Text
+        })
+        {
+            var results = Convert.ToInt32(command.ExecuteScalar());
+
+            // if the owner role does not exist, we throw an exception.
+            if (results == 0)
+            {
+                throw new InvalidOperationException($"PostgreSQL role '{owner}' does not exist.");
+            }
+        }
+
+        sqlCommandText = $"create database \"{databaseName}\" with owner = \"{owner}\";";
 
         // Create the database...
         using (var command = new NpgsqlCommand(sqlCommandText, connection)
-               {
-                   CommandType = CommandType.Text
-               })
+        {
+            CommandType = CommandType.Text
+        })
         {
             command.ExecuteNonQuery();
         }
@@ -319,7 +371,7 @@ PostgresqlConnectionOptions connectionOptions
 
         var masterConnectionStringBuilder = new NpgsqlConnectionStringBuilder(connectionString);
 
-        var databaseName = masterConnectionStringBuilder.Database;       
+        var databaseName = masterConnectionStringBuilder.Database;
 
         if (string.IsNullOrEmpty(databaseName) || databaseName.Trim() == string.Empty)
         {
@@ -351,9 +403,9 @@ PostgresqlConnectionOptions connectionOptions
 
         // check to see if the database already exists..
         using (var command = new NpgsqlCommand(sqlCommandText, connection)
-               {
-                   CommandType = CommandType.Text
-               })
+        {
+            CommandType = CommandType.Text
+        })
         {
             var results = Convert.ToInt32(command.ExecuteScalar());
 
@@ -368,9 +420,9 @@ PostgresqlConnectionOptions connectionOptions
         // prevent new connections to the database
         sqlCommandText = $"alter database \"{databaseName}\" with ALLOW_CONNECTIONS false;";
         using (var command = new NpgsqlCommand(sqlCommandText, connection)
-               {
-                   CommandType = CommandType.Text
-               })
+        {
+            CommandType = CommandType.Text
+        })
         {
             command.ExecuteNonQuery();
         }
@@ -380,9 +432,9 @@ PostgresqlConnectionOptions connectionOptions
         // terminate all existing connections to the database
         sqlCommandText = $"select pg_terminate_backend(pg_stat_activity.pid) from pg_stat_activity where pg_stat_activity.datname = \'{databaseName}\';";
         using (var command = new NpgsqlCommand(sqlCommandText, connection)
-               {
-                   CommandType = CommandType.Text
-               })
+        {
+            CommandType = CommandType.Text
+        })
         {
             command.ExecuteNonQuery();
         }
@@ -393,9 +445,9 @@ PostgresqlConnectionOptions connectionOptions
 
         // drop the database
         using (var command = new NpgsqlCommand(sqlCommandText, connection)
-               {
-                   CommandType = CommandType.Text
-               })
+        {
+            CommandType = CommandType.Text
+        })
         {
             command.ExecuteNonQuery();
         }