private static async Task SeedCadmusDatabaseAsync(
IServiceProvider serviceProvider,
IConfiguration config,
ILogger logger)
{
// build connection string
string connString = config.GetConnectionString("Default");
string databaseName = config["DatabaseNames:Data"];
if (string.IsNullOrEmpty(databaseName))
{
return;
}
connString = string.Format(connString, databaseName);
// nope if database exists
IDatabaseManager manager = new MongoDatabaseManager();
if (manager.DatabaseExists(connString))
{
return;
}
// load seed profile (nope if no profile)
string profileSource = config["Seed:ProfileSource"];
if (string.IsNullOrEmpty(profileSource))
{
return;
}
Console.WriteLine($"Loading seed profile from {profileSource}...");
logger.LogInformation($"Loading seed profile from {profileSource}...");
ResourceLoaderService loaderService =
new ResourceLoaderService(serviceProvider);
Stream stream = await loaderService.LoadAsync(profileSource);
if (stream == null)
{
Console.WriteLine("Error: seed profile could not be loaded");
return;
}
// deserialize the profile
string profileContent;
using (StreamReader reader = new StreamReader(stream, Encoding.UTF8))
{
profileContent = reader.ReadToEnd();
}
IDataProfileSerializer serializer = new JsonDataProfileSerializer();
DataProfile profile = serializer.Read(profileContent);
// issue warning on invalid profile
Console.WriteLine("Validating profile...");
string error = profile.Validate();
if (error != null)
{
logger.LogWarning(error);
}
// create database
Console.WriteLine("Creating database...");
logger.LogInformation($"Creating database {connString}...");
manager.CreateDatabase(connString, profile);
Console.WriteLine("Database created.");
logger.LogInformation("Database created.");
// get the required services
Console.WriteLine("Creating seeders factory...");
Serilog.Log.Information("Creating seeders factory...");
IPartSeederFactoryProvider seederService =
serviceProvider.GetService <IPartSeederFactoryProvider>();
PartSeederFactory factory = seederService.GetFactory(
loaderService.ResolvePath(profileSource));
Console.WriteLine("Creating repository...");
Serilog.Log.Information("Creating repository...");
IRepositoryProvider repositoryProvider =
serviceProvider.GetService <IRepositoryProvider>();
ICadmusRepository repository =
repositoryProvider.CreateRepository(databaseName);
// get seed count
int count = 0;
string configCount = config["Seed:ItemCount"];
if (configCount != null && int.TryParse(configCount, out int n))
{
count = n;
}
// if required, seed data
if (count > 0)
{
Console.WriteLine($"Seeding {count} items...");
CadmusSeeder seeder = new CadmusSeeder(factory);
foreach (IItem item in seeder.GetItems(count))
{
Console.WriteLine($"{item}: {item.Parts.Count} parts");
repository.AddItem(item, true);
foreach (IPart part in item.Parts)
{
repository.AddPart(part, true);
}
}
Console.WriteLine("Seeding completed.");
}
// import data if required
IList <string> sources = factory.GetImports();
if (sources?.Count > 0)
{
PartImporter importer = new PartImporter(repository);
foreach (string source in sources)
{
foreach (string resolved in SourceRangeResolver.Resolve(source))
{
Console.WriteLine($"Importing from {resolved}...");
using Stream jsonStream =
await loaderService.LoadAsync(resolved);
importer.Import(jsonStream);
}
}
}
}
public Task Run()
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("SEED DATABASE\n");
Console.ResetColor();
Console.WriteLine($"Database: {_database}\n" +
$"Profile file: {_profilePath}\n" +
$"Count: {_count}\n" +
$"Dry run: {_dry}\n" +
$"History: {_history}\n");
Serilog.Log.Information("SEED DATABASE: " +
$"Database: {_database}, " +
$"Profile file: {_profilePath}, " +
$"Count: {_count}, " +
$"Dry: {_dry}, " +
$"History: {_history}");
if (!_dry)
{
// create database if not exists
string connection = string.Format(CultureInfo.InvariantCulture,
_config.GetConnectionString("Mongo"),
_database);
IDatabaseManager manager = new MongoDatabaseManager();
string profileContent = LoadProfile(_profilePath);
IDataProfileSerializer serializer = new JsonDataProfileSerializer();
DataProfile profile = serializer.Read(profileContent);
if (!manager.DatabaseExists(connection))
{
Console.WriteLine("Creating database...");
Serilog.Log.Information($"Creating database {_database}...");
manager.CreateDatabase(connection, profile);
Console.WriteLine("Database created.");
Serilog.Log.Information("Database created.");
}
}
Console.WriteLine("Creating repository...");
Serilog.Log.Information("Creating repository...");
ICadmusRepository repository = _dry
? null : _repositoryService.CreateRepository(_database);
Console.WriteLine("Seeding items");
PartSeederFactory factory = _seederService.GetFactory(_profilePath);
CadmusSeeder seeder = new CadmusSeeder(factory);
foreach (IItem item in seeder.GetItems(_count))
{
Console.WriteLine($"{item}: {item.Parts.Count} parts");
if (!_dry)
{
repository.AddItem(item, _history);
foreach (IPart part in item.Parts)
{
repository.AddPart(part, _history);
}
}
}
Console.WriteLine("Completed.");
return(Task.CompletedTask);
}
