使用Redis实现简单限流

   限流是可以避免外部请求频繁的对系统施压,而且还可以控制用户行为,避免垃圾请求。

下面是利用Redis来控制用户在规定的时间只能请求N次

public boolean isExceedLimit(String userId, String key, int period, int maxCount)
      throws IOException {
    String newKey = userId + ":" + key;
    long nowDate = System.currentTimeMillis();
    Pipeline pipeline = jedis.pipelined(); // 创建管道
    pipeline.multi();// 开启事务,保证指令的原子操作,不被其它事务打断
    pipeline.zadd(newKey, nowDate, nowDate + "");
    pipeline.zrangeByScore(newKey, 0, nowDate - period * 1000); // 删除[0, nowDate -period 
                                                                // * 1000]之前的记录
    Response<Long> count = pipeline.zcard(newKey);
    pipeline.expire(newKey, period + 1);// 每次设置都能保持key的过期时间
    pipeline.exec();// 执行
    pipeline.close();
    return count.get() <= maxCount;
  }


@Test
public void test() {

    try {
      // 限定1s内操作不得超过3次
      for (int i = 0; i < 10; i++) {
        System.out.println(isExceedLimit("2333", "test", 1, 3));
      }
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值