如何在Golang中实现基础API限流功能_Golang请求频率控制与处理示例_行业资讯

如何在Golang中实现基础API限流功能_Golang请求频率控制与处理示例

发布时间：2025-12-31 00:00

发布者：P粉602998670

浏览次数：

rate.Limiter基于令牌桶算法实现轻量可靠限流，支持全局复用、中间件集成、key级分桶及标准响应头，但单机限流无法解决分布式一致性问题。

用 `golang.org/x/time/rate` 实现每秒请求数限制

Go 官方扩展包 rate 是最轻量、最可靠的选择，它基于令牌桶算法，能平滑控制突发流量。不要自己手写计数器或时间窗口逻辑——容易漏掉并发竞争、时钟漂移、goroutine 泄漏等问题。

关键点：

rate.Limiter 是线程安全的，可全局复用，无需为每个请求新建
初始化时指定 limit（每秒令牌数）和 burst（桶容量），例如 rate.NewLimiter(5, 10) 表示“平均 5 QPS，最多允许 10 次瞬时并发”
使用 Limiter.Wait(ctx) 阻塞等待可用令牌，或 Limiter.Allow() 非阻塞判断——后者需手动返回 429

在 HTTP handler 中嵌入限流逻辑

限流应放在中间件层，而不是每个业务 handler 内硬编码。避免重复逻辑，也方便统一开关和指标上报。

常见错误：在 handler 开头直接调用 limiter.Wait(r.Context()) 却没处理 context 取消，导致超时请求仍占用令牌。正确做法是确保 Wait 能响应 cancel 或 timeout：

立即学习“go语言免费学习笔记（深入）”；

func rateLimitMiddleware(next http.Handler) http.Handler {
    limiter := rate.NewLimiter(rate.Limit(10), 5)
    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)
    })
}

注意：Allow() 不阻塞，适合无延迟敏感场景；若需精确配额（如按用户 ID 分桶），需配合 sync.Map 或外部存储做 key 级限流。