mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 16:02:29 +01:00
7603d8e1ba
model tables are automatically created.
46 lines
1.2 KiB
C#
46 lines
1.2 KiB
C#
using System;
|
|
using System.Data;
|
|
using ServiceStack.OrmLite;
|
|
using ServiceStack.OrmLite.Sqlite;
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
{
|
|
public interface IDbFactory
|
|
{
|
|
IDbConnection Create(string dbPath = null);
|
|
}
|
|
|
|
public class DbFactory : IDbFactory
|
|
{
|
|
private const string MemoryConnectionString = "Data Source=:memory:;Version=3;New=True;";
|
|
|
|
static DbFactory()
|
|
{
|
|
OrmLiteConfig.DialectProvider = new SqliteOrmLiteDialectProvider();
|
|
}
|
|
|
|
public IDbConnection Create(string dbPath = null)
|
|
{
|
|
var connectionString = MemoryConnectionString;
|
|
|
|
if (!string.IsNullOrWhiteSpace(dbPath))
|
|
{
|
|
connectionString = GetConnectionString(dbPath);
|
|
}
|
|
|
|
OrmLiteConfig.DialectProvider = new SqliteOrmLiteDialectProvider();
|
|
var dbFactory = new OrmLiteConnectionFactory(connectionString);
|
|
var connection = dbFactory.Open();
|
|
|
|
Migration.CreateTables(connection);
|
|
|
|
return connection;
|
|
}
|
|
|
|
private string GetConnectionString(string dbPath)
|
|
{
|
|
return String.Format("Data Source={0};Version=3;", dbPath);
|
|
}
|
|
}
|
|
}
|