使用redis实现手机短信验证码和lua完成重置功能

文章目录

前言
一、介绍
二、代码
- 1.LoginController
- 2.reset.lua
总结

前言

2024了,各种各样的门户网站和APP都需要登录,登录方式也各种各样,由于都要绑定用户手机号码,所以大部分都支持了手机验证码登录的方式,接下来我们使用redis来完成验证码的功能。

一、介绍

方法名	描述
get	获取验证码
match	验证码配对
reset	验证码重置
check	手机号检测
sendVerificationCode	发送验证码

二、代码

1.LoginController

@Controller
@RequestMapping("verificationCode")
public class LoginController {@Autowiredprivate StringRedisTemplate stringRedisTemplate;private int maxRetryCount=3;private static final String COUNT="_COUNT";private static final String CODE="_CODE";private static final DefaultRedisScript<Boolean> RESET_SCRIPT;static {RESET_SCRIPT = new DefaultRedisScript<>();RESET_SCRIPT.setLocation(new ClassPathResource("redis/reset.lua"));RESET_SCRIPT.setResultType(Boolean.class);}@RequestMapping("get")@ResponseBodypublic boolean get(String phone){Optional.of(phone);if(check(phone)){return sendVerificationCode(phone);}return false;}@RequestMapping("match")@ResponseBodypublic boolean match(String phone, String code) {String redisCode = stringRedisTemplate.opsForValue().get(phone + CODE);System.out.println("用户["+phone+"],认证完成,跳转到首页!");return code.equals(redisCode);}@RequestMapping("reset")@ResponseBodypublic boolean reset(String phone) {boolean res=false;try {//用手机号拼接上字符串来作为key值，这是作为发送的次数String countKey = phone + COUNT;//用手机号拼接上字符串来作为key值，来表示你的验证码String codeKey = phone + CODE;if(stringRedisTemplate.execute(RESET_SCRIPT,Arrays.asList(countKey, codeKey))){res=true;System.out.println("重置已完成");}}catch (Exception e){e.printStackTrace();}return res;}private boolean check(String phone) {String regex = "^1[0-9]{10}$"; // 定义手机号格式的正则表达式Pattern pattern = Pattern.compile(regex);Matcher matcher = pattern.matcher(phone);return matcher.matches();}private boolean sendVerificationCode(String phone){//用手机号拼接上字符串来作为key值，这是作为发送的次数String countKey = phone + COUNT;//用手机号拼接上字符串来作为key值，来表示你的验证码String codeKey = phone + CODE;try {//根据countKey来获取发送的次数String countStr = stringRedisTemplate.opsForValue().get(countKey);Integer count;try {//将String类型的次数来强转为Integercount = Integer.valueOf(countStr);}catch (NumberFormatException e){count = 0;}if(count > maxRetryCount){System.out.println("24小时内短信发送次数已达"+maxRetryCount+"次，不能继续发送！24小时后自动重置次数");//做判断来表示发送三次，再次发送失败return false;}//判断codeKey是否存在boolean isTtl = stringRedisTemplate.hasKey(codeKey);if(isTtl){//如果存在的获取还有多少秒过期long ttlCode = stringRedisTemplate.getExpire(codeKey);System.out.println("验证码在1分钟内保持有效，请于"+ttlCode+"秒之后再次发送");return false;}//获取6位数的随机数//这个地方如果换成手机短信的api接口的话就可以实现手机短信验证的功能String code = RandomUtil.randomNumbers(6);//保存在随机数中设置过期时间stringRedisTemplate.opsForValue().set(codeKey,code,1, TimeUnit.MINUTES);//设置过期时间为1天long timeOut = 24 * 60 * 60;//做一个判断如果不是第一次发送，就要根据countKey来获取他的剩余的过期时间if(count!=0){//如果这个地方是短信验证的功能的话就是可以换成根据countKey来获取它是否存在来判断是否验证码过期timeOut = stringRedisTemplate.getExpire(countKey);}stringRedisTemplate.opsForValue().set(countKey,String.valueOf(count+1),timeOut,TimeUnit.SECONDS);System.out.println("短信验证码发送完毕！请注意查收!");return true;}catch (Exception e){stringRedisTemplate.delete(codeKey);stringRedisTemplate.delete(countKey);return false;}}
}

2.reset.lua

-- reset.lua
local count = redis.call('get',KEYS[1])
if (count ~= nil) thenredis.call('del',KEYS[1])
end
local code = redis.call('get',KEYS[2])
if (code ~= nil) thenredis.call('del',KEYS[2])
end
return true