/// <summary>
/// Add rate limit service
/// </summary>
/// <param name="builder"></param>
/// <param name="algorithm"></param>
/// <param name="error"></param>
/// <param name="interceptor"></param>
/// <param name="specialRule"></param>
/// <returns></returns>
public static IServiceCollection AddRateLimit(this IServiceCollection builder, IAlgorithm algorithm, HttpErrorResponse error = null, HttpInvokeInterceptor interceptor = null, RateLimitSpecialRule specialRule = null)
{
if (algorithm == null)
{
throw new ArgumentNullException("The algorithm service is not registered, please use 'AddRateLimit' in 'ConfigureServices' method.");
}
if (error == null)
{
error = new HttpErrorResponse()
{
HttpStatusCode = 429,
BuildHttpContent = (context, checkResult) =>
{
return("too many requests");
}
};
}
if (interceptor == null)
{
interceptor = new HttpInvokeInterceptor();
}
if (specialRule == null)
{
specialRule = new RateLimitSpecialRule();
}
builder.AddSingleton <IAlgorithm>(algorithm);
builder.AddSingleton <HttpErrorResponse>(error);
builder.AddSingleton <HttpInvokeInterceptor>(interceptor);
builder.AddSingleton <RateLimitSpecialRule>(specialRule);
return(builder);
}
/// <summary>
/// Create a new instance
/// </summary>
/// <param name="next"></param>
/// <param name="algorithm"></param>
/// <param name="error"></param>
/// <param name="interceptor"></param>
/// <param name="specialRule"></param>
public RateLimitMiddleware(RequestDelegate next, IAlgorithm algorithm, HttpErrorResponse error, HttpInvokeInterceptor interceptor, RateLimitSpecialRule specialRule)
{
_next = next;
_algorithm = algorithm;
_error = error;
_interceptor = interceptor;
_specialRule = specialRule;
}