Using TPL for feed downloading - more speed!

New: Faster searching and rss syncing
pull/6/head
Mark McDowall 12 years ago
parent 58557ef25c
commit 4cefc5323c

@ -1,6 +1,7 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Threading.Tasks;
using Ninject; using Ninject;
using NLog; using NLog;
using NzbDrone.Core.Model; using NzbDrone.Core.Model;
@ -8,12 +9,12 @@ using NzbDrone.Core.Model.Notification;
using NzbDrone.Core.Providers; using NzbDrone.Core.Providers;
using NzbDrone.Core.Providers.DecisionEngine; using NzbDrone.Core.Providers.DecisionEngine;
using NzbDrone.Core.Providers.Indexer; using NzbDrone.Core.Providers.Indexer;
using StackExchange.Profiling;
namespace NzbDrone.Core.Jobs namespace NzbDrone.Core.Jobs
{ {
public class RssSyncJob : IJob public class RssSyncJob : IJob
{ {
private readonly IEnumerable<IndexerBase> _indexers;
private readonly DownloadProvider _downloadProvider; private readonly DownloadProvider _downloadProvider;
private readonly IndexerProvider _indexerProvider; private readonly IndexerProvider _indexerProvider;
private readonly MonitoredEpisodeSpecification _isMonitoredEpisodeSpecification; private readonly MonitoredEpisodeSpecification _isMonitoredEpisodeSpecification;
@ -24,10 +25,9 @@ namespace NzbDrone.Core.Jobs
private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
[Inject] [Inject]
public RssSyncJob(IEnumerable<IndexerBase> indexers, DownloadProvider downloadProvider, IndexerProvider indexerProvider, public RssSyncJob(DownloadProvider downloadProvider, IndexerProvider indexerProvider,
MonitoredEpisodeSpecification isMonitoredEpisodeSpecification, AllowedDownloadSpecification allowedDownloadSpecification, UpgradeHistorySpecification upgradeHistorySpecification) MonitoredEpisodeSpecification isMonitoredEpisodeSpecification, AllowedDownloadSpecification allowedDownloadSpecification, UpgradeHistorySpecification upgradeHistorySpecification)
{ {
_indexers = indexers;
_downloadProvider = downloadProvider; _downloadProvider = downloadProvider;
_indexerProvider = indexerProvider; _indexerProvider = indexerProvider;
_isMonitoredEpisodeSpecification = isMonitoredEpisodeSpecification; _isMonitoredEpisodeSpecification = isMonitoredEpisodeSpecification;
@ -49,20 +49,22 @@ namespace NzbDrone.Core.Jobs
{ {
var reports = new List<EpisodeParseResult>(); var reports = new List<EpisodeParseResult>();
foreach (var indexer in _indexers.Where(i => _indexerProvider.GetSettings(i.GetType()).Enable)) notification.CurrentMessage = "Fetching RSS";
Parallel.ForEach(_indexerProvider.GetEnabledIndexers(), indexer =>
{ {
try try
{ {
notification.CurrentMessage = "Fetching RSS from " + indexer.Name;
reports.AddRange(indexer.FetchRss()); reports.AddRange(indexer.FetchRss());
} }
catch (Exception e) catch (Exception e)
{ {
Logger.ErrorException("An error has occurred while fetching items from " + indexer.Name, e); Logger.ErrorException("An error has occurred while fetching items from " + indexer.Name, e);
} }
} });
Logger.Debug("Finished fetching reports from all indexers. Total {0}", reports.Count); Logger.Debug("Finished fetching reports from all indexers. Total {0}", reports.Count);
notification.CurrentMessage = "Processing downloaded RSS"; notification.CurrentMessage = "Processing downloaded RSS";
foreach (var episodeParseResult in reports) foreach (var episodeParseResult in reports)

@ -2,6 +2,7 @@
using System.Collections; using System.Collections;
using System.Collections.Generic; using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Threading.Tasks;
using NLog; using NLog;
using Ninject; using Ninject;
using NzbDrone.Core.Model; using NzbDrone.Core.Model;
@ -208,7 +209,6 @@ namespace NzbDrone.Core.Providers
{ {
//If single episode, do a single episode search, if full season then do a full season search, otherwise, do a partial search //If single episode, do a single episode search, if full season then do a full season search, otherwise, do a partial search
var indexers = _indexerProvider.GetEnabledIndexers();
var reports = new List<EpisodeParseResult>(); var reports = new List<EpisodeParseResult>();
var title = _sceneMappingProvider.GetSceneName(series.SeriesId); var title = _sceneMappingProvider.GetSceneName(series.SeriesId);
@ -218,7 +218,7 @@ namespace NzbDrone.Core.Providers
title = series.Title; title = series.Title;
} }
foreach (var indexer in indexers) Parallel.ForEach(_indexerProvider.GetEnabledIndexers(), indexer =>
{ {
try try
{ {
@ -252,7 +252,7 @@ namespace NzbDrone.Core.Providers
{ {
Logger.ErrorException("An error has occurred while fetching items from " + indexer.Name, e); Logger.ErrorException("An error has occurred while fetching items from " + indexer.Name, e);
} }
} });
return reports; return reports;
} }

Loading…
Cancel
Save