Jackett/src/Jackett/Services/CacheService.cs

116 lines
4.2 KiB
C#
Raw Normal View History

2015-07-26 14:22:20 +00:00
using AutoMapper;
using Jackett.Indexers;
2015-07-26 14:22:20 +00:00
using Jackett.Models;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Jackett.Services
{
public interface ICacheService
{
2015-08-07 21:40:45 +00:00
void CacheRssResults(IIndexer indexer, IEnumerable<ReleaseInfo> releases);
List<TrackerCacheResult> GetCachedResults(string serverUrl);
2015-08-07 21:40:45 +00:00
int GetNewItemCount(IIndexer indexer, IEnumerable<ReleaseInfo> releases);
2015-07-26 14:22:20 +00:00
}
public class CacheService : ICacheService
{
private readonly List<TrackerCache> cache = new List<TrackerCache>();
2015-08-07 21:40:45 +00:00
private readonly int MAX_RESULTS_PER_TRACKER = 1000;
private readonly TimeSpan AGE_LIMIT = new TimeSpan(7, 0, 0, 0);
2015-07-26 14:22:20 +00:00
2015-08-07 21:40:45 +00:00
public void CacheRssResults(IIndexer indexer, IEnumerable<ReleaseInfo> releases)
2015-07-26 14:22:20 +00:00
{
lock (cache)
{
var trackerCache = cache.Where(c => c.TrackerId == indexer.ID).FirstOrDefault();
2015-07-26 14:22:20 +00:00
if (trackerCache == null)
{
trackerCache = new TrackerCache();
trackerCache.TrackerId = indexer.ID;
trackerCache.TrackerName = indexer.DisplayName;
2015-07-26 14:22:20 +00:00
cache.Add(trackerCache);
}
foreach(var release in releases.OrderByDescending(i=>i.PublishDate))
{
// Skip old releases
if(release.PublishDate-DateTime.Now> AGE_LIMIT)
{
continue;
}
var existingItem = trackerCache.Results.Where(i => i.Result.Guid == release.Guid).FirstOrDefault();
if (existingItem == null)
{
existingItem = new CachedResult();
existingItem.Created = DateTime.Now;
trackerCache.Results.Add(existingItem);
}
existingItem.Result = release;
}
// Prune cache
foreach(var tracker in cache)
{
tracker.Results = tracker.Results.OrderByDescending(i => i.Created).Take(MAX_RESULTS_PER_TRACKER).ToList();
}
2015-08-07 21:40:45 +00:00
}
}
public int GetNewItemCount(IIndexer indexer, IEnumerable<ReleaseInfo> releases)
{
lock (cache)
{
int newItemCount = 0;
var trackerCache = cache.Where(c => c.TrackerId == indexer.ID).FirstOrDefault();
if (trackerCache != null)
{
foreach (var release in releases)
{
if (trackerCache.Results.Where(i => i.Result.Guid == release.Guid).Count() == 0)
{
newItemCount++;
}
}
}
else {
newItemCount++;
}
return newItemCount;
2015-07-26 14:22:20 +00:00
}
}
public List<TrackerCacheResult> GetCachedResults(string serverUrl)
2015-07-26 14:22:20 +00:00
{
lock (cache)
{
var results = new List<TrackerCacheResult>();
foreach(var tracker in cache)
{
foreach(var release in tracker.Results)
{
var item = Mapper.Map<TrackerCacheResult>(release.Result);
item.FirstSeen = release.Created;
item.Tracker = tracker.TrackerName;
2015-07-26 14:22:20 +00:00
item.Peers = item.Peers - item.Seeders; // Use peers as leechers
2015-08-07 21:40:45 +00:00
item.Link = item.ConvertToProxyLink(serverUrl, tracker.TrackerId);
if(item.Link!=null && item.Link.Scheme != "magnet" && !string.IsNullOrWhiteSpace(Engine.Server.Config.BlackholeDir))
item.BlackholeLink = item.ConvertToProxyLink(serverUrl, tracker.TrackerId, "blackhole");
2015-07-26 14:22:20 +00:00
results.Add(item);
}
}
2015-08-07 21:40:45 +00:00
return results.Take(3000).OrderByDescending(i=>i.PublishDate).ToList();
2015-07-26 14:22:20 +00:00
}
}
}
}