115 lines
5.4 KiB
Java
115 lines
5.4 KiB
Java
package com.ttstd.videotablet.controller;
|
||
|
||
import com.aliyun.dysmsapi20170525.models.SendSmsResponse;
|
||
import com.aliyun.tea.TeaException;
|
||
import com.ttstd.videotablet.result.Result;
|
||
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;
|
||
|
||
@RestController
|
||
public class SmsController {
|
||
private static final int LOGIN_CODE_TTL = 5;
|
||
|
||
//引入 redis
|
||
@Autowired
|
||
private StringRedisTemplate stringRedisTemplate;
|
||
|
||
@Autowired
|
||
private RedisTemplate<String, Object> redisTemplate;
|
||
|
||
@GetMapping("/verify_code")
|
||
public Result getMethodName(@RequestParam("phone") String phone) {
|
||
if (TextUtils.isEmpty(phone)) {
|
||
return Result.error().message("phone number is empty");
|
||
}
|
||
String oldCode = stringRedisTemplate.opsForValue().get(phone);
|
||
if (TextUtils.isEmpty(oldCode)) {
|
||
String code = SendSms.generatedcode(6);
|
||
return sendCode(phone, code);
|
||
} else {
|
||
return sendCode(phone, oldCode);
|
||
}
|
||
}
|
||
|
||
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(phone)
|
||
.setTemplateParam("{\"code\":\"" + code + "\"}");
|
||
com.aliyun.teautil.models.RuntimeOptions runtime = new com.aliyun.teautil.models.RuntimeOptions();
|
||
// 复制代码运行请自行打印 API 的返回值
|
||
SendSmsResponse sendSmsResponse = client.sendSmsWithOptions(sendSmsRequest, runtime);
|
||
// System.out.println(sendSmsResponse.body.toMap());
|
||
// System.out.println(sendSmsResponse.getBody().getMessage());
|
||
// 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("phone", phone);
|
||
map.put("code", code);
|
||
map.put("sms", sendSmsResponse.getBody().getMessage());
|
||
map.put("verifyKey", randomString);
|
||
map.put("expireAt", expireAt);
|
||
//4.保存验证码到Redis,并且设置有效期5分钟
|
||
redisTemplate.opsForValue().set(phone, map, Duration.ofMinutes(5));
|
||
return Result.ok().data(map);
|
||
} else {
|
||
return Result.error().message(sendSmsResponse.getBody().getMessage());
|
||
}
|
||
} catch (TeaException error) {
|
||
// 此处仅做打印展示，请谨慎对待异常处理，在工程项目中切勿直接忽略异常。
|
||
// 错误 message
|
||
System.out.println(error.getMessage());
|
||
// 诊断地址
|
||
System.out.println(error.getData().get("Recommend"));
|
||
com.aliyun.teautil.Common.assertAsString(error.message);
|
||
return Result.error().data("sms", error.message);
|
||
} catch (Exception _error) {
|
||
TeaException error = new TeaException(_error.getMessage(), _error);
|
||
// 此处仅做打印展示，请谨慎对待异常处理，在工程项目中切勿直接忽略异常。
|
||
// 错误 message
|
||
System.out.println(error.getMessage());
|
||
// 诊断地址
|
||
System.out.println(error.getData().get("Recommend"));
|
||
com.aliyun.teautil.Common.assertAsString(error.message);
|
||
return Result.error().data("sms", error.message);
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 使用AK&SK初始化账号Client
|
||
*
|
||
* @return Client
|
||
* @throws Exception
|
||
*/
|
||
public com.aliyun.dysmsapi20170525.Client createClient() throws Exception {
|
||
// 工程代码泄露可能会导致 AccessKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考。
|
||
// 建议使用更安全的 STS 方式，更多鉴权访问方式请参见：https://help.aliyun.com/document_detail/378657.html。
|
||
com.aliyun.teaopenapi.models.Config config = new com.aliyun.teaopenapi.models.Config()
|
||
// 必填，请确保代码运行环境设置了环境变量 ALIBABA_CLOUD_ACCESS_KEY_ID。
|
||
.setAccessKeyId("LTAI5tEVYLeg7U3bP58n3Xkj")
|
||
// 必填，请确保代码运行环境设置了环境变量 ALIBABA_CLOUD_ACCESS_KEY_SECRET。
|
||
.setAccessKeySecret("bbPSlxkaGgkhUPaCOP7CANaNNY1b2p");
|
||
// Endpoint 请参考 https://api.aliyun.com/product/Dysmsapi
|
||
config.endpoint = "dysmsapi.aliyuncs.com";
|
||
return new com.aliyun.dysmsapi20170525.Client(config);
|
||
}
|
||
}
|