【Redis】---流量控制

Redis滑动窗口实现

public class SimpleRateLimiter {

   private Jedis jedis;

   public SimpleRateLimiter(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 count = pipe.zcard(key);

     pipe.expire(key, period + 1);

     pipe.exec();

     pipe.close();

    return count.get() <= maxCount;

}

  public static void main(String[] args) {

        Jedis jedis = new Jedis();

        SimpleRateLimiter limiter = new SimpleRateLimiter(jedis);

        for(int i=0;i<20;i++) {

           System.out.println(limiter.isActionAllowed("xx", "request", 60, 5));

      }

}

}