Redis实现验证码发送并限制每日发送次数的示例代码

收藏

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《Redis实现验证码发送并限制每日发送次数的示例代码》，聊聊发送、Redis验证码、限制次数，我们一起来看看吧！

1、功能

• 输入手机号，点击发送后随机生成六位数字码，2分钟有效
• 输入验证码，点击验证，返回成功或失败
• 每个手机号每天只能输3次

2、分析

• 每个手机每天只能输3次：incr每次发送之后+1，当值为3时提示不能发送，过期时间为当天结束
• 随机生成6位数字验证码：RandomUtil（hutool）
• 验证码2分钟有效：放入redis里并设置过期时间2分钟
• 判断验证码是否一致：从redis里获取验证码和输入的验证码进行比对

3、实现

package cn.ken.blog.controller.common;

import cn.hutool.core.date.DateUnit;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.RandomUtil;
import cn.ken.blog.common.constant.Constants;
import cn.ken.blog.common.domain.Result;
import cn.ken.blog.common.enums.ErrorCodeEnum;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Date;
import java.util.concurrent.TimeUnit;

/**
 * 验证码控制器
 * @author Ken-Chy129
 * @date 2022/4/17 20:28
 */
@RestController
@SuppressWarnings(value = { "unchecked", "rawtypes" })
public class CaptureController {
    
    @Autowired
    private RedisTemplate redisTemplate;
    
    // 生成验证码
    @GetMapping("getNumCode")
    public Result
   
     getNumCode(String phone) {
        String captureLimitKey = Constants.CAPTCHA_LIMIT_KEY + phone;
        Integer counts = (Integer) redisTemplate.opsForValue().get(captureLimitKey);
        if (ObjectUtils.isEmpty(counts)) {
            // 今天第一次验证，故之前缓存中无该键
            // 距离今天结束剩下多少毫秒
            long expire = DateUtil.endOfDay(new Date()).between(new Date(), DateUnit.MS);
            redisTemplate.opsForValue().set(captureLimitKey, 1, expire, TimeUnit.MILLISECONDS);
        } else if (counts ().error(ErrorCodeEnum.OVER_LIMITS);
        }
        // 生成验证码
        String code = RandomUtil.randomNumbers(6); // 随机生成六位数
        String captureCodeKey = Constants.CAPTCHA_CODE_KEY + phone;
        redisTemplate.opsForValue().set(captureCodeKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);
        return new Result
    
     ().success(captureCodeKey + ":" + code);     }          // 验证验证码     @GetMapping("verify")     public Result
     
       verify(String phone, String code) {         String captureCodeKey = Constants.CAPTCHA_CODE_KEY + phone;         String realCode = (String) redisTemplate.opsForValue().get(captureCodeKey);         if (ObjectUtils.isEmpty(realCode)) {             // redis中不存在该用户生成的验证码，证明验证码以过期销毁             return new Result
      
       ().error(ErrorCodeEnum.OVERDUE_CODE);         }         if (realCode.equals(code)) {             return new Result
       
        ().success("验证成功");         } else {             return new Result
        
         ().error(ErrorCodeEnum.ERROR_CODE);         }     }      //    @Scheduled(cron = "0 0 12 * * ?") //    private void clear() { //        redisTemplate.delete() //    } }
        
       
      
     
    
   

// Constants类

/**
 * 验证码 redis key
 */
public static final String CAPTCHA_CODE_KEY = "captcha_codes:";

/**
 * 每日限制 redis key
 */
public static final String CAPTCHA_LIMIT_KEY = "captcha_limits:";
/**
 * 验证码有效期（分钟）
 */
public static final Integer CAPTCHA_EXPIRATION = 2;