/// <summary>
/// Initializes a new instance of the <see cref="ThrottleProcesser"/> class.
/// By default, the <see cref="QuotaExceededResponseCode"/> property
/// is set to 429 (Too Many Requests).
/// </summary>
public ThrottleProcesser(
ThrottlePolicy policy,
IIpAddressParser ipAddressParser,
IPolicyRepository policyRepo = null,
IThrottleRepository throttleRepo = null)
{
Logger = new DefaultThrottleLogger();
QuotaExceededResponseCode = (HttpStatusCode)429;
processResult = new ThrottleProcessResult {
IsPass = true
};
ThrottleRepo = throttleRepo;
if (ThrottleRepo == null)
{
ThrottleRepo = new WebCacheThrottleRepository();
}
ThrottlingCore = new ThrottlingCore(ipAddressParser);
ThrottlingCore.ThrottleRepo = ThrottleRepo;
PolicyRepo = policyRepo;
if (PolicyRepo == null)
{
PolicyRepo = new WebCachePolicyRepository();
}
Policy = policy;
PolicyRepo.Save(ThrottleManager.GetPolicyKey(), policy);
}
public static ThrottlePolicy FromStore(IThrottlePolicyProvider provider)
{
var settings = provider.ReadSettings();
var whitelists = provider.AllWhitelists();
var rules = provider.AllRules();
var policy = new ThrottlePolicy(
perSecond: settings.LimitPerSecond,
perMinute: settings.LimitPerMinute,
perHour: settings.LimitPerHour,
perDay: settings.LimitPerDay,
perWeek: settings.LimitPerWeek);
policy.IpThrottling = settings.IpThrottling;
policy.ClientThrottling = settings.ClientThrottling;
policy.EndpointThrottling = settings.EndpointThrottling;
policy.StackBlockedRequests = settings.StackBlockedRequests;
policy.IpRules = new Dictionary <string, RateLimits>();
policy.ClientRules = new Dictionary <string, RateLimits>();
policy.EndpointRules = new Dictionary <string, RateLimits>();
policy.EndpointWhitelist = new List <string>();
policy.IpWhitelist = new List <string>();
policy.ClientWhitelist = new List <string>();
foreach (var item in rules)
{
var rateLimit = new RateLimits
{
PerSecond = item.LimitPerSecond,
PerMinute = item.LimitPerMinute,
PerHour = item.LimitPerHour,
PerDay = item.LimitPerDay,
PerWeek = item.LimitPerWeek
};
switch (item.PolicyType)
{
case ThrottlePolicyType.IpThrottling:
policy.IpRules.Add(item.Entry, rateLimit);
break;
case ThrottlePolicyType.ClientThrottling:
policy.ClientRules.Add(item.Entry, rateLimit);
break;
case ThrottlePolicyType.EndpointThrottling:
policy.EndpointRules.Add(item.Entry, rateLimit);
break;
}
}
if (whitelists != null)
{
policy.IpWhitelist.AddRange(whitelists.Where(x => x.PolicyType == ThrottlePolicyType.IpThrottling).Select(x => x.Entry));
policy.ClientWhitelist.AddRange(whitelists.Where(x => x.PolicyType == ThrottlePolicyType.ClientThrottling).Select(x => x.Entry));
policy.EndpointWhitelist.AddRange(whitelists.Where(x => x.PolicyType == ThrottlePolicyType.EndpointThrottling).Select(x => x.Entry));
}
return(policy);
}
