mirror of https://github.com/Jackett/Jackett
102 lines
3.5 KiB
C#
102 lines
3.5 KiB
C#
using AutoMapper;
|
|
using Jackett.Indexers;
|
|
using Jackett.Models;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Jackett.Services.Interfaces;
|
|
|
|
namespace Jackett.Services
|
|
{
|
|
|
|
public class CacheService : ICacheService
|
|
{
|
|
private readonly List<TrackerCache> cache = new List<TrackerCache>();
|
|
private readonly int MAX_RESULTS_PER_TRACKER = 1000;
|
|
private readonly TimeSpan AGE_LIMIT = new TimeSpan(7, 0, 0, 0);
|
|
|
|
public void CacheRssResults(IIndexer indexer, IEnumerable<ReleaseInfo> releases)
|
|
{
|
|
lock (cache)
|
|
{
|
|
var trackerCache = cache.Where(c => c.TrackerId == indexer.ID).FirstOrDefault();
|
|
if (trackerCache == null)
|
|
{
|
|
trackerCache = new TrackerCache();
|
|
trackerCache.TrackerId = indexer.ID;
|
|
trackerCache.TrackerName = indexer.DisplayName;
|
|
cache.Add(trackerCache);
|
|
}
|
|
|
|
foreach(var release in releases.OrderByDescending(i=>i.PublishDate))
|
|
{
|
|
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();
|
|
}
|
|
}
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
|
|
public List<TrackerCacheResult> GetCachedResults()
|
|
{
|
|
lock (cache)
|
|
{
|
|
var results = new List<TrackerCacheResult>();
|
|
|
|
foreach(var tracker in cache)
|
|
{
|
|
foreach(var release in tracker.Results.OrderByDescending(i => i.Result.PublishDate).Take(300))
|
|
{
|
|
var item = Mapper.Map<TrackerCacheResult>(release.Result);
|
|
item.FirstSeen = release.Created;
|
|
item.Tracker = tracker.TrackerName;
|
|
item.TrackerId = tracker.TrackerId;
|
|
item.Peers = item.Peers - item.Seeders; // Use peers as leechers
|
|
results.Add(item);
|
|
}
|
|
}
|
|
|
|
return results.Take(3000).OrderByDescending(i=>i.PublishDate).ToList();
|
|
}
|
|
}
|
|
}
|
|
}
|