优化验证码
This commit is contained in:
@@ -7,11 +7,13 @@ import com.ttstd.videotablet.sms.SendSms;
|
||||
import com.ttstd.videotablet.utils.TextUtils;
|
||||
import org.apache.commons.lang3.RandomStringUtils;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.data.redis.core.StringRedisTemplate;
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.RequestParam;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import java.time.Duration;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
@@ -24,27 +26,30 @@ public class SmsController {
|
||||
@Autowired
|
||||
private StringRedisTemplate stringRedisTemplate;
|
||||
|
||||
@Autowired
|
||||
private RedisTemplate<String, Object> redisTemplate;
|
||||
|
||||
@GetMapping("/verify_code")
|
||||
public Result getMethodName(@RequestParam("number") String number) {
|
||||
if (TextUtils.isEmpty(number)) {
|
||||
public Result getMethodName(@RequestParam("phone") String phone) {
|
||||
if (TextUtils.isEmpty(phone)) {
|
||||
return Result.error().message("phone number is empty");
|
||||
}
|
||||
String oldCode = stringRedisTemplate.opsForValue().get(number);
|
||||
String oldCode = stringRedisTemplate.opsForValue().get(phone);
|
||||
if (TextUtils.isEmpty(oldCode)) {
|
||||
String code = SendSms.generatedcode(6);
|
||||
return sendCode(number, code);
|
||||
return sendCode(phone, code);
|
||||
} else {
|
||||
return sendCode(number, oldCode);
|
||||
return sendCode(phone, oldCode);
|
||||
}
|
||||
}
|
||||
|
||||
private Result sendCode(String number, String code) {
|
||||
private Result sendCode(String phone, String code) {
|
||||
try {
|
||||
com.aliyun.dysmsapi20170525.Client client = createClient();
|
||||
com.aliyun.dysmsapi20170525.models.SendSmsRequest sendSmsRequest = new com.aliyun.dysmsapi20170525.models.SendSmsRequest()
|
||||
.setSignName("同同的小站")
|
||||
.setTemplateCode("SMS_468370574")
|
||||
.setPhoneNumbers(number)
|
||||
.setPhoneNumbers(phone)
|
||||
.setTemplateParam("{\"code\":\"" + code + "\"}");
|
||||
com.aliyun.teautil.models.RuntimeOptions runtime = new com.aliyun.teautil.models.RuntimeOptions();
|
||||
// 复制代码运行请自行打印 API 的返回值
|
||||
@@ -54,11 +59,15 @@ public class SmsController {
|
||||
// System.out.println(sendSmsResponse.getBody().getCode());
|
||||
if ("OK".equals(sendSmsResponse.getBody().getCode())) {
|
||||
String randomString = RandomStringUtils.randomAlphanumeric(32);
|
||||
long expireAt = System.currentTimeMillis() + Duration.ofMinutes(5).toMillis();
|
||||
Map<String, Object> map = new HashMap<>();
|
||||
map.put("verify_key", randomString);
|
||||
map.put("phone", phone);
|
||||
map.put("code", code);
|
||||
map.put("sms", sendSmsResponse.getBody().getMessage());
|
||||
map.put("verifyKey", randomString);
|
||||
map.put("expireAt", expireAt);
|
||||
//4.保存验证码到Redis,并且设置有效期5分钟
|
||||
stringRedisTemplate.opsForValue().set(number, code, LOGIN_CODE_TTL, TimeUnit.MINUTES);
|
||||
redisTemplate.opsForValue().set(phone, map, Duration.ofMinutes(5));
|
||||
return Result.ok().data(map);
|
||||
} else {
|
||||
return Result.error().message(sendSmsResponse.getBody().getMessage());
|
||||
|
||||
Reference in New Issue
Block a user