一段基于Redis-SortedSet的限流代码

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
[HttpGet]
      public async Task<ActionResult<string>> Get()
      {
          //限流周期:5秒
          int period = 5;
          //周期内请求数限制:5次,即限制每五秒只能请求五次
          int maxCount = 5;
          long uid = 123456;
          string action = "login";
          string key = string.Format("hist:{0}{1}", uid, action);
          long current = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
 
          var batch = db.CreateBatch();
          var tasks = new List<Task>();
          //加入SortedSet
          tasks.Add(batch.SortedSetAddAsync(key, current, current));
          //移除五秒之前的记录
          tasks.Add(batch.SortedSetRemoveRangeByScoreAsync(key, 0, current - period * 1000));
          //重置过期时间
          tasks.Add(batch.KeyExpireAsync(key, TimeSpan.FromSeconds(period + 1)));
          batch.Execute();
 
          //获取请求的次数
          long currentCount = await db.SortedSetLengthAsync(key);
          //返回是否大于限制次数
          bool res = currentCount <= maxCount;
 
          return res ? "未被限流,正常请求" : "超过请求频次,限制请求";
      }
posted @   丶Pz  阅读(326)  评论(0编辑  收藏  举报
编辑推荐:
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
阅读排行:
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 分享 3 个 .NET 开源的文件压缩处理库,助力快速实现文件压缩解压功能!
· Ollama——大语言模型本地部署的极速利器
· DeepSeek如何颠覆传统软件测试?测试工程师会被淘汰吗?
历史上的今天:
2016-09-05 ASP.NET SignalR 与 LayIM2.0 配合轻松实现Web聊天室(十二) 代码重构使用反射工厂解耦(一)缓存切换
2016-09-05 ASP.NET SignalR 与 LayIM2.0 配合轻松实现Web聊天室(十一) 代码重构使用反射工厂解耦
点击右上角即可分享
微信分享提示
+

"大爷常来玩呀"

微信支付