網站首頁 編程語言 正文
假設我們要做一個業務需求,這個需求就是限制用戶的訪問頻次。比如1分鐘內只能訪問20次,10分鐘內只能訪問200次。因為是用戶維度的場景,性能肯定是要首先考慮,那么適合這個場景的非Redis莫屬。
最簡單的實現,莫過于只是用incr進行計數操作,于是有了下面的代碼:
long count = redisTemplate.opsForValue().increment("user:1:60"); if (count > maxLimitCount) { throw new LimitException("訪問太頻繁"); } count = redisTemplate.opsForValue().increment("user:1:600"); if (count > maxLimitCount) { throw new LimitException("訪問太頻繁"); }
來,我們對上面這段代碼解讀一下。需求有2個時間維度的限制,所以這邊基于用戶和時間維度構建了Redis的Key。然后對每個Key進行計數,計數后的結果用于跟限制的值進行判斷,如果超出了限制的值就拋出異常。
假設限制的時間場景有10個呢?那上面的代碼是不是得寫10遍才可以。有人可能會說,這還不簡單嗎?循環呀,循環確實能夠解決這個問題。但是大家有沒有去思考,這是用戶維度的請求,如果每個請求里面都去操作10次Redis的話,這耗時至少也得10來毫秒吧。所以問題在這,并不是說這個邏輯實現的有問題。
那我們就改成批量的吧,用pipeline來批量執行。
redisTemplate.execute(new RedisCallback() { @Override public Long doInRedis(RedisConnection connection) throws DataAccessException { connection.openPipeline(); connection.incr("user:1:60".getBytes()); connection.incr("user:1:600".getBytes()); onnection.closePipeline(); return null; } });
用pipeline也有一個問題,那就是拿不到返回值,也就只能增加,但是沒辦法判斷是否超過了限制的閥值。
所以需要在第一步先查詢下,用查到的值進行判斷,這樣也就是只需要和Redis交互兩次就可以了。
上面的代碼在單節點下沒問題,但是如果在集群下,其實每個Key都可能分配到不同的節點上去,只不過是底層幫你屏蔽掉了細節,并發執行,拿到了所有結果后合并返回的。所以我們需要讓所有的Key都路由到一個節點上,本來就是用戶維度的,直接使用userId路由即可。
這個時候Redis的HashTag功能就排上用場了,將Key user:1:600改寫成user:{1}:600 。
雖然已經優化了,但是還是要發起兩次網絡請求才能完成這個邏輯,有沒有可能再進一步優化下呢?一次請求行不行。
這個時候要放大招了,Lua腳本走起,將所有邏輯都放入Lua腳本中,一次網絡交互即可完成。
local current current = redis.call("incr",KEYS[1]) if current == 1 then redis.call("expire",KEYS[1],1) end if current > ARGV[1] return 1 end return 0
上面腳本演示了如何對一個Key進行處理,返回1表示限流,返回0表示通過。不過使用lua腳本的時候要注意,某些云服務的Redis會對腳本進行校驗,像Redis的Key不能使用變量,必須用KEYS[下標]的方式,所以這里操作多個Key還不能用循環,代碼得寫多遍,這是一個惡心的點。
總結
原文鏈接:https://mp.weixin.qq.com/s/P4vh-n9lssuAAW-_t6Pc6A
相關推薦
- 2023-07-16 uniapp 微信小程序獲取當前位置的坐標
- 2022-12-29 python解決循環依賴的問題分析_python
- 2022-04-18 Android?app本地切換logo和名稱_Android
- 2023-10-14 uniapp 將base64字符串保存為圖片、Word、Excel、音頻、視頻等文件
- 2023-05-29 scipy稀疏數組coo_array的實現_python
- 2022-06-11 python中Event實現線程間同步介紹_python
- 2023-11-12 python enumerate函數用法
- 2022-11-09 React特征學習Form數據管理示例詳解_React
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支