redis 简单限流

camhan 2020-05-09

/**
 * @Description: 简单限流
 * @Author : myron
 * @Date : 2020-05-09 17:43
 **/
public class SimpleLimiter {
    private Jedis jedis;
    public SimpleLimiter(Jedis jedis) {
        this.jedis = jedis;
    }
    public boolean isActionAllowed(String userId, String actionKey, int period, int maxCount) {
        String key = String.format("hist:%s:%s", userId, actionKey);
        long nowTs = System.currentTimeMillis();
        Pipeline pipe = jedis.pipelined();
        pipe.multi();
        pipe.zadd(key, nowTs, "" + nowTs);
        pipe.zremrangeByScore(key, 0, nowTs - period * 1000);
        Response<Long> count = pipe.zcard(key);
        pipe.expire(key, period + 1);
        pipe.exec();
        try {
            pipe.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return count.get() <= maxCount;
    }
    public static void main(String[] args) {
        Jedis jedis = new Jedis();
        SimpleLimiter limiter = new SimpleLimiter(jedis);
        for(int i=0;i<20;i++) {
            //一分钟之内最多回复5次
            System.out.println(limiter.isActionAllowed("myron", "reply", 60, 5));
        }
    }
}

相关推荐