mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-12-25 02:30:20 +02:00
85 lines
2.5 KiB
C#
85 lines
2.5 KiB
C#
using System;
|
|
using System.Data.SQLite;
|
|
using Marr.Data;
|
|
using Marr.Data.Reflection;
|
|
using NzbDrone.Common.Composition;
|
|
using NzbDrone.Core.Datastore.Migration.Framework;
|
|
using NzbDrone.Core.Instrumentation;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
{
|
|
public interface IDbFactory
|
|
{
|
|
IDatabase Create(MigrationType migrationType = MigrationType.Main);
|
|
}
|
|
|
|
public class DbFactory : IDbFactory
|
|
{
|
|
private readonly IMigrationController _migrationController;
|
|
private readonly IConnectionStringFactory _connectionStringFactory;
|
|
|
|
static DbFactory()
|
|
{
|
|
MapRepository.Instance.ReflectionStrategy = new SimpleReflectionStrategy();
|
|
TableMapping.Map();
|
|
}
|
|
|
|
public static void RegisterDatabase(IContainer container)
|
|
{
|
|
var mainDb = container.Resolve<IDbFactory>().Create();
|
|
|
|
container.Register(mainDb);
|
|
|
|
var logDb = container.Resolve<IDbFactory>().Create(MigrationType.Log);
|
|
|
|
container.Register<ILogRepository>(c => new LogRepository(logDb, c.Resolve<IEventAggregator>()));
|
|
}
|
|
|
|
public DbFactory(IMigrationController migrationController, IConnectionStringFactory connectionStringFactory)
|
|
{
|
|
_migrationController = migrationController;
|
|
_connectionStringFactory = connectionStringFactory;
|
|
}
|
|
|
|
public IDatabase Create(MigrationType migrationType = MigrationType.Main)
|
|
{
|
|
string connectionString;
|
|
|
|
|
|
switch (migrationType)
|
|
{
|
|
case MigrationType.Main:
|
|
{
|
|
connectionString = _connectionStringFactory.MainDbConnectionString;
|
|
break;
|
|
}
|
|
case MigrationType.Log:
|
|
{
|
|
connectionString = _connectionStringFactory.LogDbConnectionString;
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
throw new ArgumentException("Invalid MigrationType");
|
|
}
|
|
}
|
|
|
|
_migrationController.MigrateToLatest(connectionString, migrationType);
|
|
|
|
var db = new Database(() =>
|
|
{
|
|
var dataMapper = new DataMapper(SQLiteFactory.Instance, connectionString)
|
|
{
|
|
SqlMode = SqlModes.Text,
|
|
};
|
|
|
|
return dataMapper;
|
|
});
|
|
|
|
return db;
|
|
}
|
|
}
|
|
}
|