package cn.wasu.metadata_platform_be.util;
import cn.hutool.json.JSONObject;
import com.aliyuncs.CommonRequest;
import com.aliyuncs.CommonResponse;
import com.aliyuncs.DefaultAcsClient;
import com.aliyuncs.IAcsClient;
import com.aliyuncs.exceptions.ClientException;
import com.aliyuncs.exceptions.ServerException;
import com.aliyuncs.http.MethodType;
import com.aliyuncs.profile.DefaultProfile;
import org.springframework.beans.factory.annotation.Autowired;
import javax.mail.Session;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.util.Random;
public class AliyunSendMsgUtils {
/**
* duanfy
* 2021-06-08
* 阿里云短信服务
*/
private static final String regionId = "cn-qingdao";
private static final String accessKeyId = "自己的id";
private static final String accessSecret = "accesssecret";
private static final String doMain = "dysmsapi.aliyuncs.com";
private static final String version = "2017-05-25 ";
private static final String action = "SendSms";
private static final String signName="签名名称";
private static final String templateCode="模板";
public static String sendMsg(String phone, HttpServletRequest req){
DefaultProfile profile = DefaultProfile.getProfile(regionId, accessKeyId, accessSecret);
IAcsClient client = new DefaultAcsClient(profile);
CommonRequest request = new CommonRequest();
request.setSysMethod(MethodType.POST);
request.setSysDomain(doMain);
request.setSysVersion(version);
request.setSysAction(action);
request.putQueryParameter("RegionId", regionId);
request.putQueryParameter("PhoneNumbers", phone);
request.putQueryParameter("SignName", signName);
request.putQueryParameter("TemplateCode", templateCode);
// 生成六位验证码
String code = verificationCode();
//将获取到的验证码存到session里
//通过key为phone实现验证码和手机号绑定
HttpSession session = req.getSession(true);
req.setAttribute(phone,code);
session.setAttribute(phone,code);
//短信的参数不要删掉
JSONObject jsonObject=new JSONObject();
jsonObject.put("code",code);
request.putQueryParameter("TemplateParam",jsonObject.toString());
try {
CommonResponse response = client.getCommonResponse(request);
System.out.println(response.getData());
return code;
} catch (ServerException e) {
e.printStackTrace();
} catch (ClientException e) {
e.printStackTrace();
}
return null;
}
private static String verificationCode(){
String sources = "0123456789"; // 加上一些字母,就可以生成pc站的验证码了
Random rand = new Random();
StringBuffer flag = new StringBuffer();
for (int j = 0; j < 6; j++)
{
flag.append(sources.charAt(rand.nextInt(9)) + "");
}
return flag.toString();
}
}
上面是一个工具类,接下来去写登录的方法
public ServiceResult login(@RequestParam("phone") String phone, @RequestParam("code") String code, HttpServletResponse response, HttpSession session) {
try {
UserMetadata userMetadata = userService.getUserByPhone(phone);
//手机号和验证码不可以为空
if (StringUtils.isBlank(phone) || StringUtils.isBlank(code)) {
return ServiceResult.error(StatusEnum.PHONE_CODE_ISNULL);
}
System.out.println("sessionId= "+session.getId());
if (!session.getAttribute(phone).equals(code)) {
return ServiceResult.error(StatusEnum.CODE_ERROR);
}
Map<String, Object> map = new HashMap<>();
// 只封装用户名称与用户id
String toKen = JWTUtils.sign(userMetadata.getUserName(), userMetadata.getId(), userMetadata.getPrId());
map.put("token", toKen);
map.put("user", userMetadata);
response.setHeader("token", toKen);
return ServiceResult.success(map);
} catch (Exception e) {
e.printStackTrace();
}finally {
session.removeAttribute(phone);
}
return ServiceResult.buildFAILURE();
}
上述给前端传的参数就是phone,code
下面在写一个获取验证码的方法
public ServiceResult sendMsg(@RequestParam("phone") String phone, HttpServletRequest req){
Integer count = iUserService.getPhone(phone);
if (count==0){
return ServiceResult.error(StatusEnum.PHONE_ERROR);
}
if (StringUtils.isBlank(phone)){
return ServiceResult.error(StatusEnum.PHONE_NULL);
}
//从写好的工具类里取出来穿的手机号,通过手机号发送验证码
String code = AliyunSendMsgUtils.sendMsg(phone,req);
if (StringUtils.isBlank(code)){
return ServiceResult.error(StatusEnum.GET_CODE_ERROR);
}
return ServiceResult.success(code);
}