项目中需要一个发送短信验证码的功能,但是需要限制每天发送限制码的次数,可以借助Redis的incr命令实现。
incr命令:
对存储在指定key
的数值执行原子的加1操作。
如果指定的key不存在,那么在执行incr操作之前,会先将它的值设定为0
。
返回值是执行递增操作后key
对应的值。
代码:
long count = redisTemplate.opsForValue().increment(mobile.concat(RedisKeyConstant.SMS_COUNT),1);
if (count > 5 ) {
throw new ServiceException("您的操作太过频繁!");
} else {
// 验证码次数凌晨清除
Duration duration = Duration.between(LocalDateTime.now(), LocalDate.now().plusDays(1).atTime(0,0,0));
redisTemplate.expire(mobile.concat(RedisKeyConstant.SMS_COUNT),duration.toMinutes(),TimeUnit.MINUTES);
}
由于Redis是串行执行的,所以increment会每次给key值的value加1,超过5次就抛出异常,提示操作频繁,否则就放过并设置有效时间到第二天凌晨。