2011-05-24 01:29:14 +02:00
|
|
|
|
using System;
|
2011-11-23 07:15:02 +01:00
|
|
|
|
using System.Configuration;
|
2011-11-23 06:58:26 +01:00
|
|
|
|
using System.Data.Common;
|
|
|
|
|
using System.Data.SqlServerCe;
|
2011-10-29 06:54:33 +02:00
|
|
|
|
using NzbDrone.Common;
|
2011-11-23 06:58:26 +01:00
|
|
|
|
using NzbDrone.Core.Instrumentation;
|
2011-06-15 04:31:41 +02:00
|
|
|
|
using PetaPoco;
|
2011-05-24 01:29:14 +02:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
|
|
|
{
|
2011-11-03 06:04:14 +01:00
|
|
|
|
public class Connection
|
2011-05-24 01:29:14 +02:00
|
|
|
|
{
|
2011-11-13 08:27:16 +01:00
|
|
|
|
private readonly EnviromentProvider _enviromentProvider;
|
2011-05-24 01:29:14 +02:00
|
|
|
|
|
2011-12-02 08:07:18 +01:00
|
|
|
|
static Connection()
|
2011-11-23 07:15:02 +01:00
|
|
|
|
{
|
2011-12-02 08:07:18 +01:00
|
|
|
|
Database.Mapper = new CustomeMapper();
|
2011-11-23 07:15:02 +01:00
|
|
|
|
|
2011-12-02 08:07:18 +01:00
|
|
|
|
var dataSet = ConfigurationManager.GetSection("system.data") as System.Data.DataSet;
|
|
|
|
|
dataSet.Tables[0].Rows.Add("Microsoft SQL Server Compact Data Provider 4.0"
|
|
|
|
|
, "System.Data.SqlServerCe.4.0"
|
|
|
|
|
, ".NET Framework Data Provider for Microsoft SQL Server Compact"
|
|
|
|
|
, "System.Data.SqlServerCe.SqlCeProviderFactory, System.Data.SqlServerCe, Version=4.0.0.0, Culture=neutral, PublicKeyToken=89845dcd8080cc91");
|
2011-11-23 07:15:02 +01:00
|
|
|
|
}
|
|
|
|
|
|
2011-11-13 08:27:16 +01:00
|
|
|
|
public Connection(EnviromentProvider enviromentProvider)
|
2011-05-24 01:29:14 +02:00
|
|
|
|
{
|
2011-11-13 08:27:16 +01:00
|
|
|
|
_enviromentProvider = enviromentProvider;
|
2011-05-24 01:29:14 +02:00
|
|
|
|
}
|
2011-12-02 08:07:18 +01:00
|
|
|
|
|
2011-11-03 06:04:14 +01:00
|
|
|
|
public String MainConnectionString
|
2011-05-24 01:29:14 +02:00
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2011-11-13 08:27:16 +01:00
|
|
|
|
return GetConnectionString(_enviromentProvider.GetNzbDronoeDbFile());
|
2011-05-24 01:29:14 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2011-11-03 06:04:14 +01:00
|
|
|
|
public String LogConnectionString
|
2011-05-24 01:29:14 +02:00
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2011-11-23 06:58:26 +01:00
|
|
|
|
return GetConnectionString(_enviromentProvider.GetLogDbFileDbFile());
|
2011-05-24 01:29:14 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2011-11-03 06:04:14 +01:00
|
|
|
|
public static string GetConnectionString(string path)
|
|
|
|
|
{
|
|
|
|
|
return String.Format("Data Source={0}", path);
|
|
|
|
|
}
|
2011-11-13 08:27:16 +01:00
|
|
|
|
|
2011-11-03 06:04:14 +01:00
|
|
|
|
public IDatabase GetMainPetaPocoDb(Boolean profiled = true)
|
|
|
|
|
{
|
|
|
|
|
return GetPetaPocoDb(MainConnectionString, profiled);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public IDatabase GetLogPetaPocoDb(Boolean profiled = true)
|
|
|
|
|
{
|
|
|
|
|
return GetPetaPocoDb(LogConnectionString, profiled);
|
|
|
|
|
}
|
2011-06-18 04:51:53 +02:00
|
|
|
|
|
2011-11-23 06:58:26 +01:00
|
|
|
|
public LogDbContext GetLogEfContext()
|
|
|
|
|
{
|
|
|
|
|
return GetLogDbContext(LogConnectionString);
|
|
|
|
|
}
|
|
|
|
|
|
2011-06-18 04:51:53 +02:00
|
|
|
|
public static IDatabase GetPetaPocoDb(string connectionString, Boolean profiled = true)
|
2011-06-15 04:31:41 +02:00
|
|
|
|
{
|
2011-06-17 08:58:50 +02:00
|
|
|
|
MigrationsHelper.Run(connectionString, true);
|
2011-06-23 08:56:17 +02:00
|
|
|
|
|
2011-11-23 06:58:26 +01:00
|
|
|
|
var factory = new DbProviderFactory
|
2011-08-28 19:43:33 +02:00
|
|
|
|
{
|
|
|
|
|
IsProfiled = profiled
|
|
|
|
|
};
|
2011-06-17 05:36:52 +02:00
|
|
|
|
|
2011-08-28 19:43:33 +02:00
|
|
|
|
var db = new Database(connectionString, factory, Database.DBType.SqlServerCE)
|
|
|
|
|
{
|
|
|
|
|
KeepConnectionAlive = true,
|
|
|
|
|
ForceDateTimesToUtc = false,
|
|
|
|
|
};
|
2011-06-15 04:31:41 +02:00
|
|
|
|
|
|
|
|
|
return db;
|
|
|
|
|
}
|
|
|
|
|
|
2011-11-23 06:58:26 +01:00
|
|
|
|
public static LogDbContext GetLogDbContext(string connectionString)
|
|
|
|
|
{
|
|
|
|
|
MigrationsHelper.Run(connectionString, true);
|
|
|
|
|
DbConnection connection = new SqlCeConnection(connectionString);
|
|
|
|
|
return new LogDbContext(connection);
|
|
|
|
|
}
|
|
|
|
|
|
2011-05-24 01:29:14 +02:00
|
|
|
|
}
|
|
|
|
|
}
|