CleanDatabaseScheduledTask.cs 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. using System;
  2. using System.Threading;
  3. using System.Threading.Tasks;
  4. using MediaBrowser.Common.Configuration;
  5. using MediaBrowser.Controller.Entities;
  6. using MediaBrowser.Controller.Library;
  7. using MediaBrowser.Controller.Persistence;
  8. using MediaBrowser.Model.IO;
  9. using Microsoft.Extensions.Logging;
  10. namespace Emby.Server.Implementations.Data
  11. {
  12. public class CleanDatabaseScheduledTask : ILibraryPostScanTask
  13. {
  14. private readonly ILibraryManager _libraryManager;
  15. private readonly IItemRepository _itemRepo;
  16. private readonly ILogger _logger;
  17. private readonly IFileSystem _fileSystem;
  18. private readonly IApplicationPaths _appPaths;
  19. public CleanDatabaseScheduledTask(ILibraryManager libraryManager, IItemRepository itemRepo, ILogger logger, IFileSystem fileSystem, IApplicationPaths appPaths)
  20. {
  21. _libraryManager = libraryManager;
  22. _itemRepo = itemRepo;
  23. _logger = logger;
  24. _fileSystem = fileSystem;
  25. _appPaths = appPaths;
  26. }
  27. public Task Run(IProgress<double> progress, CancellationToken cancellationToken)
  28. {
  29. CleanDeadItems(cancellationToken, progress);
  30. return Task.CompletedTask;
  31. }
  32. private void CleanDeadItems(CancellationToken cancellationToken, IProgress<double> progress)
  33. {
  34. var itemIds = _libraryManager.GetItemIds(new InternalItemsQuery
  35. {
  36. HasDeadParentId = true
  37. });
  38. var numComplete = 0;
  39. var numItems = itemIds.Count;
  40. _logger.LogDebug("Cleaning {0} items with dead parent links", numItems);
  41. foreach (var itemId in itemIds)
  42. {
  43. cancellationToken.ThrowIfCancellationRequested();
  44. var item = _libraryManager.GetItemById(itemId);
  45. if (item != null)
  46. {
  47. _logger.LogInformation("Cleaning item {0} type: {1} path: {2}", item.Name, item.GetType().Name, item.Path ?? string.Empty);
  48. _libraryManager.DeleteItem(item, new DeleteOptions
  49. {
  50. DeleteFileLocation = false
  51. });
  52. }
  53. numComplete++;
  54. double percent = numComplete;
  55. percent /= numItems;
  56. progress.Report(percent * 100);
  57. }
  58. progress.Report(100);
  59. }
  60. }
  61. }