在实际开发中,限流(Rate Limiting)是一种保护服务、避免接口被恶意刷流的常见技术。常用的限流算法有令牌桶、漏桶、固定窗口、滑动窗口等。由于Redis具备高性能和原子性操作,常常被用来实现分布式限流。
下面给出使用Golang结合Redis实现简单限流的几种常见方式(以“固定窗口计数”和“滑动窗口”为例)。使用的Go Redis库为go-redis。
1. 固定窗口计数(Fixed Window Counter)
在单位时间窗口(如1秒、1分钟)内计数,超过阈值则限流。
伪代码思路:
- 用Redis的
INCR
自增某个key记录当前窗口内的访问次数。 - 使用
EXPIRE
设置Key过期时间为窗口大小。 - 如果计数超过阈值,说明被限流。
Go代码示例:
package main
import (
"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 {