SpringBoot+Redis发送短信

SpringBoot+Redis发送短信

pom.xml

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-redis</artifactId>
        </dependency>

application.xml

spring:
  data:
    redis:
      host: 192.168.226.134
      password: 123456

实体类

package com.sin.pojo;

/**
 * @createTime 2024/6/5 9:48
 * @createAuthor SIN
 * @use 验证码信息
 */
public class VerificationCode {
    private String phoneNumber;

    private String code;
    // setter getter方法
}

RandomCodeUtil.java

package com.sin.utils;

import org.springframework.stereotype.Component;

import java.util.Random;

/**
 * @createTime 2024/6/5 9:50
 * @createAuthor SIN
 * @use
 */
@Component
public class RandomCodeUtil {
    // 生成6位随机数字验证码
    public static String generateRandomCode() {
        Random random = new Random();
        int code = 100000 + random.nextInt(900000);
        return String.valueOf(code);
    }
}

VerifcationCodeService.java

package com.sin.service;

import java.time.Duration;

/**
 * @createTime 2024/6/5 9:52
 * @createAuthor SIN
 * @use
 */
public interface VerifcationCodeService {

    // 保存验证码到 Redis 并设置过期时间为60秒
    public String saveVerificationCode(String phoneNumber) ;

    // 检查验证码是否正确
    public boolean checkVerificationCode(String phoneNumber, String code) ;
}

VerifcationCodeServiceImpl.java

package com.sin.service.impl;
import com.sin.service.VerifcationCodeService;
import com.sin.utils.RandomCodeUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.TimeoutUtils;
import org.springframework.stereotype.Service;
import java.time.Duration;
/**
* @createTime 2024/6/5 9:52
* @createAuthor SIN
* @use
*/
@Service
public class VerifcationCodeServiceImpl  implements VerifcationCodeService {
@Autowired
private StringRedisTemplate stringRedisTemplate;
@Autowired
private RandomCodeUtil randomCodeUtil;
@Override
public String saveVerificationCode(String phoneNumber) {
String codeInRedis = stringRedisTemplate.opsForValue().get(phoneNumber);
if (codeInRedis == null) {
String code = randomCodeUtil.generateRandomCode();
stringRedisTemplate.opsForValue().set(phoneNumber, code, Duration.ofSeconds(60));
long currentTimeMillis = System.currentTimeMillis();
stringRedisTemplate.opsForValue().set(phoneNumber + "_sentTime", String.valueOf(currentTimeMillis)); // 记录发送时间
return "验证码已发送至手机";
} else {
// 如果已经发送过验证码，则计算距离下次发送的时间间隔
long timeLeft = getTimeLeft(phoneNumber);
return "信息已发送，请等待 " + timeLeft + " 秒后重试";
}
}
// 计算距离下次发送的时间间隔
private long getTimeLeft(String phoneNumber) {
String lastSentTimeStr = stringRedisTemplate.opsForValue().get(phoneNumber + "_sentTime");
if (lastSentTimeStr != null) {
long lastSentTime = Long.parseLong(lastSentTimeStr);
long currentTime = System.currentTimeMillis();
long elapsedTime = currentTime - lastSentTime;
long timeLeft = 60 - elapsedTime / 1000; // 时间间隔以秒为单位
return Math.max(timeLeft, 0); // 如果时间间隔小于0，则返回0
} else {
return 0; // 如果没有记录发送时间，则可以立即重试
}
}
@Override
public boolean checkVerificationCode(String phoneNumber, String code) {
String savedCode = stringRedisTemplate.opsForValue().get(phoneNumber);
return code.equals(savedCode);
}
}

VerificationCodeController.java

package com.sin.controller;
import com.sin.service.VerifcationCodeService;
import com.sin.utils.RandomCodeUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class VerificationCodeController {
@Autowired
private VerifcationCodeService verificationCodeService;
// 发送验证码的接口
@PostMapping("/sendVerificationCode")
public String sendVerificationCode(@RequestParam String phoneNumber) {
return  verificationCodeService.saveVerificationCode(phoneNumber);
}
// 验证验证码的接口
@PostMapping("/verifyCode")
public String verifyCode(@RequestParam String phoneNumber, @RequestParam String code) {
if (verificationCodeService.checkVerificationCode(phoneNumber, code)) {
return "验证通过";
} else {
return "验证失败";
}
}
}