2013-03-25 03:51:32 +00:00
|
|
|
|
using System;
|
2019-12-18 21:56:41 +00:00
|
|
|
|
using System.Data;
|
2022-03-28 20:45:52 +00:00
|
|
|
|
using System.Text.RegularExpressions;
|
2019-12-18 21:56:41 +00:00
|
|
|
|
using Dapper;
|
2013-11-28 07:11:05 +00:00
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Common.Instrumentation;
|
2013-03-25 03:51:32 +00:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
|
|
|
{
|
|
|
|
|
public interface IDatabase
|
|
|
|
|
{
|
2019-12-18 21:56:41 +00:00
|
|
|
|
IDbConnection OpenConnection();
|
2013-11-28 07:11:05 +00:00
|
|
|
|
Version Version { get; }
|
2019-09-03 02:22:25 +00:00
|
|
|
|
int Migration { get; }
|
2022-03-28 20:45:52 +00:00
|
|
|
|
DatabaseType DatabaseType { get; }
|
2013-11-28 07:11:05 +00:00
|
|
|
|
void Vacuum();
|
2013-03-25 03:51:32 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class Database : IDatabase
|
|
|
|
|
{
|
2015-05-03 18:52:07 +00:00
|
|
|
|
private readonly string _databaseName;
|
2019-12-18 21:56:41 +00:00
|
|
|
|
private readonly Func<IDbConnection> _datamapperFactory;
|
2013-03-25 03:51:32 +00:00
|
|
|
|
|
2014-12-17 07:12:26 +00:00
|
|
|
|
private readonly Logger _logger = NzbDroneLogger.GetLogger(typeof(Database));
|
2013-11-28 07:11:05 +00:00
|
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
|
public Database(string databaseName, Func<IDbConnection> datamapperFactory)
|
2013-03-25 03:51:32 +00:00
|
|
|
|
{
|
2015-05-03 18:52:07 +00:00
|
|
|
|
_databaseName = databaseName;
|
2013-11-28 07:52:22 +00:00
|
|
|
|
_datamapperFactory = datamapperFactory;
|
2013-03-25 03:51:32 +00:00
|
|
|
|
}
|
|
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
|
public IDbConnection OpenConnection()
|
2013-05-11 20:06:57 +00:00
|
|
|
|
{
|
2013-11-28 07:52:22 +00:00
|
|
|
|
return _datamapperFactory();
|
2013-11-28 07:11:05 +00:00
|
|
|
|
}
|
|
|
|
|
|
2022-03-28 20:45:52 +00:00
|
|
|
|
public DatabaseType DatabaseType
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
|
|
|
|
using (var db = _datamapperFactory())
|
|
|
|
|
{
|
|
|
|
|
if (db.ConnectionString.Contains(".db"))
|
|
|
|
|
{
|
|
|
|
|
return DatabaseType.SQLite;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
return DatabaseType.PostgreSQL;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-11-28 07:11:05 +00:00
|
|
|
|
public Version Version
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2018-12-02 17:05:58 +00:00
|
|
|
|
using (var db = _datamapperFactory())
|
|
|
|
|
{
|
2022-03-28 20:45:52 +00:00
|
|
|
|
string version;
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
version = db.QueryFirstOrDefault<string>("SHOW server_version");
|
|
|
|
|
|
2022-11-20 18:27:45 +00:00
|
|
|
|
// Postgres can return extra info about operating system on version call, ignore this
|
2022-03-28 20:45:52 +00:00
|
|
|
|
version = Regex.Replace(version, @"\(.*?\)", "");
|
|
|
|
|
}
|
|
|
|
|
catch
|
|
|
|
|
{
|
|
|
|
|
version = db.QueryFirstOrDefault<string>("SELECT sqlite_version()");
|
|
|
|
|
}
|
|
|
|
|
|
2018-12-02 17:05:58 +00:00
|
|
|
|
return new Version(version);
|
|
|
|
|
}
|
2013-11-28 07:11:05 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2019-09-03 02:22:25 +00:00
|
|
|
|
public int Migration
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2019-12-18 21:56:41 +00:00
|
|
|
|
using (var db = _datamapperFactory())
|
|
|
|
|
{
|
2022-03-28 20:45:52 +00:00
|
|
|
|
return db.QueryFirstOrDefault<int>("SELECT \"Version\" from \"VersionInfo\" ORDER BY \"Version\" DESC LIMIT 1");
|
2019-12-18 21:56:41 +00:00
|
|
|
|
}
|
2019-09-03 02:22:25 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-11-28 07:11:05 +00:00
|
|
|
|
public void Vacuum()
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
2015-05-03 18:52:07 +00:00
|
|
|
|
_logger.Info("Vacuuming {0} database", _databaseName);
|
2018-12-02 17:05:58 +00:00
|
|
|
|
using (var db = _datamapperFactory())
|
|
|
|
|
{
|
2019-12-18 21:56:41 +00:00
|
|
|
|
db.Execute("Vacuum;");
|
2018-12-02 17:05:58 +00:00
|
|
|
|
}
|
2019-12-22 22:08:53 +00:00
|
|
|
|
|
2015-05-03 18:52:07 +00:00
|
|
|
|
_logger.Info("{0} database compressed", _databaseName);
|
2013-11-28 07:11:05 +00:00
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
2016-02-11 21:13:42 +00:00
|
|
|
|
_logger.Error(e, "An Error occurred while vacuuming database.");
|
2013-11-28 07:11:05 +00:00
|
|
|
|
}
|
2013-05-11 20:06:57 +00:00
|
|
|
|
}
|
2013-03-25 03:51:32 +00:00
|
|
|
|
}
|
2022-03-28 20:45:52 +00:00
|
|
|
|
|
|
|
|
|
public enum DatabaseType
|
|
|
|
|
{
|
|
|
|
|
SQLite,
|
|
|
|
|
PostgreSQL
|
|
|
|
|
}
|
2018-02-13 18:07:26 +00:00
|
|
|
|
}
|