1
0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-16 11:37:58 +02:00
Sonarr/NzbDrone.Core/Datastore/Connection.cs

94 lines
2.8 KiB
C#
Raw Normal View History

2011-05-24 02:29:14 +03:00
using System;
2011-11-23 09:15:02 +03:00
using System.Configuration;
using System.Data.Common;
using System.Data.SqlServerCe;
using NzbDrone.Common;
using NzbDrone.Core.Instrumentation;
2011-06-15 05:31:41 +03:00
using PetaPoco;
2011-05-24 02:29:14 +03:00
namespace NzbDrone.Core.Datastore
{
public class Connection
2011-05-24 02:29:14 +03:00
{
2011-11-13 10:27:16 +03:00
private readonly EnviromentProvider _enviromentProvider;
2011-05-24 02:29:14 +03:00
2011-12-02 10:07:18 +03:00
static Connection()
2011-11-23 09:15:02 +03:00
{
2011-12-02 10:07:18 +03:00
Database.Mapper = new CustomeMapper();
2011-11-23 09:15:02 +03:00
2011-12-02 10:07:18 +03: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 09:15:02 +03:00
}
2011-11-13 10:27:16 +03:00
public Connection(EnviromentProvider enviromentProvider)
2011-05-24 02:29:14 +03:00
{
2011-11-13 10:27:16 +03:00
_enviromentProvider = enviromentProvider;
2011-05-24 02:29:14 +03:00
}
2011-12-02 10:07:18 +03:00
public String MainConnectionString
2011-05-24 02:29:14 +03:00
{
get
{
2011-11-13 10:27:16 +03:00
return GetConnectionString(_enviromentProvider.GetNzbDronoeDbFile());
2011-05-24 02:29:14 +03:00
}
}
public String LogConnectionString
2011-05-24 02:29:14 +03:00
{
get
{
return GetConnectionString(_enviromentProvider.GetLogDbFileDbFile());
2011-05-24 02:29:14 +03:00
}
}
public static string GetConnectionString(string path)
{
return String.Format("Data Source={0}", path);
}
2011-11-13 10:27:16 +03:00
public IDatabase GetMainPetaPocoDb(Boolean profiled = true)
{
return GetPetaPocoDb(MainConnectionString, profiled);
}
public IDatabase GetLogPetaPocoDb(Boolean profiled = true)
{
return GetPetaPocoDb(LogConnectionString, profiled);
}
public LogDbContext GetLogEfContext()
{
return GetLogDbContext(LogConnectionString);
}
public static IDatabase GetPetaPocoDb(string connectionString, Boolean profiled = true)
2011-06-15 05:31:41 +03:00
{
MigrationsHelper.Run(connectionString, true);
2011-06-23 09:56:17 +03:00
var factory = new DbProviderFactory
{
IsProfiled = profiled
};
var db = new Database(connectionString, factory, Database.DBType.SqlServerCE)
{
KeepConnectionAlive = true,
ForceDateTimesToUtc = false,
};
2011-06-15 05:31:41 +03:00
return db;
}
public static LogDbContext GetLogDbContext(string connectionString)
{
MigrationsHelper.Run(connectionString, true);
DbConnection connection = new SqlCeConnection(connectionString);
return new LogDbContext(connection);
}
2011-05-24 02:29:14 +03:00
}
}