這篇文章主要為大家分析了Go中是怎么實(shí)現(xiàn)用戶的每日限額的相關(guān)知識(shí)點(diǎn),內(nèi)容詳細(xì)易懂,操作細(xì)節(jié)合理,具有一定參考價(jià)值。如果感興趣的話,不妨跟著跟隨小編一起來(lái)看看,下面跟著小編一起深入學(xué)習(xí)“Go中是怎么實(shí)現(xiàn)用戶的每日限額”的知識(shí)吧。
創(chuàng)新互聯(lián)-專業(yè)網(wǎng)站定制、快速模板網(wǎng)站建設(shè)、高性價(jià)比永州網(wǎng)站開發(fā)、企業(yè)建站全套包干低至880元,成熟完善的模板庫(kù),直接使用。一站式永州網(wǎng)站制作公司更省心,省錢,快速模板網(wǎng)站建設(shè)找我們,業(yè)務(wù)覆蓋永州地區(qū)。費(fèi)用合理售后完善,10多年實(shí)體公司更值得信賴。
Go 中實(shí)現(xiàn)用戶的每日限額(比如一天只能領(lǐng)三次福利)
如果你寫一個(gè) bug 管理系統(tǒng),用了這個(gè) PeriodLimit
你就可以限制每個(gè)測(cè)試人員每天只能給你提一個(gè) bug。工作是不是就輕松很多了?
如今微服務(wù)架構(gòu)大行其道本質(zhì)原因是因?yàn)橐档拖到y(tǒng)的整體復(fù)雜度,將系統(tǒng)風(fēng)險(xiǎn)均攤到子系統(tǒng)從而最大化保證系統(tǒng)的穩(wěn)定性,通過(guò)領(lǐng)域劃分拆成不同的子系統(tǒng)后各個(gè)子系統(tǒng)能獨(dú)立的開發(fā)、測(cè)試、發(fā)布,研發(fā)節(jié)奏和效率能明顯提高。
但同時(shí)也帶來(lái)了問(wèn)題,比如:調(diào)用鏈路過(guò)長(zhǎng),部署架構(gòu)復(fù)雜度提升,各種中間件需要支持分布式場(chǎng)景。為了確保微服務(wù)的正常運(yùn)行,服務(wù)治理就不可或缺了,通常包括:限流,降級(jí),熔斷。
其中限流指的是針對(duì)接口調(diào)用頻率進(jìn)行限制,以免超出承載上限拖垮系統(tǒng)。
下面主要講解固定時(shí)間窗口限流算法。
從某個(gè)時(shí)間點(diǎn)開始每次請(qǐng)求過(guò)來(lái)請(qǐng)求數(shù)+1,同時(shí)判斷當(dāng)前時(shí)間窗口內(nèi)請(qǐng)求數(shù)是否超過(guò)限制,超過(guò)限制則拒絕該請(qǐng)求,然后下個(gè)時(shí)間窗口開始時(shí)計(jì)數(shù)器清零等待請(qǐng)求。
優(yōu)點(diǎn)
實(shí)現(xiàn)簡(jiǎn)單高效,特別適合用來(lái)限制比如一個(gè)用戶一天只能發(fā)10篇文章、只能發(fā)送5次短信驗(yàn)證碼、只能嘗試登錄5次等場(chǎng)景,實(shí)際業(yè)務(wù)中此類場(chǎng)景非常多見。
缺點(diǎn)
固定時(shí)間窗口限流的缺點(diǎn)在于無(wú)法處理臨界區(qū)請(qǐng)求突發(fā)場(chǎng)景。
假設(shè)每 1s 限流 100 次請(qǐng)求,用戶在中間 500ms 時(shí)開始 1s 內(nèi)發(fā)起 200 次請(qǐng)求,此時(shí) 200 次請(qǐng)求是可以全部通過(guò)的。這就和我們預(yù)期 1s 限流 100 次不合了,根源在于限流的細(xì)粒度太粗。
core/limit/periodlimit.go
go-zero 中使用 redis 過(guò)期時(shí)間來(lái)模擬固定時(shí)間窗口。
redis lua 腳本:
-- KYES[1]:限流器key-- ARGV[1]:qos,單位時(shí)間內(nèi)最多請(qǐng)求次數(shù)-- ARGV[2]:單位限流窗口時(shí)間-- 請(qǐng)求最大次數(shù),等于p.quotalocal limit = tonumber(ARGV[1])-- 窗口即一個(gè)單位限流周期,這里用過(guò)期模擬窗口效果,等于p.permitlocal window = tonumber(ARGV[2])-- 請(qǐng)求次數(shù)+1,獲取請(qǐng)求總數(shù)local current = redis.call("INCRBY",KYES[1],1)-- 如果是第一次請(qǐng)求,則設(shè)置過(guò)期時(shí)間并返回 成功if current == 1 then redis.call("expire",KYES[1],window) return 1-- 如果當(dāng)前請(qǐng)求數(shù)量小于limit則返回 成功elseif current < limit then return 1-- 如果當(dāng)前請(qǐng)求數(shù)量==limit則返回 最后一次請(qǐng)求elseif current == limit then return 2-- 請(qǐng)求數(shù)量>limit則返回 失敗else return 0end
固定時(shí)間窗口限流器定義
type ( // PeriodOption defines the method to customize a PeriodLimit. // go中常見的option參數(shù)模式 // 如果參數(shù)非常多,推薦使用此模式來(lái)設(shè)置參數(shù) PeriodOption func(l *PeriodLimit) // A PeriodLimit is used to limit requests during a period of time. // 固定時(shí)間窗口限流器 PeriodLimit struct { // 窗口大小,單位s period int // 請(qǐng)求上限 quota int // 存儲(chǔ) limitStore *redis.Redis // key前綴 keyPrefix string // 線性限流,開啟此選項(xiàng)后可以實(shí)現(xiàn)周期性的限流 // 比如quota=5時(shí),quota實(shí)際值可能會(huì)是5.4.3.2.1呈現(xiàn)出周期性變化 align bool } )
注意一下 align 參數(shù),align=true 時(shí)請(qǐng)求上限將會(huì)呈現(xiàn)周期性的變化。
比如quota=5時(shí)實(shí)際quota可能是5.4.3.2.1呈現(xiàn)出周期性變化
限流邏輯
其實(shí)限流邏輯在上面的 lua 腳本實(shí)現(xiàn)了,需要注意的是返回值
0:表示錯(cuò)誤,比如可能是 redis 故障、過(guò)載
1:允許
2:允許但是當(dāng)前窗口內(nèi)已到達(dá)上限,如果是跑批業(yè)務(wù)的話此時(shí)可以休眠 sleep 一下等待下個(gè)窗口(作者考慮的非常細(xì)致)
3:拒絕
// Take requests a permit, it returns the permit state. // 執(zhí)行限流 // 注意一下返回值: // 0:表示錯(cuò)誤,比如可能是redis故障、過(guò)載 // 1:允許 // 2:允許但是當(dāng)前窗口內(nèi)已到達(dá)上限 // 3:拒絕 func (h *PeriodLimit) Take(key string) (int, error) { // 執(zhí)行l(wèi)ua腳本 resp, err := h.limitStore.Eval(periodScript, []string{h.keyPrefix + key}, []string{ strconv.Itoa(h.quota), strconv.Itoa(h.calcExpireSeconds()), }) if err != nil { return Unknown, err } code, ok := resp.(int64) if !ok { return Unknown, ErrUnknownCode } switch code { case internalOverQuota: return OverQuota, nil case internalAllowed: return Allowed, nil case internalHitQuota: return HitQuota, nil default: return Unknown, ErrUnknownCode } }
這個(gè)固定窗口限流可能用來(lái)限制比如一個(gè)用戶一天只能發(fā)送5次驗(yàn)證碼短信,此時(shí)我們就需要跟中國(guó)時(shí)區(qū)對(duì)應(yīng)(GMT+8),并且其實(shí)限流時(shí)間應(yīng)該從零點(diǎn)開始,此時(shí)我們需要額外對(duì)齊(設(shè)置 align 為 true)。
// 計(jì)算過(guò)期時(shí)間也就是窗口時(shí)間大小 // 如果align==true // 線性限流,開啟此選項(xiàng)后可以實(shí)現(xiàn)周期性的限流 // 比如quota=5時(shí),quota實(shí)際值可能會(huì)是5.4.3.2.1呈現(xiàn)出周期性變化 func (h *PeriodLimit) calcExpireSeconds() int { if h.align { now := time.Now() _, offset := now.Zone() unix := now.Unix() + int64(offset) return h.period - int(unix%int64(h.period)) } return h.period }
這篇文章主要為大家分析了Go中是怎么實(shí)現(xiàn)用戶的每日限額的相關(guān)知識(shí)點(diǎn),內(nèi)容詳細(xì)易懂,操作細(xì)節(jié)合理,具有一定參考價(jià)值。如果感興趣的話,不妨跟著跟隨小編一起來(lái)看看,下面跟著小編一起深入學(xué)習(xí)“Go中是怎么實(shí)現(xiàn)用戶的每日限額”的知識(shí)吧。
名稱欄目:Go中是怎么實(shí)現(xiàn)用戶的每日限額
URL鏈接:http://bm7419.com/article32/ijhgsc.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供自適應(yīng)網(wǎng)站、定制網(wǎng)站、手機(jī)網(wǎng)站建設(shè)、、建站公司、移動(dòng)網(wǎng)站建設(shè)
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請(qǐng)盡快告知,我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如需處理請(qǐng)聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來(lái)源: 創(chuàng)新互聯(lián)