如何使用redis做限流(golang实现小样)
在实际开发中,限流(Rate Limiting)是一种保护服务、避免接口被恶意刷流的常见技术。常用的限流算法有令牌桶、漏桶、固定窗口、滑动窗口等。由于Redis具备高性能和原子性操作,常常被用来实现分布式限流。
下面给出使用Golang结合Redis实现简单限流的几种常见方式(以“固定窗口计数”和“滑动窗口”为例)。使用的Go Redis库为go-redis。
1. 固定窗口计数(Fixed Window Counter)
在单位时间窗口(如1秒、1分钟)内计数,超过阈值则限流。
伪代码思路:
- 用Redis的
INCR
自增某个key记录当前窗口内的访问次数。 - 使用
EXPIRE
设置Key过期时间为窗口大小。 - 如果计数超过阈值,说明被限流。
Go代码示例:
package mainimport ("context""fmt""github.com/redis/go-redis/v9""time"
)var ctx = context.Background()func FixedWindowRateLimit(client *redis.Client, key string, limit int, window time.Duration) (bool, error) {// 每次请求自增计数cnt, err := client.Incr(ctx, key).Result()if err != nil {