SpringBoot基于redis zset实现滑动窗口限流

这篇文章距离上次修改已过238天，其中的内容可能已经有所变动。

在Spring Boot中，可以使用Redis的有序集合（zset）来实现滑动窗口限流。以下是一个简单的实现示例：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

@Component
public class SlidingWindowLimiter {

    @Autowired
    private StringRedisTemplate redisTemplate;

    private static final String KEY_PREFIX = "rate_limit:";
    private static final int WINDOW_SIZE = 10; // 滑动窗口的大小，单位是秒
    private static final long TIME_BUCKET_SIZE = 1000L; // 每个时间桶的毫秒数

    public boolean isAllowed(String key, int limit) {
        long now = System.currentTimeMillis();
        ZSetOperations<String, String> zSetOps = redisTemplate.opsForZSet();

        // 移除过期的元素
        zSetOps.removeRangeByScore(KEY_PREFIX + key, 0, now - WINDOW_SIZE * 1000);

        // 添加新的元素并设置过期时间
        String value = String.valueOf(now);
        zSetOps.add(KEY_PREFIX + key, value, now);
        redisTemplate.expire(KEY_PREFIX + key, WINDOW_SIZE, TimeUnit.SECONDS);

        // 计算当前窗口的访问次数
        long count = zSetOps.count(KEY_PREFIX + key, now - WINDOW_SIZE * 1000, now);

        // 如果超过限制，则不允许通过
        return count < limit;
    }
}

使用方法：

@Autowired
private SlidingWindowLimiter limiter;

public void someMethod() {
    String key = "resource_key"; // 需要限流的资源键
    int limit = 100; // 时间窗口内允许的最大请求次数

    if (limiter.isAllowed(key, limit)) {
        // 如果允许通过，则执行业务逻辑
    } else {
        // 如果不允许通过，则返回错误或执行其他操作
    }
}

这段代码定义了一个SlidingWindowLimiter组件，它使用Redis的有序集合（zset）来跟踪每个键的访问次数，并且在每次请求时检查是否超过了限制。滑动窗口的大小是10秒，并且每次请求都会更新这个窗口。如果请求次数超过了限定的限制，则不允许通过。这个实现可以很容易地集成到Spring Boot应用程序中，并用于限制对特定资源的访问频率。

SpringBoot基于redis zset实现滑动窗口限流

评论已关闭

推荐阅读