规定时间内限制发送短信次数

2019-12-09

需求是3分钟之内限制同一个手机号码发送短信验证码的次数小于3次以内。

使用redis的zset数据结构来实现，参考以下鲁棒图

短信发送次数限制

第一步：通过redis分布式锁锁定该手机号，为了防止60s内重复发送
第二步：判断3分钟之前到现在发送了多少次短信验证码

1	zcount key min max

如果count>=3那就失败

第三步：插入该次发送短信事件

1	zadd key value score

第四步：判断如果当前的key数量太多就要清理一下多余的value

zcount key min max
如果count>=3那就要删除多余的
zremrangebyscore key min max

伪代码实现

//第一步
String sendKey = "redis:" + mobile;
long nowTime = System.currentTimeMillis()/1000;
boolean couldSend = stringRedisTemplate.opsForValue().setIfAbsent(sendKey, String.valueOf(nowTime));
if(couldSend){
	stringRedisTemplate.expire(sendKey, 60, TimeUnit.SECONDS);
}else{
	throw new Exception("间隔太短");
}
//第二步
long seconds = 180L;
long times = 3L;
String sendSetKey = "redis_set:" + mobile;
//设置最后一次发送
long startTime = nowTime - seconds;
//当前redis中有效总条数
long count = stringRedisTemplate.opsForZSet().count(redisKey, startTime, nowTime);
//如果条数大于或等于条数限制，则抛出异常
if(count >= times){
	throw new Exception("超过三次不能再发送");
}
//第三步
String value = "生成一个随机值即可";
//向set中添加新纪录
stringRedisTemplate.opsForZSet().add(redisKey, value, nowTime);
//第四步
count = stringRedisTemplate.opsForZSet().count(redisKey, startTime, nowTime);
if(count >= times){
	stringRedisTemplate.opsForZSet().removeRangeByScore(redisKey, 0, startTime);
}