项目中需要一个发送短信验证码的功能,但是需要限制每天发送限制码的次数,可以借助Redis实现,代码:
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次就抛出异常,提示操作频繁,否则就放过并设置有效时间到第二天凌晨。