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;
 }
}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。