New: Stats filters

2026-04-20 22:14:34 -04:00 · 2021-06-24 00:12:37 -04:00
parent 01e7e924c4
commit a61d4ab88c
11 changed files with 397 additions and 139 deletions
@@ -1,43 +1,174 @@
+using System;
 using System.Collections.Generic;
 using System.Linq;
+using NzbDrone.Core.History;
+using NzbDrone.Core.Indexers;

 namespace NzbDrone.Core.IndexerStats
 {
    public interface IIndexerStatisticsService
    {
-        List<IndexerStatistics> IndexerStatistics();
-        List<UserAgentStatistics> UserAgentStatistics();
-        List<HostStatistics> HostStatistics();
+        CombinedStatistics IndexerStatistics(DateTime start, DateTime end);
    }

    public class IndexerStatisticsService : IIndexerStatisticsService
    {
-        private readonly IIndexerStatisticsRepository _indexerStatisticsRepository;
+        private readonly IIndexerFactory _indexerFactory;
+        private readonly IHistoryService _historyService;

-        public IndexerStatisticsService(IIndexerStatisticsRepository indexerStatisticsRepository)
+        public IndexerStatisticsService(IHistoryService historyService, IIndexerFactory indexerFactory)
        {
-            _indexerStatisticsRepository = indexerStatisticsRepository;
+            _historyService = historyService;
+            _indexerFactory = indexerFactory;
        }

-        public List<IndexerStatistics> IndexerStatistics()
+        public CombinedStatistics IndexerStatistics(DateTime start, DateTime end)
        {
-            var indexerStatistics = _indexerStatisticsRepository.IndexerStatistics();
+            var history = _historyService.Between(start, end);

-            return indexerStatistics.ToList();
-        }
+            var groupedByIndexer = history.GroupBy(h => h.IndexerId);
+            var groupedByUserAgent = history.GroupBy(h => h.Data.GetValueOrDefault("source") ?? "");
+            var groupedByHost = history.GroupBy(h => h.Data.GetValueOrDefault("host") ?? "");

-        public List<UserAgentStatistics> UserAgentStatistics()
-        {
-            var userAgentStatistics = _indexerStatisticsRepository.UserAgentStatistics();
+            var indexerStatsList = new List<IndexerStatistics>();
+            var userAgentStatsList = new List<UserAgentStatistics>();
+            var hostStatsList = new List<HostStatistics>();

-            return userAgentStatistics.ToList();
-        }
+            var indexers = _indexerFactory.All();

-        public List<HostStatistics> HostStatistics()
-        {
-            var hostStatistics = _indexerStatisticsRepository.HostStatistics();
+            foreach (var indexer in groupedByIndexer)
+            {
+                var indexerDef = indexers.SingleOrDefault(i => i.Id == indexer.Key);

-            return hostStatistics.ToList();
+                if (indexerDef == null)
+                {
+                    continue;
+                }
+
+                var indexerStats = new IndexerStatistics
+                {
+                    IndexerId = indexer.Key,
+                    IndexerName = indexerDef.Name
+                };
+
+                var sortedEvents = indexer.OrderBy(v => v.Date)
+                                          .ThenBy(v => v.Id)
+                                          .ToArray();
+
+                indexerStats.AverageResponseTime = (int)sortedEvents.Where(h => h.Data.ContainsKey("elapsedTime")).Select(h => int.Parse(h.Data.GetValueOrDefault("elapsedTime"))).Average();
+
+                foreach (var historyEvent in sortedEvents)
+                {
+                    var failed = !historyEvent.Successful;
+                    switch (historyEvent.EventType)
+                    {
+                        case HistoryEventType.IndexerQuery:
+                            indexerStats.NumberOfQueries++;
+                            if (failed)
+                            {
+                                indexerStats.NumberOfFailedQueries++;
+                            }
+
+                            break;
+                        case HistoryEventType.IndexerAuth:
+                            indexerStats.NumberOfAuthQueries++;
+                            if (failed)
+                            {
+                                indexerStats.NumberOfFailedAuthQueries++;
+                            }
+
+                            break;
+                        case HistoryEventType.ReleaseGrabbed:
+                            indexerStats.NumberOfGrabs++;
+                            if (failed)
+                            {
+                                indexerStats.NumberOfFailedGrabs++;
+                            }
+
+                            break;
+                        case HistoryEventType.IndexerRss:
+                            indexerStats.NumberOfRssQueries++;
+                            if (failed)
+                            {
+                                indexerStats.NumberOfFailedRssQueries++;
+                            }
+
+                            break;
+                        default:
+                            break;
+                    }
+                }
+
+                indexerStatsList.Add(indexerStats);
+            }
+
+            foreach (var indexer in groupedByUserAgent)
+            {
+                var indexerStats = new UserAgentStatistics
+                {
+                    UserAgent = indexer.Key
+                };
+
+                var sortedEvents = indexer.OrderBy(v => v.Date)
+                                          .ThenBy(v => v.Id)
+                                          .ToArray();
+
+                foreach (var historyEvent in sortedEvents)
+                {
+                    switch (historyEvent.EventType)
+                    {
+                        case HistoryEventType.IndexerRss:
+                        case HistoryEventType.IndexerQuery:
+                            indexerStats.NumberOfQueries++;
+
+                            break;
+                        case HistoryEventType.ReleaseGrabbed:
+                            indexerStats.NumberOfGrabs++;
+                            break;
+                        default:
+                            break;
+                    }
+                }
+
+                userAgentStatsList.Add(indexerStats);
+            }
+
+            foreach (var indexer in groupedByHost)
+            {
+                var indexerStats = new HostStatistics
+                {
+                    Host = indexer.Key
+                };
+
+                var sortedEvents = indexer.OrderBy(v => v.Date)
+                                          .ThenBy(v => v.Id)
+                                          .ToArray();
+
+                foreach (var historyEvent in sortedEvents)
+                {
+                    switch (historyEvent.EventType)
+                    {
+                        case HistoryEventType.IndexerRss:
+                        case HistoryEventType.IndexerQuery:
+                            indexerStats.NumberOfQueries++;
+                            break;
+                        case HistoryEventType.ReleaseGrabbed:
+                            indexerStats.NumberOfGrabs++;
+                            break;
+                        default:
+                            break;
+                    }
+                }
+
+                hostStatsList.Add(indexerStats);
+            }
+
+            return new CombinedStatistics
+            {
+                IndexerStatistics = indexerStatsList,
+                UserAgentStatistics = userAgentStatsList,
+                HostStatistics = hostStatsList
+            };
        }
    }
 }