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

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。