self-hosted/Sonarr
1
0
Fork 0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-16 11:37:58 +02:00
Code Issues Releases Activity

Fixed: Blacklist Retry logic will now properly handle Sabnzbd changing the unique id.

Browse Source
This commit is contained in:
Taloth Saldono 2014-07-19 19:37:06 +02:00
parent 47089d360d
commit 53e723a301
11 changed files with 116 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/NzbDrone.Api/History/HistoryModule.cs
View File

@ -10,12 +10,12 @@ namespace NzbDrone.Api.History
public class HistoryModule : NzbDroneRestModule<HistoryResource>
{
private readonly IHistoryService _historyService;
private readonly IFailedDownloadService _failedDownloadService;
private readonly IDownloadTrackingService _downloadTrackingService;
public HistoryModule(IHistoryService historyService, IFailedDownloadService failedDownloadService)
public HistoryModule(IHistoryService historyService, IDownloadTrackingService downloadTrackingService)
{
_historyService = historyService;
_failedDownloadService = failedDownloadService;
_downloadTrackingService = downloadTrackingService;
GetResourcePaged = GetHistory;
Post["/failed"] = x => MarkAsFailed();
@ -51,7 +51,7 @@ private PagingResource<HistoryResource> GetHistory(PagingResource<HistoryResourc
private Response MarkAsFailed()
{
var id = (int)Request.Form.Id;
_failedDownloadService.MarkAsFailed(id);
_downloadTrackingService.MarkAsFailed(id);
return new Object().AsResponse();
}
}

10
src/NzbDrone.Core/DecisionEngine/Specifications/NotInQueueSpecification.cs
View File

@ -11,12 +11,12 @@ namespace NzbDrone.Core.DecisionEngine.Specifications
{
public class NotInQueueSpecification : IDecisionEngineSpecification
{
private readonly IQueueService _queueService;
private readonly IDownloadTrackingService _downloadTrackingService;
private readonly Logger _logger;
public NotInQueueSpecification(IQueueService queueService, Logger logger)
public NotInQueueSpecification(IDownloadTrackingService downloadTrackingService, Logger logger)
{
_queueService = queueService;
_downloadTrackingService = downloadTrackingService;
_logger = logger;
}
@ -30,7 +30,9 @@ public string RejectionReason
public bool IsSatisfiedBy(RemoteEpisode subject, SearchCriteriaBase searchCriteria)
{
var queue = _queueService.GetQueue().Select(q => q.RemoteEpisode);
var queue = _downloadTrackingService.GetQueuedDownloads()
.Where(v => v.State == TrackedDownloadState.Downloading)
.Select(q => q.DownloadItem.RemoteEpisode).ToList();
if (IsInQueue(subject, queue))
{

6
src/NzbDrone.Core/Download/Clients/Nzbget/Nzbget.cs
View File

@ -223,14 +223,16 @@ public override IEnumerable<DownloadClientItem> GetItems()
}
}
public override void RemoveItem(string id)
public override void RemoveItem(String id)
{
_proxy.RemoveFromHistory(id, Settings);
}
public override void RetryDownload(string id)
public override String RetryDownload(String id)
{
_proxy.RetryDownload(id, Settings);
return id;
}
public override DownloadClientStatus GetStatus()

4
src/NzbDrone.Core/Download/Clients/Pneumatic/Pneumatic.cs
View File

@ -75,12 +75,12 @@ public override IEnumerable<DownloadClientItem> GetItems()
return new DownloadClientItem[0];
}
public override void RemoveItem(string id)
public override void RemoveItem(String id)
{
throw new NotSupportedException();
}
public override void RetryDownload(string id)
public override String RetryDownload(String id)
{
throw new NotSupportedException();
}

22
src/NzbDrone.Core/Download/Clients/Sabnzbd/Sabnzbd.cs
View File

@ -215,7 +215,7 @@ public override IEnumerable<DownloadClientItem> GetItems()
}
}
public override void RemoveItem(string id)
public override void RemoveItem(String id)
{
if (GetQueue().Any(v => v.DownloadClientId == id))
{
@ -227,9 +227,27 @@ public override void RemoveItem(string id)
}
}
public override void RetryDownload(string id)
public override String RetryDownload(String id)
{
// Sabnzbd changed the nzo_id for retried downloads without reporting it back to us. We need to try to determine the new ID.
var history = GetHistory().Where(v => v.DownloadClientId == id).ToList();
_proxy.RetryDownload(id, Settings);
if (history.Count() != 1)
{
return id;
}
var queue = GetQueue().Where(v => v.Category == history.First().Category && v.Title == history.First().Title).ToList();
if (queue.Count() != 1)
{
return id;
}
return queue.First().DownloadClientId;
}
protected IEnumerable<SabnzbdCategory> GetCategories(SabnzbdConfig config)

4
src/NzbDrone.Core/Download/Clients/UsenetBlackhole/UsenetBlackhole.cs
View File

@ -124,12 +124,12 @@ public override IEnumerable<DownloadClientItem> GetItems()
}
}
public override void RemoveItem(string id)
public override void RemoveItem(String id)
{
throw new NotSupportedException();
}
public override void RetryDownload(string id)
public override String RetryDownload(String id)
{
throw new NotSupportedException();
}

4
src/NzbDrone.Core/Download/DownloadClientBase.cs
View File

@ -67,10 +67,10 @@ public abstract DownloadProtocol Protocol
get;
}
public abstract string Download(RemoteEpisode remoteEpisode);
public abstract String Download(RemoteEpisode remoteEpisode);
public abstract IEnumerable<DownloadClientItem> GetItems();
public abstract void RemoveItem(string id);
public abstract void RetryDownload(string id);
public abstract String RetryDownload(string id);
public abstract DownloadClientStatus GetStatus();
protected RemoteEpisode GetRemoteEpisode(String title)

18
src/NzbDrone.Core/Download/DownloadTrackingService.cs
View File

@ -18,6 +18,8 @@ public interface IDownloadTrackingService
TrackedDownload[] GetTrackedDownloads();
TrackedDownload[] GetCompletedDownloads();
TrackedDownload[] GetQueuedDownloads();
void MarkAsFailed(Int32 historyId);
}
public class DownloadTrackingService : IDownloadTrackingService, IExecute<CheckForFinishedDownloadCommand>, IHandleAsync<ApplicationStartedEvent>, IHandle<EpisodeGrabbedEvent>
@ -78,6 +80,22 @@ public TrackedDownload[] GetQueuedDownloads()
}, TimeSpan.FromSeconds(5.0));
}
public void MarkAsFailed(Int32 historyId)
{
var item = _historyService.Get(historyId);
var trackedDownload = GetTrackedDownloads()
.Where(h => h.DownloadItem.DownloadClientId.Equals(item.Data.GetValueOrDefault(DOWNLOAD_CLIENT_ID)))
.FirstOrDefault();
if (trackedDownload != null && trackedDownload.State == TrackedDownloadState.Unknown)
{
ProcessTrackedDownloads();
}
_failedDownloadService.MarkAsFailed(trackedDownload, item);
}
private TrackedDownload[] FilterQueuedDownloads(IEnumerable<TrackedDownload> trackedDownloads)
{
var enabledFailedDownloadHandling = _configService.EnableFailedDownloadHandling;

74
src/NzbDrone.Core/Download/FailedDownloadService.cs
View File

@ -13,7 +13,7 @@ namespace NzbDrone.Core.Download
{
public interface IFailedDownloadService
{
void MarkAsFailed(int historyId);
void MarkAsFailed(TrackedDownload trackedDownload, History.History grabbedHistory);
void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload trackedDownload, List<History.History> grabbedHistory, List<History.History> failedHistory);
}
@ -35,10 +35,14 @@ public FailedDownloadService(IHistoryService historyService,
_logger = logger;
}
public void MarkAsFailed(int historyId)
public void MarkAsFailed(TrackedDownload trackedDownload, History.History grabbedHistory)
{
var item = _historyService.Get(historyId);
PublishDownloadFailedEvent(new List<History.History> { item }, "Manually marked as failed");
if (trackedDownload != null && trackedDownload.State == TrackedDownloadState.Downloading)
{
trackedDownload.State = TrackedDownloadState.DownloadFailed;
}
PublishDownloadFailedEvent(new List<History.History> { grabbedHistory }, "Manually marked as failed");
}
public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload trackedDownload, List<History.History> grabbedHistory, List<History.History> failedHistory)
@ -54,7 +58,7 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
if (!grabbedItems.Any())
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Download was not grabbed by drone, ignoring download");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Download was not grabbed by drone, ignoring download");
return;
}
@ -64,7 +68,7 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
if (failedItems.Any())
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Already added to history as failed");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Already added to history as failed.");
}
else
{
@ -78,7 +82,7 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
if (!grabbedItems.Any())
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Download wasn't grabbed by drone or not in a category, ignoring download.");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Download wasn't grabbed by drone or not in a category, ignoring download.");
return;
}
@ -86,13 +90,13 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
if (trackedDownload.DownloadItem.Message.Equals("Unpacking failed, write error or disk is full?",
StringComparison.InvariantCultureIgnoreCase))
{
UpdateStatusMessage(LogLevel.Error, trackedDownload, "Download failed due to lack of disk space, not blacklisting.");
UpdateStatusMessage(trackedDownload, LogLevel.Error, "Download failed due to lack of disk space, not blacklisting.");
return;
}
if (FailedDownloadForRecentRelease(downloadClient, trackedDownload, grabbedItems))
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Recent release Failed, do not blacklist.");
_logger.Debug("[{0}] Recent release Failed, do not blacklist.", trackedDownload.DownloadItem.Title);
return;
}
@ -102,7 +106,7 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
if (failedItems.Any())
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Already added to history as failed.");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Already added to history as failed.");
}
else
{
@ -117,7 +121,7 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
if (grabbedItems.Any() && failedItems.Any())
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Already added to history as failed, updating tracked state.");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Already added to history as failed, updating tracked state.");
trackedDownload.State = TrackedDownloadState.DownloadFailed;
}
}
@ -126,14 +130,14 @@ public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload t
{
try
{
_logger.Debug("[{0}] Removing failed download from client", trackedDownload.DownloadItem.Title);
_logger.Debug("[{0}] Removing failed download from client.", trackedDownload.DownloadItem.Title);
downloadClient.RemoveItem(trackedDownload.DownloadItem.DownloadClientId);
trackedDownload.State = TrackedDownloadState.Removed;
}
catch (NotSupportedException)
{
UpdateStatusMessage(LogLevel.Debug, trackedDownload, "Removing item not supported by your download client.");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Removing item not supported by your download client.");
}
}
}
@ -144,38 +148,62 @@ private bool FailedDownloadForRecentRelease(IDownloadClient downloadClient, Trac
if (!Double.TryParse(matchingHistoryItems.First().Data.GetValueOrDefault("ageHours"), out ageHours))
{
_logger.Info("[{0}] Unable to determine age of failed download.", trackedDownload.DownloadItem.Title);
UpdateStatusMessage(trackedDownload, LogLevel.Info, "Unable to determine age of failed download.");
return false;
}
if (ageHours > _configService.BlacklistGracePeriod)
{
_logger.Info("[{0}] Failed download is older than the grace period.", trackedDownload.DownloadItem.Title);
UpdateStatusMessage(trackedDownload, LogLevel.Info, "Download Failed, Failed download is older than the grace period.");
return false;
}
if (trackedDownload.RetryCount >= _configService.BlacklistRetryLimit)
{
_logger.Info("[{0}] Retry limit reached.", trackedDownload.DownloadItem.Title);
UpdateStatusMessage(trackedDownload, LogLevel.Info, "Download Failed, Retry limit reached.");
return false;
}
if (trackedDownload.RetryCount == 0 || trackedDownload.LastRetry.AddMinutes(_configService.BlacklistRetryInterval) < DateTime.UtcNow)
if (trackedDownload.LastRetry == new DateTime())
{
trackedDownload.LastRetry = DateTime.UtcNow;
}
if (trackedDownload.LastRetry.AddMinutes(_configService.BlacklistRetryInterval) < DateTime.UtcNow)
{
_logger.Info("[{0}] Retrying failed release.", trackedDownload.DownloadItem.Title);
trackedDownload.LastRetry = DateTime.UtcNow;
trackedDownload.RetryCount++;
UpdateStatusMessage(trackedDownload, LogLevel.Info, "Download Failed, initiating retry attempt {0}/{1}.", trackedDownload.RetryCount, _configService.BlacklistRetryLimit);
try
{
downloadClient.RetryDownload(trackedDownload.DownloadItem.DownloadClientId);
var newDownloadClientId = downloadClient.RetryDownload(trackedDownload.DownloadItem.DownloadClientId);
if (newDownloadClientId != trackedDownload.DownloadItem.DownloadClientId)
{
var oldTrackingId = trackedDownload.TrackingId;
var newTrackingId = String.Format("{0}-{1}", downloadClient.Definition.Id, newDownloadClientId);
trackedDownload.TrackingId = newTrackingId;
trackedDownload.DownloadItem.DownloadClientId = newDownloadClientId;
_logger.Debug("[{0}] Changed id from {1} to {2}.", trackedDownload.DownloadItem.Title, oldTrackingId, newTrackingId);
var newHistoryData = new Dictionary<String, String>(matchingHistoryItems.First().Data);
newHistoryData[DownloadTrackingService.DOWNLOAD_CLIENT_ID] = newDownloadClientId;
_historyService.UpdateHistoryData(matchingHistoryItems.First().Id, newHistoryData);
}
}
catch (NotSupportedException ex)
{
_logger.Debug("Retrying failed downloads is not supported by your download client");
UpdateStatusMessage(trackedDownload, LogLevel.Debug, "Retrying failed downloads is not supported by your download client.");
return false;
}
}
else
{
UpdateStatusMessage(trackedDownload, LogLevel.Warn, "Download Failed, waiting for retry interval to expire.");
}
return true;
}
@ -206,16 +234,16 @@ private void PublishDownloadFailedEvent(List<History.History> historyItems, stri
_eventAggregator.PublishEvent(downloadFailedEvent);
}
private void UpdateStatusMessage(LogLevel logLevel, TrackedDownload trackedDownload, String message, params object[] args)
private void UpdateStatusMessage(TrackedDownload trackedDownload, LogLevel logLevel, String message, params object[] args)
{
var statusMessage = String.Format(message, args);
var logMessage = String.Format("[{0}] {1}", trackedDownload.DownloadItem.Title, message);
var logMessage = String.Format("[{0}] {1}", trackedDownload.DownloadItem.Title, statusMessage);
if (trackedDownload.StatusMessage != statusMessage)
{
trackedDownload.HasError = logLevel >= LogLevel.Warn;
trackedDownload.StatusMessage = statusMessage;
_logger.Log(logLevel, statusMessage);
_logger.Log(logLevel, logMessage);
}
else
{

9
src/NzbDrone.Core/Download/IDownloadClient.cs
View File

@ -1,4 +1,5 @@
using System.Collections.Generic;
using System;
using System.Collections.Generic;
using NzbDrone.Core.Indexers;
using NzbDrone.Core.Parser.Model;
using NzbDrone.Core.ThingiProvider;
@ -9,10 +10,10 @@ public interface IDownloadClient : IProvider
{
DownloadProtocol Protocol { get; }
string Download(RemoteEpisode remoteEpisode);
String Download(RemoteEpisode remoteEpisode);
IEnumerable<DownloadClientItem> GetItems();
void RemoveItem(string id);
void RetryDownload(string id);
void RemoveItem(String id);
String RetryDownload(String id);
DownloadClientStatus GetStatus();
}