CleanDatabaseScheduledTask.cs 2.5 KB

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