Backup_Repos/Prowlarr
1
0
Fork 0
mirror of https://github.com/Prowlarr/Prowlarr synced 2025-12-30 04:14:56 +01:00
Code Issues Projects Releases Wiki Activity

New: Additional logging for requests when query/grab limit are set

Browse source
This commit is contained in:
Qstick 2021-11-21 18:46:12 -06:00
parent 19862cda6c
commit 03cde56333
1 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/NzbDrone.Core/Indexers/IndexerLimitService.cs
View file

@ -3,7 +3,6 @@
using NLog;
using NzbDrone.Common.Extensions;
using NzbDrone.Core.History;
using NzbDrone.Core.Messaging.Events;
namespace NzbDrone.Core.Indexers
{
@ -15,15 +14,12 @@ public interface IIndexerLimitService
public class IndexerLimitService : IIndexerLimitService
{
private readonly IEventAggregator _eventAggregator;
private readonly IHistoryService _historyService;
private readonly Logger _logger;
public IndexerLimitService(IEventAggregator eventAggregator,
IHistoryService historyService,
Logger logger)
public IndexerLimitService(IHistoryService historyService,
Logger logger)
{
_eventAggregator = eventAggregator;
_historyService = historyService;
_logger = logger;
}
@ -32,14 +28,17 @@ public bool AtDownloadLimit(IndexerDefinition indexer)
{
if (indexer.Id > 0 && ((IIndexerSettings)indexer.Settings).BaseSettings.GrabLimit.HasValue)
{
var queryCount = _historyService.CountSince(indexer.Id, DateTime.Now.StartOfDay(), new List<HistoryEventType> { HistoryEventType.ReleaseGrabbed });
var grabCount = _historyService.CountSince(indexer.Id, DateTime.Now.StartOfDay(), new List<HistoryEventType> { HistoryEventType.ReleaseGrabbed });
var grabLimit = ((IIndexerSettings)indexer.Settings).BaseSettings.QueryLimit;
if (queryCount > ((IIndexerSettings)indexer.Settings).BaseSettings.GrabLimit)
if (grabCount > grabLimit)
{
_logger.Info("Indexer {0} has exceeded maximum grab limit for today", indexer.Name);
return true;
}
_logger.Debug("Indexer {0} has performed {1} of possible {2} grabs for today, proceeding", indexer.Name, grabCount, grabLimit);
}
return false;
@ -50,13 +49,16 @@ public bool AtQueryLimit(IndexerDefinition indexer)
if (indexer.Id > 0 && ((IIndexerSettings)indexer.Settings).BaseSettings.QueryLimit.HasValue)
{
var queryCount = _historyService.CountSince(indexer.Id, DateTime.Now.StartOfDay(), new List<HistoryEventType> { HistoryEventType.IndexerQuery, HistoryEventType.IndexerRss });
var queryLimit = ((IIndexerSettings)indexer.Settings).BaseSettings.QueryLimit;
if (queryCount > ((IIndexerSettings)indexer.Settings).BaseSettings.QueryLimit)
if (queryCount > queryLimit)
{
_logger.Info("Indexer {0} has exceeded maximum query limit for today", indexer.Name);
return true;
}
_logger.Debug("Indexer {0} has performed {1} of possible {2} queries for today, proceeding", indexer.Name, queryCount, queryLimit);
}
return false;