Sonarr/NzbDrone.Core/Datastore/DbFactory.cs

86 lines
2.5 KiB
C#
Raw Normal View History

2013-03-24 00:08:23 +00:00
using System;
using System.Data.SQLite;
2013-03-25 03:51:32 +00:00
using Marr.Data;
using Marr.Data.Reflection;
2013-06-28 01:03:04 +00:00
using NzbDrone.Common.Composition;
using NzbDrone.Common.Messaging;
2013-03-25 06:13:53 +00:00
using NzbDrone.Core.Datastore.Migration.Framework;
2013-06-28 01:03:04 +00:00
using NzbDrone.Core.Instrumentation;
2013-03-25 03:51:32 +00:00
2013-03-24 00:08:23 +00:00
namespace NzbDrone.Core.Datastore
{
public interface IDbFactory
{
2013-06-28 01:03:04 +00:00
IDatabase Create(MigrationType migrationType = MigrationType.Main);
2013-03-24 00:08:23 +00:00
}
2013-07-05 03:56:27 +00:00
2013-03-24 00:08:23 +00:00
public class DbFactory : IDbFactory
{
2013-03-25 06:13:53 +00:00
private readonly IMigrationController _migrationController;
2013-07-05 03:56:27 +00:00
private readonly IConnectionStringFactory _connectionStringFactory;
2013-03-24 00:08:23 +00:00
static DbFactory()
2013-03-24 00:08:23 +00:00
{
MapRepository.Instance.ReflectionStrategy = new SimpleReflectionStrategy();
2013-03-25 06:13:53 +00:00
TableMapping.Map();
}
2013-06-28 01:03:04 +00:00
public static void RegisterDatabase(IContainer container)
{
container.Register(c => c.Resolve<IDbFactory>().Create());
container.Register<ILogRepository>(c =>
{
var db = c.Resolve<IDbFactory>().Create(MigrationType.Log);
return new LogRepository(db, c.Resolve<IMessageAggregator>());
});
}
2013-07-05 03:56:27 +00:00
public DbFactory(IMigrationController migrationController, IConnectionStringFactory connectionStringFactory)
{
2013-03-25 06:13:53 +00:00
_migrationController = migrationController;
2013-07-05 03:56:27 +00:00
_connectionStringFactory = connectionStringFactory;
2013-03-25 06:13:53 +00:00
}
2013-03-24 04:56:59 +00:00
2013-06-28 01:03:04 +00:00
public IDatabase Create(MigrationType migrationType = MigrationType.Main)
2013-03-25 06:13:53 +00:00
{
2013-07-05 03:56:27 +00:00
string connectionString;
2013-06-28 01:03:04 +00:00
switch (migrationType)
{
case MigrationType.Main:
{
2013-07-05 03:56:27 +00:00
connectionString = _connectionStringFactory.MainDbConnectionString;
2013-06-28 01:03:04 +00:00
break;
}
case MigrationType.Log:
{
2013-07-05 03:56:27 +00:00
connectionString = _connectionStringFactory.LogDbConnectionString;
2013-06-28 01:03:04 +00:00
break;
}
default:
{
throw new ArgumentException("Invalid MigrationType");
}
}
2013-03-24 00:08:23 +00:00
2013-03-25 06:13:53 +00:00
_migrationController.MigrateToLatest(connectionString, migrationType);
2013-05-11 20:06:57 +00:00
return new Database(() =>
{
var dataMapper = new DataMapper(SQLiteFactory.Instance, connectionString)
{
SqlMode = SqlModes.Text,
};
return dataMapper;
});
2013-03-25 03:51:32 +00:00
}
2013-03-24 00:08:23 +00:00
}
}