0Pricing
Go Academy · 课时

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 反馈 — 无需本地设置。

此课程中的所有课时

  1. 令牌桶概念
  2. golang.org/x/time/rate
  3. 限制 Goroutine
  4. 退避与重试
← 返回 Go Academy