利用Redis实现高并发计数器\请求限制

>>强大,10k+点赞的 SpringBoot 后台管理系统竟然出了详细教程!

利用Redis实现高并发计数器\请求限制

  作者:李秀才

  liboyi.blog.csdn.net/article/details/79427271

业务需求中经常有需要用到计数器的场景:譬如一个手机号一天限制发送5条短信、一个接口一分钟限制多少请求、一个接口一天限制调用多少次等等。使用Redis的Incr自增命令可以轻松实现以上需求。以一个接口一天限制调用次数为例:

/**
 * 是否拒绝服务
 * @return
 */

private boolean denialOfService(String userId){
 long count=JedisUtil.setIncr(DateUtil.getDate()+"&"+userId+"&"+"queryCarViolation"86400);
 if(count<=10){
  return false;
 }
 return true;
}


/**
  * 查询违章
  * @param plateNumber车牌
  * @param vin 车架号
  * @param engineNo发动机
  * @param request
  * @param response
  * @throws Exception
  */

 @RequestMapping("/queryCarViolationList.json")
 @AuthorizationApi
 public void queryCarViolationList(@CurrentToken Token token,String plateNumber,String vin,
       String engineNo,HttpServletRequest request,HttpServletResponse response)
 throws Exception 
{
     String userId=token.getUserId();
           //超过限制,拦截请求
     if(denialOfService(userId)){
    apiData(request, response, ReqJson.error(CarError.ONLY_5_TIMES_A_DAY_CAN_BE_FOUND));
    return;
     }
  //没超过限制,业务逻辑……
}

每次调用接口之前,先获得下计数器自增后的值,如果小于限制,放行,执行后面的代码。如果大于限制,则拦截掉。

JedisUtil工具类:

public class JedisUtil {
 protected final static Logger logger = Logger.getLogger(JedisUtil.class);
 private static  JedisPool jedisPool;
 
 @Autowired(required = true)
 public void setJedisPool(JedisPool jedisPool) {
  JedisUtil.jedisPool = jedisPool;
 }
 /**
  * 对某个键的值自增
  * @author liboyi
  * @param key 键
  * @param cacheSeconds 超时时间,0为不超时
  * @return
  */

 public static long setIncr(String key, int cacheSeconds) {
  long result = 0;
  Jedis jedis = null;
  try {
   jedis = jedisPool.getResource();
   result =jedis.incr(key);
   if (result<=1 && cacheSeconds != 0) {
    jedis.expire(key, cacheSeconds);
   }
   logger.debug("set "+ key + " = " + result);
  } catch (Exception e) {
   logger.warn("set "+ key + " = " + result);
  } finally {
   jedisPool.returnResource(jedis);
  }
  return result;
 }

END

推荐好文

强大,10k+点赞的 SpringBoot 后台管理系统竟然出了详细教程!

分享一套基于SpringBoot和Vue的企业级中后台开源项目,代码很规范!

能挣钱的,开源 SpringBoot 商城系统,功能超全,超漂亮!

利用Redis实现高并发计数器\请求限制

原文始发于微信公众号(Java知音):利用Redis实现高并发计数器\请求限制