如何在Gin框架中实现请求限流功能-Golang-PHP中文网

gin 默认不带限流因其定位为轻量路由框架，限流需自行实现或引入第三方方案；单机用golang.org/x/time/rate，分布式用redis+lua脚本保证原子性。

如何在gin框架中实现请求限流功能

为什么 Gin 默认不带限流，而要用第三方库

Gin 本身是轻量 HTTP 路由框架，不内置限流逻辑——它连中间件都只提供注册机制，具体策略（如令牌桶、漏桶、滑动窗口）得自己实现或引入成熟方案。硬写一个 time.Sleep 或简单计数器，在并发场景下容易因竞态或时钟漂移失效，且无法跨进程共享状态。生产环境通常选 golang.org/x/time/rate（单机）或 redis + Lua（分布式），前者够用且无外部依赖。

用 `rate.Limiter` 实现每秒最多 10 次请求的中间件

核心是把 rate.NewLimiter 实例挂到 Gin 的 gin.Context 上，每次请求调用 Allow 或 Wait 判断是否放行。注意：不能每个请求都新建 Limiter，必须复用；也不能直接在 func(c *gin.Context) 里初始化，否则所有路由共用同一个限流器（除非你真想全局限流）。

常见错误现象：429 Too Many Requests 响应没返回、限流阈值始终不准、高并发下 panic（因 rate.Limiter 非 goroutine-safe 初始化）。

按路由路径单独配置限流器：用 sync.Map 缓存不同 c.FullPath() 对应的 *rate.Limiter
使用 limiter.Wait(c.Request.Context()) 而非 Allow()，避免因时钟误差导致瞬时超发
在 Wait 返回 error 时显式写入 c.AbortWithStatusJSON(429, gin.H{"error": "rate limited"})，否则 Gin 会继续执行后续 handler
若需区分用户/IP，把 key 改成 c.ClientIP() + ":" + c.FullPath()，但注意 IPv6 地址长度和冒号冲突问题

func RateLimitMiddleware(rps int) gin.HandlerFunc {
    limiter := rate.NewLimiter(rate.Limit(rps), rps)
    return func(c *gin.Context) {
        if err := limiter.Wait(c.Request.Context()); err != nil {
            c.AbortWithStatusJSON(429, gin.H{"error": "rate limited"})
            return
        }
    }
}

当需要分布式限流时，为什么 Redis + Lua 是更稳的选择

单机 rate.Limiter 在多实例部署下完全失效——每个进程维护自己的桶，总请求量轻松突破阈值。Redis 的原子性 + Lua 脚本能保证“读-判-写”一步完成，且支持精确时间窗口（如每分钟 60 次）。但要注意：redis.SetNX 类命令无法实现滑动窗口，必须用 INCR + EXPIRE 组合，或直接用 Redis 6.2+ 的 INCRBYEX。

Gin框架 1.9.0

Gin框架 1.9.0版本源码包下载，版本号 1.9.0，适合需要 sonic JSON 支持、路由修复和内容协商改进的 Go Web 开发场景。

下载

性能影响明显：每次请求多一次 Redis RTT，若 Redis 延迟高或连接池不足，反而拖慢整体响应。建议设置超时（context.WithTimeout）并降级为本地限流（fallback）。

脚本中用 KEYS[1] 表示限流 key（如 "rl:ip:192.168.1.1:/api/v1/users"），ARGV[1] 传窗口秒数，ARGV[2] 传最大请求数
不要用 GETSET，它无法处理过期逻辑，INCR 后立刻 EXPIRE 才能保证窗口重置
若用 github.com/go-redis/redis/v9，记得用 script.Load().Run() 预加载脚本提升性能