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

本文介绍了如何通过实现IAsyncActionFilter接口并利用MemoryCache服务,创建一个自定义限流过滤器,限制用户在一秒钟内只能访问一次API接口,防止频繁请求。在ASP.NET Core应用中,将限流器注册到IoC容器,并配置Swagger、数据库上下文等相关服务。

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值