DownloadClientRootFolderCheck Improvements

This commit is contained in:
Qstick 2023-01-16 22:38:05 -06:00 committed by Mark McDowall
parent 1f20bc63c6
commit ed22bb719d
2 changed files with 28 additions and 4 deletions

View File

@ -10,7 +10,7 @@ namespace NzbDrone.Core.Download
public interface IProvideDownloadClient public interface IProvideDownloadClient
{ {
IDownloadClient GetDownloadClient(DownloadProtocol downloadProtocol, int indexerId = 0); IDownloadClient GetDownloadClient(DownloadProtocol downloadProtocol, int indexerId = 0);
IEnumerable<IDownloadClient> GetDownloadClients(); IEnumerable<IDownloadClient> GetDownloadClients(bool filterBlockedClients = false);
IDownloadClient Get(int id); IDownloadClient Get(int id);
} }
@ -86,14 +86,37 @@ namespace NzbDrone.Core.Download
return provider; return provider;
} }
public IEnumerable<IDownloadClient> GetDownloadClients() public IEnumerable<IDownloadClient> GetDownloadClients(bool filterBlockedClients = false)
{ {
return _downloadClientFactory.GetAvailableProviders(); var enabledClients = _downloadClientFactory.GetAvailableProviders();
if (filterBlockedClients)
{
return FilterBlockedDownloadClients(enabledClients).ToList();
}
return enabledClients;
} }
public IDownloadClient Get(int id) public IDownloadClient Get(int id)
{ {
return _downloadClientFactory.GetAvailableProviders().Single(d => d.Definition.Id == id); return _downloadClientFactory.GetAvailableProviders().Single(d => d.Definition.Id == id);
} }
private IEnumerable<IDownloadClient> FilterBlockedDownloadClients(IEnumerable<IDownloadClient> clients)
{
var blockedClients = _downloadClientStatusService.GetBlockedProviders().ToDictionary(v => v.ProviderId, v => v);
foreach (var client in clients)
{
if (blockedClients.TryGetValue(client.Definition.Id, out var blockedClientStatus))
{
_logger.Debug("Temporarily ignoring client {0} till {1} due to recent failures.", client.Definition.Name, blockedClientStatus.DisabledTill.Value.ToLocalTime());
continue;
}
yield return client;
}
}
} }
} }

View File

@ -35,7 +35,8 @@ namespace NzbDrone.Core.HealthCheck.Checks
public override HealthCheck Check() public override HealthCheck Check()
{ {
// Only check clients not in failure status, those get another message // Only check clients not in failure status, those get another message
var clients = _downloadClientProvider.GetDownloadClients(); var clients = _downloadClientProvider.GetDownloadClients(true);
var rootFolders = _rootFolderService.All(); var rootFolders = _rootFolderService.All();
foreach (var client in clients) foreach (var client in clients)