private async Task CreateDatabaseIfNotExists(string databaseName, CancellationToken cancellationToken)
{
using (var connection = await _sqlConnectionFactory.GetSqlConnectionAsync(cancellationToken: cancellationToken))
{
await connection.TryOpenAsync(cancellationToken);
bool doesDatabaseExist = await SchemaInitializer.DoesDatabaseExistAsync(connection, databaseName, cancellationToken);
if (!doesDatabaseExist)
{
_logger.LogInformation("The database does not exists.");
bool created = await SchemaInitializer.CreateDatabaseAsync(connection, databaseName, cancellationToken);
if (created)
{
_logger.LogInformation("The database is created.");
}
else
{
throw new SchemaManagerException(Resources.InsufficientDatabasePermissionsMessage);
}
}
}
}
private async Task InitializeAsync(CancellationToken cancellationToken)
{
var configuredConnectionBuilder = new SqlConnectionStringBuilder(_sqlServerDataStoreConfiguration.ConnectionString);
string databaseName = configuredConnectionBuilder.InitialCatalog;
SchemaInitializer.ValidateDatabaseName(databaseName);
SqlConnectionStringBuilder connectionBuilder = new SqlConnectionStringBuilder(_sqlServerDataStoreConfiguration.ConnectionString)
{
InitialCatalog = string.Empty
};
using (var connection = new SqlConnection(connectionBuilder.ToString()))
{
bool doesDatabaseExist = await SchemaInitializer.DoesDatabaseExistAsync(connection, databaseName, cancellationToken);
// database creation is allowed
if (!doesDatabaseExist)
{
Console.WriteLine("The database does not exists.");
bool created = await SchemaInitializer.CreateDatabaseAsync(connection, databaseName, cancellationToken);
if (created)
{
Console.WriteLine("The database is created.");
}
else
{
throw new SchemaManagerException(Resources.InsufficientDatabasePermissionsMessage);
}
connection.Close();
}
}
bool canInitialize = false;
// now switch to the target database
using (var connection = await _sqlConnectionFactory.GetSqlConnectionAsync(cancellationToken: cancellationToken))
{
canInitialize = await SchemaInitializer.CheckDatabasePermissionsAsync(connection, cancellationToken);
}
if (!canInitialize)
{
throw new SchemaManagerException(Resources.InsufficientTablesPermissionsMessage);
}
}
public async Task DatabaseExists_DoesDatabaseExistAsync_ReturnsTrue()
{
const string dbName = "willexist";
try
{
Assert.False(await SchemaInitializer.DoesDatabaseExistAsync(Connection, dbName, CancellationToken.None));
Assert.True(await SchemaInitializer.CreateDatabaseAsync(Connection, dbName, CancellationToken.None));
Assert.True(await SchemaInitializer.DoesDatabaseExistAsync(Connection, dbName, CancellationToken.None));
}
finally
{
await DeleteDatabaseAsync(dbName);
}
}
public async Task DatabaseDoesNotExist_DoesDatabaseExistAsync_ReturnsFalse()
{
Assert.False(await SchemaInitializer.DoesDatabaseExistAsync(Connection, "doesnotexist", CancellationToken.None));
}