shpaq/Jackett-Jackett
1
0
Fork 0
mirror of https://github.com/Jackett/Jackett.git synced 2025-09-17 17:34:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

core: fix parsing Retry-After header in TooManyRequestsException

Browse Source
This commit is contained in:
Bogdan
2023-06-18 12:59:59 +03:00
parent cc44db15a4
commit 99204599d9
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/Jackett.Common/Exceptions/TooManyRequestsException.cs
View File

@@ -1,4 +1,5 @@
using System;
using System.Linq;
using Jackett.Common.Utils.Clients;
namespace Jackett.Common.Exceptions
@@ -16,15 +17,17 @@ namespace Jackett.Common.Exceptions
public TooManyRequestsException(string message, WebResult response)
: base(message)
{
if (response.Headers.ContainsKey("Retry-After"))
if (response.Headers.TryGetValue("Retry-After", out var header) && header.FirstOrDefault() is {} retryAfter)
{
var retryAfter = response.Headers["Retry-After"].ToString();
if (int.TryParse(retryAfter, out var seconds))
{
RetryAfter = TimeSpan.FromSeconds(seconds);
}
else if (DateTime.TryParse(retryAfter, out var date))
{
RetryAfter = date.ToUniversalTime() - DateTime.UtcNow;
}
}
}
}
}

2
src/Jackett.Server/Controllers/ResultsController.cs
View File

@@ -453,8 +453,10 @@ namespace Jackett.Server.Controllers
var retryAfter = Convert.ToInt32(tooManyRequestsException.RetryAfter.TotalSeconds);
if (retryAfter > 0)
{
HttpContext.Response.Headers.Add("Retry-After", $"{retryAfter}");
}
}
return GetErrorXML(900, ex.Message, StatusCodes.Status429TooManyRequests);
}