利用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")@AuthorizationApipublic 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;}}