利用Redis實作高并發計數器

業務需求中經常有需要用到計數器的場景：譬如一個手機号一天限制發送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 (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;
}           

原文：

https://blog.csdn.net/qq_33556185/article/details/79427271

版權聲明：本文為部落客原創文章，轉載請附上博文連結！