mirror/Sonarr
1
0
Fork 0
mirror of https://github.com/Sonarr/Sonarr synced 2025-01-19 05:38:53 +00:00
Code Issues Releases Wiki Activity

Fixed: Tweaked ratelimit logic for rarbg api.

Browse source
This commit is contained in:
Taloth Saldono 2016-09-29 21:40:52 +02:00
parent e0d1e08f94
commit 7ef1ca8a00
2 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/NzbDrone.Common/Http/HttpRequestBuilder.cs
View file

@ -22,6 +22,7 @@ namespace NzbDrone.Common.Http
public bool UseSimplifiedUserAgent { get; set; }
public bool AllowAutoRedirect { get; set; }
public bool ConnectionKeepAlive { get; set; }
public TimeSpan RateLimit { get; set; }
public bool LogResponseContent { get; set; }
public NetworkCredential NetworkCredential { get; set; }
public Dictionary<string, string> Cookies { get; private set; }
@ -75,7 +76,7 @@ namespace NzbDrone.Common.Http
protected virtual HttpUri CreateUri()
{
var url = BaseUrl.CombinePath(ResourceUrl).AddQueryParams(QueryParams.Concat(SuffixQueryParams));
if (Segments.Any())
{
var fullUri = url.FullUri;
@ -103,6 +104,7 @@ namespace NzbDrone.Common.Http
request.UseSimplifiedUserAgent = UseSimplifiedUserAgent;
request.AllowAutoRedirect = AllowAutoRedirect;
request.ConnectionKeepAlive = ConnectionKeepAlive;
request.RateLimit = RateLimit;
request.LogResponseContent = LogResponseContent;
if (NetworkCredential != null)
@ -245,6 +247,13 @@ namespace NzbDrone.Common.Http
return this;
}
public virtual HttpRequestBuilder WithRateLimit(double seconds)
{
RateLimit = TimeSpan.FromSeconds(seconds);
return this;
}
public virtual HttpRequestBuilder Post()
{
Method = HttpMethod.POST;
@ -371,4 +380,4 @@ namespace NzbDrone.Common.Http
}
}
}
}

3
src/NzbDrone.Core/Indexers/Rarbg/RarbgTokenProvider.cs
View file

@ -30,7 +30,8 @@ namespace NzbDrone.Core.Indexers.Rarbg
return _tokenCache.Get(settings.BaseUrl, () =>
{
var requestBuilder = new HttpRequestBuilder(settings.BaseUrl.Trim('/'))
.Resource("/pubapi_v2.php?get_token=get_token")
.WithRateLimit(3.0)
.Resource("/pubapi_v2.php?get_token=get_token&app_id=Sonarr")
.Accept(HttpAccept.Json);
if (settings.CaptchaToken.IsNotNullOrWhiteSpace())