nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-07-02 16:42:50 -05:00
Code Activity

Add rerunable options and folder

Browse Source
This commit is contained in:
Michal Checinski
2023-08-28 16:41:01 +02:00
parent d82025b236
commit 5ecffdad74
1 changed files with 45 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
util/MsSqlMigratorUtility/Program.cs
View File

@ -3,6 +3,8 @@ using Microsoft.Extensions.Logging;
internal class Program internal class Program
{ {
private IDictionary<string, string> Parameters { get; set; }
private static int Main(string[] args) private static int Main(string[] args)
{ {
if (args.Length == 0) if (args.Length == 0)
@ -23,12 +25,27 @@ internal class Program
var verbose = false; var verbose = false;
if (args.Length == 2 && (args[1] == "--verbose" || args[1] == "-v")) if (Parameters.ContainsKey("--verbose") || Parameters.ContainsKey("-v"))
{ {
verbose = true; verbose = true;
} }
var success = MigrateDatabase(databaseConnectionString, verbose); var rerunable = false;
if (Parameters.ContainsKey("--rerunable") || Parameters.ContainsKey("-r"))
{
rerunable = true;
}
var folderName = "";
if (Parameters.ContainsKey("--folder") || Parameters.ContainsKey("-f"))
{
folderName = Parameters["--folder"] ?? Parameters["-f"];
}
var success = MigrateDatabase(databaseConnectionString, verbose, rerunable, folderName);
if (!success) if (!success)
{ {
@ -42,14 +59,24 @@ internal class Program
{ {
Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string>"); Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string>");
Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string> -v|--verbose (for verbose output of migrator logs)"); Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string> -v|--verbose (for verbose output of migrator logs)");
Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string> -r|--rerunable (for marking scripts as rerunable) -f|--folder <folder-name-in-migrator-project> (for specifying folder name of scripts)");
Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string> -v|--verbose (for verbose output of migrator logs) -r|--rerunable (for marking scripts as rerunable) -f|--folder <folder-name-in-migrator-project> (for specifying folder name of scripts)");
} }
private static bool MigrateDatabase(string databaseConnectionString, bool verbose = false, int attempt = 1) private static bool MigrateDatabase(string databaseConnectionString, bool verbose = false, bool rerunable = false, string folderName = "")
{ {
var logger = CreateLogger(verbose); var logger = CreateLogger(verbose);
var migrator = new DbMigrator(databaseConnectionString, logger); var migrator = new DbMigrator(databaseConnectionString, logger);
var success = migrator.MigrateMsSqlDatabaseWithRetries(verbose); bool success = false;
if (string.IsNullOrWhiteSpace(folderName))
{
success = migrator.MigrateMsSqlDatabaseWithRetries(verbose, rerunable, folderName);
}
else
{
success = migrator.MigrateMsSqlDatabaseWithRetries(verbose, rerunable);
}
return success; return success;
} }
@ -75,4 +102,18 @@ internal class Program
var logger = loggerFactory.CreateLogger<DbMigrator>(); var logger = loggerFactory.CreateLogger<DbMigrator>();
return logger; return logger;
} }
private static void ParseParameters(string[] args)
{
Parameters = new Dictionary<string, string>();
for (var i = 0; i < args.Length; i = i + 2)
{
if (!args[i].StartsWith("-"))
{
continue;
}
Parameters.Add(args[i].Substring(1), args[i + 1]);
}
}
} }