Redis实现高并发计数器
业务需求中经常有需要用到计数器的场景：譬如一个手机号一天限制发送5条短信、一个接口一分钟限制多少请求、一个接口一天限制调用多少次等等。使用Redis的Incr自增命令可以轻松实现以上需求。以一个接口一天限制调用次数为例：
/**
*是否拒绝服务
*@return
*/
privatebooleandenialOfService(StringuserId){
longcount=JedisUtil.setIncr(DateUtil.getDate()+"&"+userId+"&"+"queryCarViolation",86400);
if(count<=10){
returnfalse;
}
returntrue;
}
/**
*查询违章
*@paramplateNumber车牌
*@paramvin车架号
*@paramengineNo发动机
*@paramrequest
*@paramresponse
*@throwsException
*/
@RequestMapping("/queryCarViolationList.json")
@AuthorizationApi
publicvoidqueryCarViolationList(@CurrentTokenTokentoken,StringplateNumber,Stringvin,
StringengineNo,HttpServletRequestrequest,HttpServletResponseresponse)throwsException{
StringuserId=token.getUserId();
//超过限制，拦截请求
if(denialOfService(userId)){
apiData(request,response,ReqJson.error(CarError.ONLY_5_TIMES_A_DAY_CAN_BE_FOUND));
return;
}
//没超过限制，业务逻辑……
}
每次调用接口之前，先获得下计数器自增后的值，如果小于限制，放行，执行后面的代码。如果大于限制，则拦截掉。
JedisUtil工具类：
publicclassJedisUtil{
protectedfinalstaticLoggerlogger=Logger.getLogger(JedisUtil.class);
privatestaticJedisPooljedisPool;
@Autowired(required=true)
publicvoidsetJedisPool(JedisPooljedisPool){
JedisUtil.jedisPool=jedisPool;
}
/**
*对某个键的值自增
*@authorliboyi
*@paramkey键
*@paramcacheSeconds超时时间，0为不超时
*@return
*/
publicstaticlongsetIncr(Stringkey,intcacheSeconds){
longresult=0;
Jedisjedis=null;
try{
jedis=jedisPool.getResource();
result=jedis.incr(key);
if(cacheSeconds!=0){
jedis.expire(key,cacheSeconds);
}
logger.debug("set"+key+"="+result);
}catch(Exceptione){
logger.warn("set"+key+"="+result);
}finally{
jedisPool.returnResource(jedis);
}
returnresult;
}
}
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。