Sonarr/NzbDrone.Core/Datastore/DbFactory.cs

40 lines
1008 B
C#
Raw Normal View History

2013-03-24 00:08:23 +00:00
using System;
using System.Data;
2013-03-25 03:51:32 +00:00
using Marr.Data;
using Mono.Data.Sqlite;
2013-03-24 00:08:23 +00:00
namespace NzbDrone.Core.Datastore
{
public interface IDbFactory
{
2013-03-25 03:51:32 +00:00
IDatabase Create(string dbPath = null);
2013-03-24 00:08:23 +00:00
}
public class DbFactory : IDbFactory
{
2013-03-24 04:56:59 +00:00
private const string MemoryConnectionString = "Data Source=:memory:;Version=3;New=True;";
2013-03-24 00:08:23 +00:00
2013-03-25 03:51:32 +00:00
public IDatabase Create(string dbPath = null)
2013-03-24 00:08:23 +00:00
{
2013-03-24 04:56:59 +00:00
var connectionString = MemoryConnectionString;
if (!string.IsNullOrWhiteSpace(dbPath))
2013-03-24 00:08:23 +00:00
{
2013-03-24 04:56:59 +00:00
connectionString = GetConnectionString(dbPath);
2013-03-24 00:08:23 +00:00
}
MigrationHelper.MigrateToLatest(connectionString, MigrationType.Main);
2013-03-25 03:51:32 +00:00
var dataMapper = new DataMapper(SqliteFactory.Instance, connectionString);
return new Database(dataMapper);
}
2013-03-24 00:08:23 +00:00
private string GetConnectionString(string dbPath)
{
return String.Format("Data Source={0};Version=3;", dbPath);
}
}
}