mirror of https://github.com/Radarr/Radarr
72 lines
2.1 KiB
C#
72 lines
2.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Data.SqlServerCe;
|
|
using System.IO;
|
|
using System.Reflection;
|
|
using NLog;
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
{
|
|
public class MigrationsHelper
|
|
{
|
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
|
|
|
public static readonly Dictionary<String, String> _migrated = new Dictionary<string, string>();
|
|
|
|
public static void Run(string connetionString, bool trace)
|
|
{
|
|
if (_migrated.ContainsKey(connetionString)) return;
|
|
_migrated.Add(connetionString, string.Empty);
|
|
|
|
EnsureDatabase(connetionString);
|
|
|
|
Logger.Info("Preparing run database migration");
|
|
|
|
try
|
|
{
|
|
Migrator.Migrator migrator;
|
|
if (trace)
|
|
{
|
|
migrator = new Migrator.Migrator("sqlserverce", connetionString, Assembly.GetAssembly(typeof(MigrationsHelper)), true, new MigrationLogger());
|
|
}
|
|
else
|
|
{
|
|
migrator = new Migrator.Migrator("sqlserverce", connetionString, Assembly.GetAssembly(typeof(MigrationsHelper)));
|
|
}
|
|
|
|
|
|
|
|
migrator.MigrateToLastVersion();
|
|
|
|
//ForceSubSonicMigration(Connection.CreateSimpleRepository(connetionString));
|
|
|
|
Logger.Info("Database migration completed");
|
|
|
|
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.FatalException("An error has occured while migrating database", e);
|
|
}
|
|
}
|
|
|
|
private static void EnsureDatabase(string constr)
|
|
{
|
|
var connection = new SqlCeConnection(constr);
|
|
if (!File.Exists(connection.Database))
|
|
{
|
|
var engine = new SqlCeEngine(constr);
|
|
engine.CreateDatabase();
|
|
}
|
|
}
|
|
|
|
public static string GetIndexName(string tableName, params string[] columns)
|
|
{
|
|
return String.Format("IX_{0}_{1}", tableName, String.Join("_", columns));
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
} |