Skip to content

Commit

Permalink
New: Missing/Cutoff Unmet searches will search for episodes that have…
Browse files Browse the repository at this point in the history
…n't been searched recently first

Closes #3067
  • Loading branch information
markus101 committed Dec 28, 2022
1 parent 2491da0 commit f504dfc
Show file tree
Hide file tree
Showing 4 changed files with 66 additions and 30 deletions.
14 changes: 14 additions & 0 deletions src/NzbDrone.Core/IndexerSearch/EpisodeSearchGroup.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
using System;
using System.Collections.Generic;
using System.Linq;
using NzbDrone.Core.Tv;

namespace NzbDrone.Core.IndexerSearch
{
public class EpisodeSearchGroup
{
public int SeriesId { get; set; }
public int SeasonNumber { get; set; }
public List<Episode> Episodes { get; set; }
}
}
74 changes: 45 additions & 29 deletions src/NzbDrone.Core/IndexerSearch/EpisodeSearchService.cs
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using NLog;
using NzbDrone.Common.Extensions;
using NzbDrone.Common.Instrumentation.Extensions;
Expand Down Expand Up @@ -40,46 +39,63 @@ public class EpisodeSearchService : IExecute<EpisodeSearchCommand>,
_logger = logger;
}

private void SearchForEpisodes(List<Episode> episodes, bool monitoredOnly, bool userInvokedSearch)
private void SearchForBulkEpisodes(List<Episode> episodes, bool monitoredOnly, bool userInvokedSearch)
{
_logger.ProgressInfo("Performing search for {0} episodes", episodes.Count);
var downloadedCount = 0;
var groups = new List<EpisodeSearchGroup>();

foreach (var series in episodes.GroupBy(e => e.SeriesId))
{
foreach (var season in series.Select(e => e).GroupBy(e => e.SeasonNumber))
{
List<DownloadDecision> decisions;
groups.Add(new EpisodeSearchGroup
{
SeriesId = series.Key,
SeasonNumber = season.Key,
Episodes = season.ToList()
});
}
}

foreach (var group in groups.OrderBy(g => g.Episodes.Min(e => e.LastSearchTime ?? DateTime.MinValue)))
{
List<DownloadDecision> decisions;

if (season.Count() > 1)
var seriesId = group.SeriesId;
var seasonNumber = group.SeasonNumber;
var groupEpisodes = group.Episodes;

if (groupEpisodes.Count > 1)
{
try
{
try
{
decisions = _releaseSearchService.SeasonSearch(series.Key, season.Key, season.ToList(), monitoredOnly, userInvokedSearch, false);
}
catch (Exception ex)
{
_logger.Error(ex, "Unable to search for episodes in season {0} of [{1}]", season.Key, series.Key);
continue;
}
decisions = _releaseSearchService.SeasonSearch(seriesId, seasonNumber, groupEpisodes, monitoredOnly, userInvokedSearch, false);
}
else
catch (Exception ex)
{
try
{
decisions = _releaseSearchService.EpisodeSearch(season.First(), userInvokedSearch, false);
}
catch (Exception ex)
{
_logger.Error(ex, "Unable to search for episode: [{0}]", season.First());
continue;
}
_logger.Error(ex, "Unable to search for episodes in season {0} of [{1}]", seasonNumber, seriesId);
continue;
}
}
else
{
var episode = groupEpisodes.First();

var processed = _processDownloadDecisions.ProcessDecisions(decisions);

downloadedCount += processed.Grabbed.Count;
try
{
decisions = _releaseSearchService.EpisodeSearch(episode, userInvokedSearch, false);
}
catch (Exception ex)
{
_logger.Error(ex, "Unable to search for episode: [{0}]", episode);
continue;
}
}

var processed = _processDownloadDecisions.ProcessDecisions(decisions);

downloadedCount += processed.Grabbed.Count;
}

_logger.ProgressInfo("Completed search for {0} episodes. {1} reports downloaded.", episodes.Count, downloadedCount);
Expand Down Expand Up @@ -120,7 +136,7 @@ public void Execute(MissingEpisodeSearchCommand message)
var pagingSpec = new PagingSpec<Episode>
{
Page = 1,
PageSize = 100000,
PageSize = 1000000,
SortDirection = SortDirection.Ascending,
SortKey = "Id"
};
Expand All @@ -140,7 +156,7 @@ public void Execute(MissingEpisodeSearchCommand message)
var queue = _queueService.GetQueue().Where(q => q.Episode != null).Select(q => q.Episode.Id);
var missing = episodes.Where(e => !queue.Contains(e.Id)).ToList();

SearchForEpisodes(missing, monitored, message.Trigger == CommandTrigger.Manual);
SearchForBulkEpisodes(missing, monitored, message.Trigger == CommandTrigger.Manual);
}

public void Execute(CutoffUnmetEpisodeSearchCommand message)
Expand Down Expand Up @@ -174,7 +190,7 @@ public void Execute(CutoffUnmetEpisodeSearchCommand message)
var queue = _queueService.GetQueue().Where(q => q.Episode != null).Select(q => q.Episode.Id);
var cutoffUnmet = episodes.Where(e => !queue.Contains(e.Id)).ToList();

SearchForEpisodes(cutoffUnmet, monitored, message.Trigger == CommandTrigger.Manual);
SearchForBulkEpisodes(cutoffUnmet, monitored, message.Trigger == CommandTrigger.Manual);
}
}
}
2 changes: 1 addition & 1 deletion src/NzbDrone.Core/IndexerSearch/ReleaseSearchService.cs
Original file line number Diff line number Diff line change
Expand Up @@ -540,7 +540,7 @@ private List<DownloadDecision> Dispatch(Func<IIndexer, IEnumerable<ReleaseInfo>>
_logger.Debug("Setting last search time to: {0}", lastSearchTime);

criteriaBase.Episodes.ForEach(e => e.LastSearchTime = lastSearchTime);
_episodeService.UpdateEpisodes(criteriaBase.Episodes);
_episodeService.UpdateLastSearchTime(criteriaBase.Episodes);
}

return _makeDownloadDecision.GetSearchDecision(reports, criteriaBase).ToList();
Expand Down
6 changes: 6 additions & 0 deletions src/NzbDrone.Core/Tv/EpisodeService.cs
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ public interface IEpisodeService
void SetEpisodeMonitored(int episodeId, bool monitored);
void SetMonitored(IEnumerable<int> ids, bool monitored);
void UpdateEpisodes(List<Episode> episodes);
void UpdateLastSearchTime(List<Episode> episodes);
List<Episode> EpisodesBetweenDates(DateTime start, DateTime end, bool includeUnmonitored);
void InsertMany(List<Episode> episodes);
void UpdateMany(List<Episode> episodes);
Expand Down Expand Up @@ -187,6 +188,11 @@ public void UpdateEpisodes(List<Episode> episodes)
_episodeRepository.UpdateMany(episodes);
}

public void UpdateLastSearchTime(List<Episode> episodes)
{
_episodeRepository.SetFields(episodes, e => e.LastSearchTime);
}

public List<Episode> EpisodesBetweenDates(DateTime start, DateTime end, bool includeUnmonitored)
{
var episodes = _episodeRepository.EpisodesBetweenDates(start.ToUniversalTime(), end.ToUniversalTime(), includeUnmonitored);
Expand Down

0 comments on commit f504dfc

Please sign in to comment.