【Redis】解析Redisson 限流器源码

Redis-son是一个Redis的Java客户端，它提供了一系列的注解和AOP的方式来简化开发者对Redis操作的编码。

以下是一个使用Redis-son的限流器的简单示例：

import com.redis.son.JedisSonicSession;
import com.redis.son.SonicEdits;
 
public class RedisRateLimiter {
    private JedisSonicSession session;
 
    public RedisRateLimiter(JedisSonicSession session) {
        this.session = session;
    }
 
    public boolean isAllowed(String userId, int maxRequests, int timeInterval) {
        String key = "rate_limiter:" + userId;
        // 使用SonicEdits的incr方法来递增键的值，并返回递增后的值
        long requests = session.sonicEdits().incr(key).ret(0L).get();
        // 如果递增后的值是1，说明这是第一次请求，直接返回true允许通过
        if (requests == 1) {
            return true;
        }
        // 如果递增后的值大于1，说明不是第一次请求，检查是否超过了限制
        if (requests > maxRequests) {
            // 如果超过了，则返回false不允许通过
            return false;
        }
        // 如果没有超过限制，则计算当前时间和最后一次请求的时间差
        long currentTime = System.currentTimeMillis() / 1000;
        long lastRequestTime = session.get(key + ":time").ret(0L).get();
        long elapsedTime = currentTime - lastRequestTime;
        // 如果经过的时间超过了设定的间隔，重置请求计数器
        if (elapsedTime > timeInterval) {
            session.set(key, 1);
            session.set(key + ":time", currentTime);
            return true;
        }
        // 否则返回false不允许通过
        return false;
    }
}

在这个例子中，我们创建了一个简单的本地限流器，它使用Redis的INCR命令来跟踪用户的请求数量，并且使用Redis的键的过期特性来重置请求计数器。这个限流器可以很容易地被集成到任何需要限流的应用程序中。