Sonarr/NzbDrone.Core/Datastore/Connection.cs

79 lines
2.1 KiB
C#
Raw Normal View History

2011-05-23 23:29:14 +00:00
using System;
using System.Data;
using System.Data.Common;
2011-06-23 06:56:17 +00:00
using System.Data.SqlServerCe;
2011-05-23 23:29:14 +00:00
using System.IO;
2011-06-14 01:35:44 +00:00
using MvcMiniProfiler.Data;
using NzbDrone.Common;
using NzbDrone.Core.Providers;
2011-06-15 02:31:41 +00:00
using PetaPoco;
2011-05-23 23:29:14 +00:00
namespace NzbDrone.Core.Datastore
{
public class Connection
2011-05-23 23:29:14 +00:00
{
private readonly PathProvider _pathProvider;
2011-05-23 23:29:14 +00:00
public Connection(PathProvider pathProvider)
2011-05-23 23:29:14 +00:00
{
_pathProvider = pathProvider;
2011-05-23 23:29:14 +00:00
}
static Connection()
2011-05-23 23:29:14 +00:00
{
Database.Mapper = new CustomeMapper();
2011-05-23 23:29:14 +00:00
}
public String MainConnectionString
2011-05-23 23:29:14 +00:00
{
get
{
return GetConnectionString(_pathProvider.NzbDronoeDbFile);
2011-05-23 23:29:14 +00:00
}
}
public String LogConnectionString
2011-05-23 23:29:14 +00:00
{
get
{
return GetConnectionString(_pathProvider.LogDbFile);
2011-05-23 23:29:14 +00:00
}
}
public static string GetConnectionString(string path)
{
//return String.Format("Data Source={0};Version=3;Cache Size=30000;Pooling=true;Default Timeout=2", path);
return String.Format("Data Source={0}", path);
}
public IDatabase GetMainPetaPocoDb(Boolean profiled = true)
{
return GetPetaPocoDb(MainConnectionString, profiled);
}
public IDatabase GetLogPetaPocoDb(Boolean profiled = true)
{
return GetPetaPocoDb(LogConnectionString, profiled);
}
public static IDatabase GetPetaPocoDb(string connectionString, Boolean profiled = true)
2011-06-15 02:31:41 +00:00
{
MigrationsHelper.Run(connectionString, true);
2011-06-23 06:56:17 +00:00
var factory = new PetaDbProviderFactory
{
IsProfiled = profiled
};
var db = new Database(connectionString, factory, Database.DBType.SqlServerCE)
{
KeepConnectionAlive = true,
ForceDateTimesToUtc = false,
};
2011-06-15 02:31:41 +00:00
return db;
}
2011-05-23 23:29:14 +00:00
}
}