package com.qianwen.core.sms; import com.github.qcloudsms.SmsMultiSender; import com.github.qcloudsms.SmsMultiSenderResult; import com.github.qcloudsms.httpclient.HTTPException; import java.io.IOException; import java.time.Duration; import java.util.Collection; import com.qianwen.core.redis.cache.BladeRedis; import com.qianwen.core.sms.model.SmsCode; import com.qianwen.core.sms.model.SmsData; import com.qianwen.core.sms.model.SmsResponse; import com.qianwen.core.sms.props.SmsProperties; import com.qianwen.core.tool.utils.Func; import com.qianwen.core.tool.utils.StringUtil; import org.springframework.http.HttpStatus; 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 BladeRedis bladeRedis; public TencentSmsTemplate(final SmsProperties smsProperties, final SmsMultiSender smsSender, final BladeRedis bladeRedis) { this.smsProperties = smsProperties; this.smsSender = smsSender; this.bladeRedis = bladeRedis; } @Override // com.qianwen.core.sms.SmsTemplate public SmsResponse sendMessage(SmsData smsData, Collection phones) { try { Collection values = smsData.getParams().values(); String[] params = StringUtil.toStringArray(values); SmsMultiSenderResult senderResult = this.smsSender.sendWithParam(NATION_CODE, StringUtil.toStringArray(phones), Func.toInt(this.smsProperties.getTemplateId()), params, this.smsProperties.getSignName(), "", ""); return new SmsResponse(senderResult.result == 0, Integer.valueOf(senderResult.result), senderResult.toString()); } catch (HTTPException | IOException e) { e.printStackTrace(); return new SmsResponse(Boolean.FALSE.booleanValue(), Integer.valueOf(HttpStatus.INTERNAL_SERVER_ERROR.value()), e.getMessage()); } } @Override // com.qianwen.core.sms.SmsTemplate public SmsCode sendValidate(SmsData smsData, String phone) { SmsCode smsCode = new SmsCode(); boolean temp = sendSingle(smsData, phone); if (temp && StringUtil.isNotBlank(smsData.getKey())) { String id = StringUtil.randomUUID(); String value = smsData.getParams().get(smsData.getKey()); this.bladeRedis.setEx(cacheKey(phone, id), value, Duration.ofMinutes(30L)); smsCode.setId(id).setValue(value); } else { smsCode.setSuccess(Boolean.FALSE.booleanValue()); } return smsCode; } @Override // com.qianwen.core.sms.SmsTemplate public boolean validateMessage(SmsCode smsCode) { String id = smsCode.getId(); String value = smsCode.getValue(); String phone = smsCode.getPhone(); String cache = (String) this.bladeRedis.get(cacheKey(phone, id)); if (StringUtil.isNotBlank(value) && StringUtil.equalsIgnoreCase(cache, value)) { this.bladeRedis.del(cacheKey(phone, id)); return true; } return false; } }