golang.org/x/time/rate
使用速率限制器
golang.org/x/time/rate 是 CoddyKit 上的免费 Go Academy 课时。 这是第 2 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 Go Academy 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 Go Academy 课程共包含 4 节课。
rate 软件包
golang.org/x/time/rate 提供可用于生产环境的令牌桶 Limiter。它是 Go 中实现限速的标准选择。
- 经过大量实践验证的实现
- Allow、Wait 和 Reserve 方法
- 支持上下文
创建限速器
rate.NewLimiter 接收一个速率(每秒的令牌数)和一个突发容量。rate.Limit 是表示每秒事件数的 float64 值。
limiter := rate.NewLimiter(rate.Limit(10), 5)rate.Every
如果您更习惯使用时间间隔而不是每秒速率,rate.Every 可以将一个时长转换为速率。每 200 毫秒发生一次事件,就相当于每秒发生 5 次。
limiter := rate.NewLimiter(rate.Every(200*time.Millisecond), 1)Allow
Allow 会报告当前是否有可用令牌。它会返回 true 并消耗一个令牌,或者立即返回 false。您可以使用它拒绝超出限制的请求。
if !limiter.Allow() {
fmt.Println("rejected: too many requests")
return
}Wait
Wait 会阻塞,直到令牌可用或上下文被取消。当您希望控制工作节奏而不是丢弃工作时,它非常适合。
if err := limiter.Wait(ctx); err != nil {
return err
}
doWork()Reserve
Reserve 会返回一个描述等待时长的 Reservation。调用 Delay() 可以了解等待时间,如果决定不继续,则调用 Cancel()。
r := limiter.Reserve()
if !r.OK() {
return
}
time.Sleep(r.Delay())注意:外部软件包
rate 软件包不属于标准库,因此需要运行 go get golang.org/x/time/rate 并使用模块。导入该软件包的代码片段仅用于说明,无法在此处直接运行。
可运行的标准库替代实现
由于(Since)rate 软件包属于外部软件包,这里提供一个使用标准库的节奏控制循环,通过每 200 毫秒计时一次来模拟 Wait。
package main
import (
"fmt"
"time"
)
func main() {
tick := time.NewTicker(200 * time.Millisecond)
defer tick.Stop()
for i := 0; i < 4; i++ {
<-tick.C
fmt.Println("processed event", i)
}
}运行时调整
您可以使用 SetLimit 和 SetBurst 动态更改限制。当配置重新加载或上游返回 429 响应,要求您降低速度时,这会非常方便。
limiter.SetLimit(rate.Limit(20))
limiter.SetBurst(10)每个客户端的限制器
请维护一个从客户端键到 *rate.Limiter 的映射,并使用互斥锁保护它。首次收到该键的请求时创建限制器,后续请求则复用它。
func getLimiter(m map[string]*rate.Limiter, key string) *rate.Limiter {
if l, ok := m[key]; ok {
return l
}
l := rate.NewLimiter(rate.Limit(5), 10)
m[key] = l
return l
}HTTP 中间件
请包装处理程序,使每个请求先调用 Allow()。如果返回 false,就以 HTTP 429 请求过多进行响应。这是服务器端规范的速率限制方式。
快速检查
请测试您对速率包的了解。
总结
您学会了 golang.org/x/time/rate:
- NewLimiter(rate, burst) 会创建令牌桶
- rate.Every 会将时间间隔转换为速率
- Allow 会拒绝请求,Wait 会控制节奏,Reserve 会报告延迟
- SetLimit 和 SetBurst 可在运行时调整参数
- 服务器可使用每个客户端的映射配合 HTTP 429
常见问题解答
「golang.org/x/time/rate」课时是免费的吗?
是的 — 「golang.org/x/time/rate」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 Go Academy 课程的其余内容,请升级到 CoddyKit PRO。 Go Academy 课程共包含 4 节课。
「golang.org/x/time/rate」这节课中我会学到什么?
使用速率限制器 你通过在浏览器中直接运行的动手代码来练习 Go Academy,全天候 AI 导师会在你学习这节课的过程中回答你的问题。
学习 Go Academy 需要有经验吗?
无需任何先前经验。CoddyKit 上的 Go Academy 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 2 节课,共 4 节。
「golang.org/x/time/rate」课时需要多长时间?
大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。
我能在这节 Go Academy 课中编写并运行代码吗?
能。每节 Go Academy 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。
此课程中的所有课时
- 令牌桶概念
- golang.org/x/time/rate
- 限制 Goroutine
- 退避与重试