a023d33d90
- Applied to TorBox, can be extended easily to other providers - Reads response headers commonplace for pre-emptive rate limit throttling and retry-after - When a rate limit is reached, displays a warning in the UI - Fix socket leak from direct allocation of HttpClient, replaced with factory which handles pooling and re-use of sockets. While HttpClient is Disposable, it doesn't gaurantee (and does not) directly release underlying sockets for queries at the time the client is disposed. These sockets will go into a TCP WAIT state often for a very long time. The expected pattern in C# is to always use the HttClientFactory which will correctly handle re-use of the OS sockets in suqsequent queries reducing resource and memory leaks.
75 lines
2.6 KiB
C#
75 lines
2.6 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Microsoft.Extensions.Logging;
|
|
using RdtClient.Data.Enums;
|
|
using RdtClient.Data.Models.Internal;
|
|
using RdtClient.Service.Services;
|
|
|
|
namespace RdtClient.Service.BackgroundServices;
|
|
|
|
public class ProviderUpdater(ILogger<ProviderUpdater> logger, IServiceProvider serviceProvider) : BackgroundService
|
|
{
|
|
private static DateTime _nextUpdate = DateTime.UtcNow;
|
|
|
|
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
|
|
{
|
|
while (!Startup.Ready)
|
|
{
|
|
await Task.Delay(1000, stoppingToken);
|
|
}
|
|
|
|
using var scope = serviceProvider.CreateScope();
|
|
var torrentService = scope.ServiceProvider.GetRequiredService<Torrents>();
|
|
var torrentRunner = scope.ServiceProvider.GetRequiredService<TorrentRunner>();
|
|
logger.LogInformation("ProviderUpdater started.");
|
|
|
|
while (!stoppingToken.IsCancellationRequested)
|
|
{
|
|
try
|
|
{
|
|
var torrents = await torrentService.Get();
|
|
|
|
if (_nextUpdate < DateTime.UtcNow && (Settings.Get.Provider.AutoImport || torrents.Any(t => t.RdStatus != TorrentStatus.Finished) || RdtHub.HasConnections))
|
|
{
|
|
logger.LogDebug($"Updating torrent info from debrid provider");
|
|
|
|
var updateTime = Settings.Get.Provider.CheckInterval * 3;
|
|
|
|
if (updateTime < 30)
|
|
{
|
|
updateTime = 30;
|
|
}
|
|
|
|
if (RdtHub.HasConnections)
|
|
{
|
|
updateTime = Settings.Get.Provider.CheckInterval;
|
|
|
|
if (updateTime < 1)
|
|
{
|
|
updateTime = 1;
|
|
}
|
|
}
|
|
|
|
_nextUpdate = DateTime.UtcNow.AddSeconds(updateTime);
|
|
|
|
await torrentService.UpdateRdData();
|
|
|
|
logger.LogDebug("Finished updating torrent info from debrid provider, next update in {updateTime} seconds", updateTime);
|
|
}
|
|
}
|
|
catch (RateLimitException ex)
|
|
{
|
|
await torrentRunner.SetRateLimit(ex.RetryAfter, ex.Message);
|
|
_nextUpdate = DateTime.UtcNow.Add(ex.RetryAfter);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
logger.LogError(ex, "Unexpected error occurred in ProviderUpdater: {ex.Message}", ex.Message);
|
|
}
|
|
|
|
await Task.Delay(TimeSpan.FromSeconds(1), stoppingToken);
|
|
}
|
|
|
|
logger.LogInformation("ProviderUpdater stopped.");
|
|
}
|
|
}
|