public async Task PrepareTwitterRequestAsync(ITwitterRequest request)
{
var twitterQuery = request.Query;
twitterQuery.Url = CleanupQueryUrl(twitterQuery.Url); // TODO : THIS LOGIC SHOULD HAPPEN BEFORE ARRIVING HERE
var rateLimitTrackerMode = request.ExecutionContext.RateLimitTrackerMode;
if (rateLimitTrackerMode == RateLimitTrackerMode.None)
{
return;
}
// Use the RateLimitCacheManager instead of RateLimitHelper to get the queryRateLimits to ensure the cache is up to date!
var credentialRateLimits = await _rateLimitCacheManager.GetCredentialsRateLimitsAsync(twitterQuery.TwitterCredentials).ConfigureAwait(false);
IEndpointRateLimit queryRateLimit = null;
// If we were not able to retrieve the credentials few ms before there is no reason why it would work now.
if (credentialRateLimits != null)
{
var getEndpointRateLimitsFromCache = new GetEndpointRateLimitsParameters(twitterQuery.Url, RateLimitsSource.CacheOnly);
queryRateLimit = await _rateLimitCacheManager.GetQueryRateLimitAsync(getEndpointRateLimitsFromCache, twitterQuery.TwitterCredentials).ConfigureAwait(false);
}
var timeToWait = _rateLimitAwaiter.GetTimeToWaitFromQueryRateLimit(queryRateLimit, request.ExecutionContext);
twitterQuery.CredentialsRateLimits = credentialRateLimits;
twitterQuery.QueryRateLimit = queryRateLimit;
twitterQuery.DateWhenCredentialsWillHaveTheRequiredRateLimits = DateTime.UtcNow.Add(timeToWait);
}
public async Task QueryExecuted(string query, ITwitterCredentials credentials, int numberOfRequests = 1)
{
var getRateLimitsFromCacheParameters = new GetEndpointRateLimitsParameters(query, RateLimitsSource.CacheOnly);
var rateLimit = await _rateLimitCacheManager.GetQueryRateLimit(getRateLimitsFromCacheParameters, credentials).ConfigureAwait(false);
if (rateLimit != null)
{
var newRemainingValue = Math.Max(rateLimit.Remaining - numberOfRequests, 0);
rateLimit.Remaining = newRemainingValue;
}
}
private async Task QueryCompleted(
ITwitterRequest request,
ITwitterResponse twitterResponse,
IRateLimitUpdater rateLimitUpdater)
{
if (request.ExecutionContext.RateLimitTrackerMode != RateLimitTrackerMode.None)
{
var credentials = request.Query.TwitterCredentials;
var rateLimitHeaders = twitterResponse.Headers
.Where(kvp => kvp.Key.StartsWith("x-rate-limit-"))
.ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
await rateLimitUpdater.QueryExecutedAsync(request.Query.Url, credentials, rateLimitHeaders).ConfigureAwait(false);
if (request.Query.QueryRateLimit == null)
{
var getEndPointParameters = new GetEndpointRateLimitsParameters(request.Query.Url);
request.Query.QueryRateLimit = await _rateLimitCacheManager.GetQueryRateLimitAsync(getEndPointParameters, credentials).ConfigureAwait(false);
}
}
request.ExecutionContext.Events.RaiseAfterExecutingQuery(new AfterExecutingQueryEventArgs(request.Query, twitterResponse.Content, twitterResponse.Headers));
}
