原理:利用MemoryCache服务组件,记录用户最后一次访问接口的时间,如果本次访问距离最后一次访问不超过1秒,提示用户访问过于频繁,否则,接口可以正常访问。然后利用拦截器可以拦截action的能力,自定义自己的限流器,结合MemoryCache组件,可以达到简单限流的目的。
1、实现限流器,主要是实现IAsyncActionFilter接口,给构造器注入缓存服务,并实现OnActionExecutionAsync方法,结合缓存功能,记录用户的ip信息作为key值,访问时间作为value。
public class RateLimitFilter : IAsyncActionFilter
{
private IMemoryCache memoryCache;
public RateLimitFilter(IMemoryCache memoryCache)
{
this.memoryCache = memoryCache;
}
public Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
{
string remoteIP = context.HttpContext.Connection.RemoteIpAddress.ToString();
string cacheKey = $"LastVisitTick_{remoteIP}";
long? lastTick = memoryCache.Get<long?>(cacheKey);
if (lastTick == null || Environment.TickCount64 - lastTick > 1000)
{
memoryCache.Set(cacheKey, Environment.TickCount64, TimeSpan.FromSeconds(10));//距离现在10秒过期
return next();
}
else//1秒之内只允许访问一次
{
context.Result = new ContentResult{ StatusCode = 429, Content="你访问太频繁了" };
return Task.CompletedTask;
}
}
}
2、注册限流器,实现了限流器之后,我们要把它注入到IOC容器中,具体代码如下:
public void ConfigureServices(IServiceCollection services)
{
services.AddControllers();
services.AddMvcCore(opt => {
opt.Filters.Add<RateLimitFilter>();//添加限流支持
opt.Filters.Add<TransactionScopeFilter>();//添加事务支持
opt.Filters.Add<ExceptionFilter>();//添加异常处理支持
opt.Filters.Add<LogFilter>();
});
services.AddDbContext<BookDbContext>(opt=> {
var connectString = this.Configuration.GetSection("DbConnectionStr").Value;
opt.UseSqlServer(connectString);
});
services.AddSwaggerGen(c =>
{
c.SwaggerDoc("v1", new OpenApiInfo { Title = "WebApplication1", Version = "v1" });
});
services.AddMemoryCache();//添加缓存服务
}
3、这样限流的功能算是启用了。