在 Go 中實(shí)現(xiàn) rate limiting 可通過官方包 golang.org/x/time/rate 實(shí)現(xiàn)，1. 使用 rate.NewLimiter 創(chuàng)建限流器，控制單位時(shí)間請(qǐng)求次數(shù)和突發(fā)請(qǐng)求數(shù)；2. 在 HTTP 服務(wù)中可封裝為中間件進(jìn)行全局限流；3. 分布式環(huán)境下需結(jié)合 Redis 和 Lua 腳本實(shí)現(xiàn)跨節(jié)點(diǎn)限流。示例代碼展示了如何在循環(huán)和 HTTP 接口中使用限流邏輯，確保系統(tǒng)穩(wěn)定性和請(qǐng)求可控性。
Rate limiting 是保護(hù)系統(tǒng)不被過多請(qǐng)求壓垮的關(guān)鍵手段，尤其在 API 服務(wù)中非常常見。Go 語言因?yàn)樾阅芎?、并發(fā)能力強(qiáng)，經(jīng)常用來做后端服務(wù)，所以實(shí)現(xiàn) rate limiting 的場(chǎng)景也比較多。
下面通過一個(gè)簡單的例子來說明怎么在 Go 中實(shí)現(xiàn)基本的 rate limiting。
使用 golang.org/x/time/rate 包
Go 官方提供的 x/time/rate 包是一個(gè)輕量級(jí)但功能完整的限流庫。它基于“令牌桶”算法，可以控制單位時(shí)間內(nèi)允許執(zhí)行的操作次數(shù)。
import (
"fmt"
"time"
"golang.org/x/time/rate"
)
func main() {
// 每秒允許 3 個(gè)請(qǐng)求，burst 表示最多能容忍的突發(fā)請(qǐng)求數(shù)
limiter := rate.NewLimiter(3, 1)
for i := 0; i < 10; i {
if limiter.Allow() {
fmt.Println("Request", i 1, "processed")
} else {
fmt.Println("Request", i 1, "rejected")
}
time.Sleep(200 * time.Millisecond)
}
}在這個(gè)例子里：
rate.NewLimiter(3, 1)表示每秒最多處理 3 個(gè)請(qǐng)求，最多允許一次突發(fā) 1 個(gè)請(qǐng)求。- 如果請(qǐng)求頻率太快，就會(huì)被拒絕。
這個(gè)包適合用在單機(jī)服務(wù)里，比如 HTTP 接口限流。
在 HTTP 服務(wù)中加入限流中間件
如果你正在寫一個(gè) HTTP 服務(wù)，可以把限流邏輯封裝成中間件，這樣每個(gè)請(qǐng)求都會(huì)先經(jīng)過限流判斷。
func limit(next http.HandlerFunc) http.HandlerFunc {
limiter := rate.NewLimiter(5, 2) // 每秒允許 5 個(gè)請(qǐng)求，最多同時(shí)處理 2 個(gè)突發(fā)請(qǐng)求
return func(w http.ResponseWriter, r *http.Request) {
if !limiter.Allow() {
http.Error(w, "Too many requests", http.StatusTooManyRequests)
return
}
next.ServeHTTP(w, r)
}
}然后注冊(cè)中間件到你的 handler 上：
http.HandleFunc("/api", limit(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, "OK")
}))這樣就可以輕松給接口加上限流功能了。
分布式環(huán)境下需要額外考慮
上面的例子都是單實(shí)例下的限流方式。如果你的服務(wù)部署了多個(gè)節(jié)點(diǎn)，或者用了負(fù)載均衡，就需要考慮分布式限流。
常見的做法是借助 Redis Lua 腳本，使用滑動(dòng)窗口或令牌桶的分布式版本。比如：
- 用 Redis 記錄每個(gè)用戶/IP 的請(qǐng)求次數(shù)
- 結(jié)合 Lua 實(shí)現(xiàn)原子操作，防止并發(fā)問題
- 可以選擇按 IP、用戶 ID 或 API Key 來做維度限流
這部分稍微復(fù)雜一點(diǎn)，但也是生產(chǎn)環(huán)境中常用的方案。
總的來說，在 Go 中做 rate limiting 并不難，尤其是有官方包支持的情況下。關(guān)鍵是要根據(jù)業(yè)務(wù)需求選擇合適的限流策略和實(shí)現(xiàn)方式。如果是單實(shí)例服務(wù)，直接用 rate.Limiter 就夠用了；如果涉及多實(shí)例或分布式系統(tǒng)，就需要引入外部存儲(chǔ)配合限流算法。
基本上就這些。
以上是進(jìn)行限制示例的詳細(xì)內(nèi)容。更多信息請(qǐng)關(guān)注PHP中文網(wǎng)其他相關(guān)文章！
熱AI工具
Undress AI Tool
免費(fèi)脫衣服圖片
Undresser.AI Undress
人工智能驅(qū)動(dòng)的應(yīng)用程序，用于創(chuàng)建逼真的裸體照片
AI Clothes Remover
用于從照片中去除衣服的在線人工智能工具。
Clothoff.io
AI脫衣機(jī)
Video Face Swap
使用我們完全免費(fèi)的人工智能換臉工具輕松在任何視頻中換臉！
熱門文章
熱工具
記事本++7.3.1
好用且免費(fèi)的代碼編輯器
SublimeText3漢化版
中文版，非常好用
禪工作室 13.0.1
功能強(qiáng)大的PHP集成開發(fā)環(huán)境
Dreamweaver CS6
視覺化網(wǎng)頁開發(fā)工具
SublimeText3 Mac版
神級(jí)代碼編輯軟件(SublimeText3)
如何在GO中創(chuàng)建緩沖頻道？ （例如，make（chan int，10））
Jun 20, 2025 am 01:07 AM
在Go中創(chuàng)建緩沖通道只需在make函數(shù)中指定容量參數(shù)即可。緩沖通道允許發(fā)送操作在沒有接收者時(shí)暫存數(shù)據(jù)，只要未超過指定容量，例如ch:=make(chanint,10)創(chuàng)建了一個(gè)可存儲(chǔ)最多10個(gè)整型值的緩沖通道；與無緩沖通道不同，發(fā)送數(shù)據(jù)時(shí)不會(huì)立即阻塞，而是將數(shù)據(jù)暫存于緩沖區(qū)中，直到被接收者取走；使用時(shí)需注意：1.容量設(shè)置應(yīng)合理以避免內(nèi)存浪費(fèi)或頻繁阻塞；2.需防止緩沖區(qū)無限堆積數(shù)據(jù)導(dǎo)致內(nèi)存問題；3.可用chanstruct{}類型傳遞信號(hào)以節(jié)省資源；常見場(chǎng)景包括控制并發(fā)數(shù)量、生產(chǎn)者-消費(fèi)者模型及異
如何在GO中的結(jié)構(gòu)實(shí)例上調(diào)用方法？
Jun 24, 2025 pm 03:17 PM
在Go語言中，調(diào)用結(jié)構(gòu)體方法需先定義結(jié)構(gòu)體和綁定接收者的方法，使用點(diǎn)號(hào)訪問。定義結(jié)構(gòu)體Rectangle后，可通過值接收者或指針接收者聲明方法；1.使用值接收者如func(rRectangle)Area()int，通過rect.Area()直接調(diào)用；2.若需修改結(jié)構(gòu)體，應(yīng)使用指針接收者如func(r*Rectangle)SetWidth(...)，Go會(huì)自動(dòng)處理指針與值的轉(zhuǎn)換；3.嵌入結(jié)構(gòu)體時(shí)，內(nèi)嵌結(jié)構(gòu)體的方法會(huì)被提升，可直接通過外層結(jié)構(gòu)體調(diào)用；4.Go無需強(qiáng)制使用getter/setter，字
GO中的接口是什么？如何定義它們？
Jun 22, 2025 pm 03:41 PM
在Go語言中，接口是一種定義行為而不指定實(shí)現(xiàn)方式的類型。接口由方法簽名組成，任何實(shí)現(xiàn)這些方法的類型都自動(dòng)滿足該接口。例如，定義一個(gè)Speaker接口包含Speak()方法，則所有實(shí)現(xiàn)該方法的類型均可視為Speaker。接口適用于編寫通用函數(shù)、抽象實(shí)現(xiàn)細(xì)節(jié)和測(cè)試中使用mock對(duì)象。定義接口使用interface關(guān)鍵字并列出方法簽名，無需顯式聲明類型實(shí)現(xiàn)了接口。常見用例包括日志、格式化、不同數(shù)據(jù)庫或服務(wù)的抽象，以及通知系統(tǒng)等。例如，Dog和Robot類型均可實(shí)現(xiàn)Speak方法，并傳遞給同一個(gè)Anno
如何在GO中使用字符串軟件包中的字符串函數(shù)？ （例如len（），strings.contains（），strings.index（），strings.replaceall（））
Jun 20, 2025 am 01:06 AM
在Go語言中，字符串操作主要通過strings包和內(nèi)置函數(shù)實(shí)現(xiàn)。1.strings.Contains()用于判斷字符串是否包含子串，返回布爾值；2.strings.Index()可查找子串首次出現(xiàn)的位置，若不存在則返回-1；3.strings.ReplaceAll()能替換所有匹配的子串，還可通過strings.Replace()控制替換次數(shù)；4.len()函數(shù)用于獲取字符串字節(jié)數(shù)長度，但處理Unicode時(shí)需注意字符與字節(jié)的區(qū)別。這些功能常用于數(shù)據(jù)過濾、文本解析及字符串處理等場(chǎng)景。
如何使用IO軟件包在GO中使用輸入和輸出流？
Jun 20, 2025 am 11:25 AM
TheGoiopackageprovidesinterfaceslikeReaderandWritertohandleI/Ooperationsuniformlyacrosssources.1.io.Reader'sReadmethodenablesreadingfromvarioussourcessuchasfilesorHTTPresponses.2.io.Writer'sWritemethodfacilitateswritingtodestinationslikestandardoutpu
我如何使用時(shí)間軟件包來處理GO的時(shí)間和持續(xù)時(shí)間？
Jun 23, 2025 pm 11:21 PM
Go的time包提供了處理時(shí)間和持續(xù)時(shí)間的功能，包括獲取當(dāng)前時(shí)間、格式化日期、計(jì)算時(shí)間差、處理時(shí)區(qū)、調(diào)度和休眠等操作。要獲取當(dāng)前時(shí)間，使用time.Now()獲取Time結(jié)構(gòu)體，并可通過Year()、Month()、Day()等方法提取具體時(shí)間信息；通過Format("2006-01-0215:04:05")可將時(shí)間格式化為字符串；計(jì)算時(shí)間差時(shí)，用Sub()或Since()獲取Duration對(duì)象，再通過Seconds()、Minutes()、Hours()轉(zhuǎn)換為對(duì)應(yīng)單位；添
我如何根據(jù)語句使用語句執(zhí)行代碼？
Jun 23, 2025 pm 07:02 PM
Ingo，ifstatementSexecuteCodeBasedonConconditions.1.BasicsStructurerunsablockifaconditionistrue，例如IFX> 10 {...}。2.Elseclausehan dlesfalseconditions，例如，else {...}。3。elseifchainsmultipleconditions，例如，elseifx == 10 {...}。4.variableInitializationInsideIndifif，l
去支持并發(fā)如何？
Jun 23, 2025 pm 12:37 PM
Gohandlesconcurrencyusinggoroutinesandchannels.1.GoroutinesarelightweightfunctionsmanagedbytheGoruntime,enablingthousandstorunco??ncurrentlywithminimalresourceuse.2.Channelsprovidesafecommunicationbetweengoroutines,allowingvaluestobesentandreceivedinas
