用操作筛选器实现请求限流器 filter
我们在操作筛选器中不仅可以在操作方法之前或者之后添加代码,还可以在满足条件的时候终止操作方法的执行。
为了避免恶意客户端频繁发送大量请求而消耗服务器资源,我们要实现 1 秒内只允许同一个IP访问一次请求。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | /// <summary> /// 请求限流器,1秒内同一个IP只允许访问一次 /// </summary> public class RateLimitFilter : IAsyncActionFilter { private readonly IMemoryCache memCache; public RateLimitFilter(IMemoryCache memCache) { this .memCache = memCache; } public Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next) { string removeIP = context.HttpContext.Connection.RemoteIpAddress.ToString(); //string removeIP = "192.168.0.8";//测试 string cacheKey = $ "LastVisitTick_{removeIP}" ; long ? lastTick = memCache.Get< long >(cacheKey); if (lastTick == null || Environment.TickCount64 - lastTick > 1000) { //首次访问或访问间隔超过1秒,则更新缓存时间 memCache.Set(cacheKey, Environment.TickCount64, TimeSpan.FromSeconds(10)); ; return next(); } else { //不调用next()方法,即终止操作执行,被调用的方法不再执行。 context.Result = new ContentResult { StatusCode = 429 }; return Task.CompletedTask; } } } |
注册服务
1 2 3 4 5 6 | builder.Services.Configure<MvcOptions>(options => { options.Filters.Add<RateLimitFilter>(); }); builder.Services.AddMemoryCache(); |
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律