New: Monitor and Process downloads separately

Co-Authored-By: Mark McDowall <markus101@users.noreply.github.com>
2026-04-21 22:05:43 -04:00 · 2020-03-29 13:48:34 -04:00
parent f430d1aab2
commit 72caab1b2b
57 changed files with 1231 additions and 478 deletions
@@ -0,0 +1,78 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NLog;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.Download.TrackedDownloads;
+using NzbDrone.Core.Messaging.Commands;
+using NzbDrone.Core.Messaging.Events;
+
+namespace NzbDrone.Core.Download
+{
+    public class DownloadProcessingService : IExecute<ProcessMonitoredDownloadsCommand>
+    {
+        private readonly IConfigService _configService;
+        private readonly ICompletedDownloadService _completedDownloadService;
+        private readonly IFailedDownloadService _failedDownloadService;
+        private readonly ITrackedDownloadService _trackedDownloadService;
+        private readonly IEventAggregator _eventAggregator;
+        private readonly Logger _logger;
+
+        public DownloadProcessingService(IConfigService configService,
+                                         ICompletedDownloadService completedDownloadService,
+                                         IFailedDownloadService failedDownloadService,
+                                         ITrackedDownloadService trackedDownloadService,
+                                         IEventAggregator eventAggregator,
+                                         Logger logger)
+        {
+            _configService = configService;
+            _completedDownloadService = completedDownloadService;
+            _failedDownloadService = failedDownloadService;
+            _trackedDownloadService = trackedDownloadService;
+            _eventAggregator = eventAggregator;
+            _logger = logger;
+        }
+
+        private void RemoveCompletedDownloads(List<TrackedDownload> trackedDownloads)
+        {
+            foreach (var trackedDownload in trackedDownloads.Where(c => c.DownloadItem.CanBeRemoved && c.State == TrackedDownloadState.Imported))
+            {
+                _eventAggregator.PublishEvent(new DownloadCompletedEvent(trackedDownload));
+            }
+        }
+
+        public void Execute(ProcessMonitoredDownloadsCommand message)
+        {
+            var enableCompletedDownloadHandling = _configService.EnableCompletedDownloadHandling;
+            var trackedDownloads = _trackedDownloadService.GetTrackedDownloads();
+
+            foreach (var trackedDownload in trackedDownloads)
+            {
+                try
+                {
+                    if (trackedDownload.State == TrackedDownloadState.FailedPending)
+                    {
+                        _failedDownloadService.ProcessFailed(trackedDownload);
+                    }
+
+                    if (enableCompletedDownloadHandling && trackedDownload.State == TrackedDownloadState.ImportPending)
+                    {
+                        _completedDownloadService.Import(trackedDownload);
+                    }
+                }
+                catch (Exception e)
+                {
+                    _logger.Debug(e, "Failed to process download: {0}", trackedDownload.DownloadItem.Title);
+                }
+            }
+
+            if (enableCompletedDownloadHandling && _configService.RemoveCompletedDownloads)
+            {
+                // Remove tracked downloads that are now complete
+                RemoveCompletedDownloads(trackedDownloads);
+            }
+
+            _eventAggregator.PublishEvent(new DownloadsProcessedEvent());
+        }
+    }
+}