Redis频率限制工具类

在API级别限流中，可以使用Redis的频率计数工具实现。

短信验证码的每天发送次数，短信验证码登录的错误次数，等，都需要用到Redis的频率计数工具。

本文提供Redis频率限制函数级方法。

redis.java

/**
 * 1分钟内限制key的最大计数值
 *
 * @param key key
 * @param standCount 最大计数
 * @return 是否触发最大计数值，触发返回true
 */
public Boolean limitCount(String key, Long standCount) {

    Long count = stringRedisTemplate.opsForValue().increment(key);
  
    Long expire = stringRedisTemplate.getExpire(key);
    if (expire == null || expire <= 0) {
        stringRedisTemplate.expire(key, 60, TimeUnit.SECONDS);
    }

    return count != null && count > standCount;
}

/**
 * x分钟内限制key的最大计数值
 *
 * @param key key
 * @param timeout 超时时间，单位s
 * @param standCount 最大计数
 * @return 是否触发最大计数值，触发返回true
 */
public Boolean limitCount(String key, Integer timeout, Long standCount) {

    Long count = stringRedisTemplate.opsForValue().increment(key);

    Long expire = stringRedisTemplate.getExpire(key);
    if (expire == null || expire <= 0) {
        stringRedisTemplate.expire(key, timeout, TimeUnit.SECONDS);
    }

    return count != null && count > standCount;
}

/**
 * 限制key的最大计数值
 *
 * <p>不增加计数
 *
 * @param key key
 * @param standCount 最大计数
 * @return 是否触发最大计数值，触发返回true
 */
public Boolean limitCountNoAdd(String key, Long standCount) {

    String countStr = stringRedisTemplate.opsForValue().get(key);
    long count;
    if (StringUtils.isBlank(countStr)) {
        count = 0L;
    } else {
        count = Long.parseLong(countStr);
    }

    return count >= standCount;
}