Skip to content

Added a function to allow creation of database with owner #37

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 9 commits into
base: main
Choose a base branch
from
Open

Added a function to allow creation of database with owner #37

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
c824d96
Added function to create postgres database with owner
Nov 5, 2025
a686ef3
Remonved using statement in NoPublicApiChanges.Run.approved.cs
Nov 5, 2025
3375b7e
Fixed test
Nov 5, 2025
31d7eac
Merge branch 'main' into main
droyad Nov 6, 2025
8a002b2
Apply suggestion from @droyad
droyad Nov 6, 2025
daea170
Update src/dbup-postgresql/PostgresqlExtensions.cs
droyad Nov 6, 2025
a9f9ae3
Addressed PR comments around SQLi and using if/else for creating a da…
Nov 6, 2025
7e17862
Flipped boolean logic around
Nov 6, 2025
0a58605
formatting
Nov 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/Tests/ApprovalFiles/NoPublicApiChanges.Run.approved.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForDropDatabas
public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForDropDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, System.Security.Cryptography.X509Certificates.X509Certificate2 certificate) { }
public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, DbUp.Postgresql.PostgresqlConnectionOptions connectionOptions) { }
public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, System.Security.Cryptography.X509Certificates.X509Certificate2 certificate) { }
public static void PostgresqlDatabase(this DbUp.SupportedDatabasesForEnsureDatabase supported, string connectionString, DbUp.Engine.Output.IUpgradeLog logger, DbUp.Postgresql.PostgresqlConnectionOptions connectionOptions, string owner) { }
}
namespace DbUp.Postgresql
{
Expand Down
121 changes: 89 additions & 32 deletions src/dbup-postgresql/PostgresqlExtensions.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
using System;
using System;
using System.Data;
using System.Security.Cryptography.X509Certificates;
using System.Text.RegularExpressions;
Expand All @@ -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>
Expand Down Expand Up @@ -175,7 +175,7 @@ public static void PostgresqlDatabase(this SupportedDatabasesForEnsureDatabase s
public static void PostgresqlDatabase(this SupportedDatabasesForEnsureDatabase supported, string connectionString, IUpgradeLog logger, X509Certificate2 certificate)
{
var options = new PostgresqlConnectionOptions
{
{
ClientCertificate = certificate
};
PostgresqlDatabase(supported, connectionString, logger, options);
Expand All @@ -189,11 +189,30 @@ public static void PostgresqlDatabase(this SupportedDatabasesForEnsureDatabase s
/// <param name="logger">The <see cref="DbUp.Engine.Output.IUpgradeLog"/> used to record actions.</param>
/// <param name="connectionOptions">Connection options to set SSL parameters</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 options to set SSL parameters</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");

Expand All @@ -205,7 +224,7 @@ 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)
Expand All @@ -232,9 +251,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());

Expand All @@ -245,18 +264,56 @@ PostgresqlConnectionOptions connectionOptions
}
}

sqlCommandText = $"create database \"{databaseName}\";";

// Create the database...
using (var command = new NpgsqlCommand(sqlCommandText, connection)
{
CommandType = CommandType.Text
})
if (string.IsNullOrEmpty(owner))
{
command.ExecuteNonQuery();
sqlCommandText = $"create database \"{databaseName}\";";

// Create the database...
using (var command = new NpgsqlCommand(sqlCommandText, connection)
{
CommandType = CommandType.Text
})
{
command.ExecuteNonQuery();
}

logger.LogInformation(@"Created database {0}", databaseName);
}
else
{
sqlCommandText = "select exists (select 1 from pg_roles where rolname = @owner);";
// check to see if the owner exists..
using (var command = new NpgsqlCommand(sqlCommandText, connection)
{
CommandType = CommandType.Text
})
{
command.Parameters.AddWithValue("@owner", owner);

var roleExists = (bool)command.ExecuteScalar();
// if the owner role does not exist, we throw an exception.
if (!roleExists)
{
throw new InvalidOperationException($"PostgreSQL role '{owner}' does not exist.");
}
}

logger.LogInformation(@"Created database {0}", databaseName);
using var formattedSql = new NpgsqlCommand("select format('create database %I with owner = %I', @databaseName, @owner);", connection);
formattedSql.Parameters.AddWithValue("databaseName", databaseName);
formattedSql.Parameters.AddWithValue("owner", owner);
sqlCommandText = (string)formattedSql.ExecuteScalar();

// Create the database..
using (var command = new NpgsqlCommand(sqlCommandText, connection)
{
CommandType = CommandType.Text,
})
{
command.ExecuteNonQuery();
}

logger.LogInformation(@"Created database {0} with owner {1}", databaseName, owner);
}
}

/// <summary>
Expand Down Expand Up @@ -347,7 +404,7 @@ PostgresqlConnectionOptions connectionOptions

var masterConnectionStringBuilder = new NpgsqlConnectionStringBuilder(connectionString);

var databaseName = masterConnectionStringBuilder.Database;
var databaseName = masterConnectionStringBuilder.Database;

if (string.IsNullOrEmpty(databaseName) || databaseName.Trim() == string.Empty)
{
Expand Down Expand Up @@ -379,9 +436,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());

Expand All @@ -396,9 +453,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();
}
Expand All @@ -408,9 +465,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();
}
Expand All @@ -421,9 +478,9 @@ PostgresqlConnectionOptions connectionOptions

// drop the database
using (var command = new NpgsqlCommand(sqlCommandText, connection)
{
CommandType = CommandType.Text
})
{
CommandType = CommandType.Text
})
{
command.ExecuteNonQuery();
}
Expand Down