本篇文章為大家展示了Redis和Lua實現(xiàn)分布式限流器，內(nèi)容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細介紹希望你能有所收獲。

專注于為中小企業(yè)提供成都網(wǎng)站設(shè)計、網(wǎng)站建設(shè)服務(wù),電腦端+手機端+微信端的三站合一,更高效的管理,為中小企業(yè)洛川免費做網(wǎng)站提供優(yōu)質(zhì)的服務(wù)。我們立足成都，凝聚了一批互聯(lián)網(wǎng)行業(yè)人才，有力地推動了千余家企業(yè)的穩(wěn)健成長，幫助中小企業(yè)通過網(wǎng)站建設(shè)實現(xiàn)規(guī)模擴充和轉(zhuǎn)變。

原理

計數(shù)器算法是指在一段窗口時間內(nèi)允許通過的固定數(shù)量的請求, 比如10次/秒, 500次/30秒.

如果設(shè)置的時間粒度越細, 那么限流會更平滑.

實現(xiàn)

所使用的 Lua 腳本

-- 計數(shù)器限流
-- 此處支持的最小單位時間是秒, 若將 expire 改成 pexpire 則可支持毫秒粒度.
-- KEYS[1] string 限流的key
-- ARGV[1] int  限流數(shù)
-- ARGV[2] int  單位時間(秒)

local cnt = tonumber(redis.call("incr", KEYS[1]))

if (cnt == 1) then
 -- cnt 值為1說明之前不存在該值, 因此需要設(shè)置其過期時間
 redis.call("expire", KEYS[1], tonumber(ARGV[2]))
elseif (cnt > tonumber(ARGV[1])) then
 return -1
end 

return cnt

返回 -1 表示超過限流, 否則返回當(dāng)前單位時間已通過的請求數(shù)

key 可以但不限于以下的情況

• ip + 接口

• user_id + 接口

優(yōu)點

• 實現(xiàn)簡單

缺點

• 粒度不夠細的情況下, 會出現(xiàn)在同一個窗口時間內(nèi)出現(xiàn)雙倍請求數(shù)

注意

• 盡量保持時間粒度精細

場景分析

eg. 1000/3s 的限流

極端情況1:

第1秒請求數(shù) 10

第2秒請求數(shù) 10

第3秒請求數(shù) 980

第4秒請求數(shù) 900

第5秒請求數(shù) 100

第6秒請求數(shù) 0

此時注意第3~5秒內(nèi)的總請求數(shù)高達 1980

極端情況2:

第1秒請求數(shù) 1000

第2秒請求數(shù) 0

第3秒請求數(shù) 0

此時后續(xù)的第2~3秒會出現(xiàn)大量拒絕請求

令牌桶模式

原理

令牌桶的

1. 桶中保存有令牌, 存在上限, 且一開始是滿的

2. 每次請求都要消耗令牌(可根據(jù)不同請求消耗不同數(shù)量的令牌)

3. 每隔一段時間(固定速率)會往桶中放令牌

桶的實現(xiàn)還分為:

可預(yù)消費

提前預(yù)支令牌數(shù): 前人挖坑, 后人跳

不可預(yù)消費

令牌數(shù)不夠直接拒絕

實現(xiàn)

此處實現(xiàn)的不可預(yù)消費的令牌桶, 具體Lua代碼:

-- 令牌桶限流: 不支持預(yù)消費, 初始桶是滿的
-- KEYS[1] string 限流的key

-- ARGV[1] int  桶大容量
-- ARGV[2] int  每次添加令牌數(shù)
-- ARGV[3] int  令牌添加間隔(秒)
-- ARGV[4] int  當(dāng)前時間戳

local bucket_capacity = tonumber(ARGV[1])
local add_token = tonumber(ARGV[2])
local add_interval = tonumber(ARGV[3])
local now = tonumber(ARGV[4])

-- 保存上一次更新桶的時間的key
local LAST_TIME_KEY = KEYS[1].."_time";   
-- 獲取當(dāng)前桶中令牌數(shù)
local token_cnt = redis.call("get", KEYS[1]) 
-- 桶完全恢復(fù)需要的大時長
local reset_time = math.ceil(bucket_capacity / add_token) * add_interval;

if token_cnt then -- 令牌桶存在
 -- 上一次更新桶的時間
 local last_time = redis.call('get', LAST_TIME_KEY)
 -- 恢復(fù)倍數(shù)
 local multiple = math.floor((now - last_time) / add_interval)
 -- 恢復(fù)令牌數(shù)
 local recovery_cnt = multiple * add_token
 -- 確保不超過桶容量
 local token_cnt = math.min(bucket_capacity, token_cnt + recovery_cnt) - 1
 
 if token_cnt < 0 then
  return -1;
 end
 
 -- 重新設(shè)置過期時間, 避免key過期
 redis.call('set', KEYS[1], token_cnt, 'EX', reset_time)      
 redis.call('set', LAST_TIME_KEY, last_time + multiple * add_interval, 'EX', reset_time)
 return token_cnt
 
else -- 令牌桶不存在
 token_cnt = bucket_capacity - 1
 -- 設(shè)置過期時間避免key一直存在
 redis.call('set', KEYS[1], token_cnt, 'EX', reset_time);
 redis.call('set', LAST_TIME_KEY, now, 'EX', reset_time + 1); 
 return token_cnt 
end

令牌桶的關(guān)鍵是以下幾個參數(shù):

• 桶大容量

• 每次放入的令牌數(shù)

• 放入令牌的間隔時間

令牌桶的實現(xiàn)不會出現(xiàn)計數(shù)器模式中單位時間內(nèi)雙倍流量的問題.

上述內(nèi)容就是Redis和Lua實現(xiàn)分布式限流器，你們學(xué)到知識或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識儲備，歡迎關(guān)注創(chuàng)新互聯(lián)成都網(wǎng)站設(shè)計公司行業(yè)資訊頻道。

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務(wù)器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場景需求。

名稱欄目：Redis和Lua實現(xiàn)分布式限流器-創(chuàng)新互聯(lián)
鏈接URL：http://jinyejixie.com/article42/dijphc.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供Google、靜態(tài)網(wǎng)站、手機網(wǎng)站建設(shè)、網(wǎng)站制作、電子商務(wù)、外貿(mào)網(wǎng)站建設(shè)

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)