mirror of
https://github.com/Prowlarr/Prowlarr.git
synced 2026-04-21 22:25:03 -04:00
New: Application Status Warnings
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using NLog;
|
||||
using NzbDrone.Common.Composition;
|
||||
using NzbDrone.Core.Messaging.Events;
|
||||
@@ -8,13 +9,53 @@ namespace NzbDrone.Core.Applications
|
||||
{
|
||||
public interface IApplicationFactory : IProviderFactory<IApplication, ApplicationDefinition>
|
||||
{
|
||||
List<IApplication> SyncEnabled(bool filterBlockedIndexers = true);
|
||||
}
|
||||
|
||||
public class ApplicationFactory : ProviderFactory<IApplication, ApplicationDefinition>, IApplicationFactory
|
||||
{
|
||||
public ApplicationFactory(IApplicationsRepository providerRepository, IEnumerable<IApplication> providers, IContainer container, IEventAggregator eventAggregator, Logger logger)
|
||||
private readonly IApplicationStatusService _applicationStatusService;
|
||||
private readonly Logger _logger;
|
||||
|
||||
public ApplicationFactory(IApplicationStatusService applicationStatusService,
|
||||
IApplicationsRepository providerRepository,
|
||||
IEnumerable<IApplication> providers,
|
||||
IContainer container,
|
||||
IEventAggregator eventAggregator,
|
||||
Logger logger)
|
||||
: base(providerRepository, providers, container, eventAggregator, logger)
|
||||
{
|
||||
_applicationStatusService = applicationStatusService;
|
||||
_logger = logger;
|
||||
}
|
||||
|
||||
public List<IApplication> SyncEnabled(bool filterBlockedClients = true)
|
||||
{
|
||||
var enabledClients = GetAvailableProviders().Where(n => ((ApplicationDefinition)n.Definition).Enable);
|
||||
|
||||
if (filterBlockedClients)
|
||||
{
|
||||
return FilterBlockedApplications(enabledClients).ToList();
|
||||
}
|
||||
|
||||
return enabledClients.ToList();
|
||||
}
|
||||
|
||||
private IEnumerable<IApplication> FilterBlockedApplications(IEnumerable<IApplication> applications)
|
||||
{
|
||||
var blockedApplications = _applicationStatusService.GetBlockedProviders().ToDictionary(v => v.ProviderId, v => v);
|
||||
|
||||
foreach (var application in applications)
|
||||
{
|
||||
ApplicationStatus blockedApplicationStatus;
|
||||
if (blockedApplications.TryGetValue(application.Definition.Id, out blockedApplicationStatus))
|
||||
{
|
||||
_logger.Debug("Temporarily ignoring application {0} till {1} due to recent failures.", application.Definition.Name, blockedApplicationStatus.DisabledTill.Value.ToLocalTime());
|
||||
continue;
|
||||
}
|
||||
|
||||
yield return application;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user