当前位置:网站首页>net6的Web MVC项目实现限流功能

net6的Web MVC项目实现限流功能

2022-08-11 05:24:00 Three Big Stones

原理:利用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、这样限流的功能算是启用了。

原网站

版权声明
本文为[Three Big Stones]所创,转载请带上原文链接,感谢
https://blog.csdn.net/dandingwangzi/article/details/125445681