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

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