mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 11:34:57 +02:00
249 lines
No EOL
8.4 KiB
C#
249 lines
No EOL
8.4 KiB
C#
using MediaBrowser.Common.Progress;
|
|
using MediaBrowser.Common.ScheduledTasks;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.LiveTv;
|
|
using MediaBrowser.Controller.Persistence;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Logging;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using CommonIO;
|
|
using MediaBrowser.Controller.Channels;
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
|
|
namespace MediaBrowser.Server.Implementations.Persistence
|
|
{
|
|
public class CleanDatabaseScheduledTask : IScheduledTask
|
|
{
|
|
private readonly ILibraryManager _libraryManager;
|
|
private readonly IItemRepository _itemRepo;
|
|
private readonly ILogger _logger;
|
|
private readonly IServerConfigurationManager _config;
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
public const int MigrationVersion = 7;
|
|
|
|
public CleanDatabaseScheduledTask(ILibraryManager libraryManager, IItemRepository itemRepo, ILogger logger, IServerConfigurationManager config, IFileSystem fileSystem)
|
|
{
|
|
_libraryManager = libraryManager;
|
|
_itemRepo = itemRepo;
|
|
_logger = logger;
|
|
_config = config;
|
|
_fileSystem = fileSystem;
|
|
}
|
|
|
|
public string Name
|
|
{
|
|
get { return "Clean Database"; }
|
|
}
|
|
|
|
public string Description
|
|
{
|
|
get { return "Deletes obsolete content from the database."; }
|
|
}
|
|
|
|
public string Category
|
|
{
|
|
get { return "Library"; }
|
|
}
|
|
|
|
public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
|
|
{
|
|
var innerProgress = new ActionableProgress<double>();
|
|
innerProgress.RegisterAction(p => progress.Report(.4 * p));
|
|
|
|
await UpdateToLatestSchema(cancellationToken, innerProgress).ConfigureAwait(false);
|
|
|
|
innerProgress = new ActionableProgress<double>();
|
|
innerProgress.RegisterAction(p => progress.Report(40 + (.05 * p)));
|
|
await CleanDeadItems(cancellationToken, innerProgress).ConfigureAwait(false);
|
|
progress.Report(45);
|
|
|
|
innerProgress = new ActionableProgress<double>();
|
|
innerProgress.RegisterAction(p => progress.Report(45 + (.55 * p)));
|
|
await CleanDeletedItems(cancellationToken, innerProgress).ConfigureAwait(false);
|
|
progress.Report(100);
|
|
|
|
await _itemRepo.UpdateInheritedValues(cancellationToken).ConfigureAwait(false);
|
|
}
|
|
|
|
private async Task UpdateToLatestSchema(CancellationToken cancellationToken, IProgress<double> progress)
|
|
{
|
|
var itemIds = _libraryManager.GetItemIds(new InternalItemsQuery
|
|
{
|
|
IsCurrentSchema = false
|
|
});
|
|
|
|
var numComplete = 0;
|
|
var numItems = itemIds.Count;
|
|
|
|
_logger.Debug("Upgrading schema for {0} items", numItems);
|
|
|
|
foreach (var itemId in itemIds)
|
|
{
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
if (itemId == Guid.Empty)
|
|
{
|
|
// Somehow some invalid data got into the db. It probably predates the boundary checking
|
|
continue;
|
|
}
|
|
|
|
var item = _libraryManager.GetItemById(itemId);
|
|
|
|
if (item != null)
|
|
{
|
|
try
|
|
{
|
|
await _itemRepo.SaveItem(item, cancellationToken).ConfigureAwait(false);
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
throw;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error saving item", ex);
|
|
}
|
|
}
|
|
|
|
numComplete++;
|
|
double percent = numComplete;
|
|
percent /= numItems;
|
|
progress.Report(percent * 100);
|
|
}
|
|
|
|
if (_config.Configuration.MigrationVersion < MigrationVersion)
|
|
{
|
|
_config.Configuration.MigrationVersion = MigrationVersion;
|
|
_config.SaveConfiguration();
|
|
}
|
|
|
|
progress.Report(100);
|
|
}
|
|
|
|
private async Task CleanDeadItems(CancellationToken cancellationToken, IProgress<double> progress)
|
|
{
|
|
var itemIds = _libraryManager.GetItemIds(new InternalItemsQuery
|
|
{
|
|
HasDeadParentId = true
|
|
});
|
|
|
|
var numComplete = 0;
|
|
var numItems = itemIds.Count;
|
|
|
|
_logger.Debug("Cleaning {0} items with dead parent links", numItems);
|
|
|
|
foreach (var itemId in itemIds)
|
|
{
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
var item = _libraryManager.GetItemById(itemId);
|
|
|
|
if (item != null)
|
|
{
|
|
_logger.Info("Cleaning item {0} type: {1} path: {2}", item.Name, item.GetType().Name, item.Path ?? string.Empty);
|
|
|
|
await _libraryManager.DeleteItem(item, new DeleteOptions
|
|
{
|
|
DeleteFileLocation = false
|
|
});
|
|
}
|
|
|
|
numComplete++;
|
|
double percent = numComplete;
|
|
percent /= numItems;
|
|
progress.Report(percent * 100);
|
|
}
|
|
|
|
progress.Report(100);
|
|
}
|
|
|
|
private async Task CleanDeletedItems(CancellationToken cancellationToken, IProgress<double> progress)
|
|
{
|
|
var result = _itemRepo.GetItemIdsWithPath(new InternalItemsQuery
|
|
{
|
|
LocationType = LocationType.FileSystem,
|
|
//Limit = limit,
|
|
|
|
// These have their own cleanup routines
|
|
ExcludeItemTypes = new[]
|
|
{
|
|
typeof(Person).Name,
|
|
typeof(Genre).Name,
|
|
typeof(MusicGenre).Name,
|
|
typeof(GameGenre).Name,
|
|
typeof(Studio).Name,
|
|
typeof(Year).Name,
|
|
typeof(Channel).Name,
|
|
typeof(AggregateFolder).Name,
|
|
typeof(CollectionFolder).Name
|
|
}
|
|
});
|
|
|
|
var numComplete = 0;
|
|
var numItems = result.Items.Length;
|
|
|
|
foreach (var item in result.Items)
|
|
{
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
var path = item.Item2;
|
|
|
|
try
|
|
{
|
|
if (_fileSystem.FileExists(path) || _fileSystem.DirectoryExists(path))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
var libraryItem = _libraryManager.GetItemById(item.Item1);
|
|
|
|
if (libraryItem.IsTopParent)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (Folder.IsPathOffline(path))
|
|
{
|
|
libraryItem.IsOffline = true;
|
|
await libraryItem.UpdateToRepository(ItemUpdateType.None, cancellationToken).ConfigureAwait(false);
|
|
continue;
|
|
}
|
|
|
|
_logger.Info("Deleting item from database {0} because path no longer exists. type: {1} path: {2}", libraryItem.Name, libraryItem.GetType().Name, libraryItem.Path ?? string.Empty);
|
|
|
|
await _libraryManager.DeleteItem(libraryItem, new DeleteOptions
|
|
{
|
|
DeleteFileLocation = false
|
|
});
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
throw;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error in CleanDeletedItems. File {0}", ex, path);
|
|
}
|
|
|
|
numComplete++;
|
|
double percent = numComplete;
|
|
percent /= numItems;
|
|
progress.Report(percent * 100);
|
|
}
|
|
}
|
|
|
|
public IEnumerable<ITaskTrigger> GetDefaultTriggers()
|
|
{
|
|
return new ITaskTrigger[]
|
|
{
|
|
new IntervalTrigger{ Interval = TimeSpan.FromHours(24)}
|
|
};
|
|
}
|
|
}
|
|
} |