登录
首页 >  Golang >  Go教程

Go rate.Limiter:用令牌桶把入口流量变成可承受的速度

来源:Golang学习网专题原创

时间:2026-06-08 20:06:00 365浏览 收藏

所属专题:Go 并发治理实战

限流不是为了拒绝用户,而是为了保护系统在高峰期仍然可用。x/time/rate 提供的令牌桶可以同时表达平均速率和突发容量,适合大多数 Go 服务入口限流。

Allow 适合快速拒绝

Allow 不阻塞,拿不到令牌就返回 false,适合 API 网关、写接口和明确要求快速失败的场景。

Wait 适合短暂排队

Wait 会等待令牌,但必须传入 context,避免无限排队。对于用户请求,排队时间不能超过接口总预算。

按用户或租户分桶

全局限流只能保护实例,不能阻止单个用户占满配额。更常见的做法是按用户、IP 或租户维护 limiter,并给 limiter 加过期清理。

代码示例

limiter := rate.NewLimiter(rate.Limit(100), 200)

func middleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if !limiter.Allow() {
            http.Error(w, "too many requests", http.StatusTooManyRequests)
            return
        }
        next.ServeHTTP(w, r)
    })
}

上线检查

  • 限流返回明确的 429。
  • Burst 不能大到压垮下游。
  • 按用户维度清理长时间不用的 limiter。
声明:本文转载于:Golang学习网专题原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>