Go语言怎么做API限流_Go语言令牌桶限流教程【实战】

Go 推荐用 golang.org/x/time/rate.Limiter 实现令牌桶限流，因其轻量、线程安全、支持突发控制；需复用实例、合理设置 rate 和 burst（如 rate.Every(100ms), burst=20），并在路由前统一拦截，避免误用 Allow() 或 per-request 新建。Go 用 golang.org/x/time/rate 做令牌桶限流最稳妥标准库没限流，但官方扩展包 rate.Limiter 就是为这个场景设计的——轻量、线程安全、支持突发流量控制。别自己手写计数器或用第三方轮子，除非你明确要绕过它的精度/阻塞逻辑。常见错误是把 rate.NewLimiter 当成“每秒放行 N 个请求”的开关，其实它默认允许突发（burst > 1），且 Allow() 不阻塞，Wait() 才会等令牌。用错就等于没限流。rate.NewLimiter(rate.Every(100*time.Millisecond), 3) 表示：平均 100ms 生成 1 个令牌，最多攒 3 个（burst=3）高并发下优先用 limiter.Wait(ctx)，避免 Allow() 返回 true 后实际被后端压垮HTTP handler 中不要每个请求 new 一个 Limiter，必须复用实例，否则 burst 完全失效HTTP middleware 里怎么嵌入限流逻辑限流必须在路由分发前做，否则中间件顺序错乱会导致部分 handler 被跳过。别在 handler 内部调 limiter.Wait()，那是给单个业务逻辑用的，不是 API 网关级控制。典型错误是把限流写在 http.HandleFunc 闭包里，导致每个路径都独立限流——而你真正想控的是总 QPS 或按 IP 分组。立即学习“go语言免费学习笔记（深入）”；用 http.Handler 包装原始 mux，统一拦截：http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if err := limiter.Wait(r.Context()); err != nil { http.Error(w, "too many requests", http.StatusTooManyRequests); return } next.ServeHTTP(w, r) })需要按 IP 限流？从 r.RemoteAddr 提取客户端 IP（注意 X-Forwarded-For 可伪造，生产需结合真实 IP 提取逻辑）别在 middleware 里用 time.Sleep 模拟等待——这会占 goroutine，limiter.Wait() 是非阻塞调度，更省资源rate.Limiter 的 burst 参数为什么不能设太小burst 是令牌桶的“水池大小”，设成 1 就退化成严格匀速限流，但真实流量天然有毛刺。用户连续点两次刷新、前端重试、CDN 预热都会触发瞬时高峰，burst=1 会让合法请求直接 429。 Vozo Vozo是一款强大的AI视频编辑工具，可以帮助用户轻松重写、配音和编辑视频。