package com.newfiber.system.template; import cn.hutool.core.convert.Convert; import cn.hutool.core.util.IdUtil; import com.alibaba.csp.sentinel.util.StringUtil; import com.baomidou.mybatisplus.core.toolkit.StringPool; import com.github.qcloudsms.SmsMultiSender; import com.github.qcloudsms.SmsMultiSenderResult; import com.github.qcloudsms.httpclient.HTTPException; import com.newfiber.common.redis.service.RedisService; import com.newfiber.system.api.enums.ESystemRedisKey; import com.newfiber.system.domain.model.SmsCode; import com.newfiber.system.domain.model.SmsData; import com.newfiber.system.domain.model.SmsProperties; import com.newfiber.system.domain.model.SmsResponse; import java.io.IOException; import java.util.Collection; import java.util.concurrent.TimeUnit; import lombok.AllArgsConstructor; import org.springframework.http.HttpStatus; /** * 腾讯云短信发送类 * * @author Chill */ @AllArgsConstructor public class TencentSmsTemplate implements SmsTemplate { private static final int SUCCESS = 0; private static final String NATION_CODE = "86"; private final SmsProperties smsProperties; private final SmsMultiSender smsSender; private final RedisService redisService; @Override public SmsResponse sendMessage(SmsData smsData, Collection<String> phones) { try { Collection<String> values = smsData.getParams().values(); String[] params = Convert.toStrArray(values); SmsMultiSenderResult senderResult = smsSender.sendWithParam( NATION_CODE, Convert.toStrArray(phones), Integer.parseInt(smsProperties.getTemplateId()), params, smsProperties.getSignName(), StringPool.EMPTY, StringPool.EMPTY ); return new SmsResponse(senderResult.result == SUCCESS, senderResult.result, senderResult.toString()); } catch (HTTPException | IOException e) { e.printStackTrace(); return new SmsResponse(Boolean.FALSE, HttpStatus.INTERNAL_SERVER_ERROR.value(), e.getMessage()); } } @Override public SmsCode sendValidate(SmsData smsData, String phone) { SmsCode smsCode = new SmsCode(); boolean temp = sendSingle(smsData, phone); if (temp && StringUtil.isNotBlank(smsData.getKey())) { String id = IdUtil.randomUUID(); String value = smsData.getParams().get(smsData.getKey()); redisService.setCacheObject(ESystemRedisKey.SmsCaptcha, cacheKey(phone, id), value, 30L, TimeUnit.MINUTES); smsCode.setId(id).setValue(value); } else { smsCode.setSuccess(Boolean.FALSE); } return smsCode; } @Override public boolean validateMessage(SmsCode smsCode) { String id = smsCode.getId(); String value = smsCode.getValue(); String phone = smsCode.getPhone(); String cache = redisService.getCacheObject(ESystemRedisKey.SmsCaptcha, cacheKey(phone, id)); if (StringUtil.isNotBlank(value) && StringUtil.equalsIgnoreCase(cache, value)) { redisService.deleteObject(ESystemRedisKey.SmsCaptcha, cacheKey(phone, id)); return true; } return false; } }