|
@@ -1,9 +1,11 @@
|
|
|
package cn.iocoder.dashboard.modules.system.service.sms.impl;
|
|
|
|
|
|
+import cn.hutool.core.collection.CollUtil;
|
|
|
import cn.hutool.core.util.ReUtil;
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
import cn.iocoder.dashboard.common.enums.CommonStatusEnum;
|
|
|
import cn.iocoder.dashboard.common.pojo.PageResult;
|
|
|
+import cn.iocoder.dashboard.framework.mybatis.core.dataobject.BaseDO;
|
|
|
import cn.iocoder.dashboard.framework.sms.core.client.SmsClient;
|
|
|
import cn.iocoder.dashboard.framework.sms.core.client.SmsClientFactory;
|
|
|
import cn.iocoder.dashboard.framework.sms.core.client.SmsCommonResult;
|
|
@@ -16,17 +18,19 @@ import cn.iocoder.dashboard.modules.system.convert.sms.SysSmsTemplateConvert;
|
|
|
import cn.iocoder.dashboard.modules.system.dal.dataobject.sms.SysSmsChannelDO;
|
|
|
import cn.iocoder.dashboard.modules.system.dal.dataobject.sms.SysSmsTemplateDO;
|
|
|
import cn.iocoder.dashboard.modules.system.dal.mysql.sms.SysSmsTemplateMapper;
|
|
|
+import cn.iocoder.dashboard.modules.system.mq.producer.sms.SysSmsProducer;
|
|
|
import cn.iocoder.dashboard.modules.system.service.sms.SysSmsChannelService;
|
|
|
import cn.iocoder.dashboard.modules.system.service.sms.SysSmsTemplateService;
|
|
|
import com.google.common.annotations.VisibleForTesting;
|
|
|
+import com.google.common.collect.ImmutableMap;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.springframework.scheduling.annotation.Scheduled;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import org.springframework.util.Assert;
|
|
|
|
|
|
+import javax.annotation.PostConstruct;
|
|
|
import javax.annotation.Resource;
|
|
|
-import java.util.Collection;
|
|
|
-import java.util.List;
|
|
|
-import java.util.Map;
|
|
|
-import java.util.Objects;
|
|
|
+import java.util.*;
|
|
|
import java.util.regex.Pattern;
|
|
|
|
|
|
import static cn.iocoder.dashboard.common.exception.util.ServiceExceptionUtil.exception;
|
|
@@ -39,6 +43,7 @@ import static cn.iocoder.dashboard.modules.system.enums.SysErrorCodeConstants.*;
|
|
|
* @date 2021/1/25 9:25
|
|
|
*/
|
|
|
@Service
|
|
|
+@Slf4j
|
|
|
public class SysSmsTemplateServiceImpl implements SysSmsTemplateService {
|
|
|
|
|
|
/**
|
|
@@ -46,6 +51,24 @@ public class SysSmsTemplateServiceImpl implements SysSmsTemplateService {
|
|
|
*/
|
|
|
private static final Pattern PATTERN_PARAMS = Pattern.compile("\\{(.*?)}");
|
|
|
|
|
|
+ /**
|
|
|
+ * 定时执行 {@link #schedulePeriodicRefresh()} 的周期
|
|
|
+ * 因为已经通过 Redis Pub/Sub 机制,所以频率不需要高
|
|
|
+ */
|
|
|
+ private static final long SCHEDULER_PERIOD = 5 * 60 * 1000L;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 短信模板缓存
|
|
|
+ * key:短信模板编码 {@link SysSmsTemplateDO#getCode()}
|
|
|
+ *
|
|
|
+ * 这里声明 volatile 修饰的原因是,每次刷新时,直接修改指向
|
|
|
+ */
|
|
|
+ private volatile Map<String, SysSmsTemplateDO> smsTemplateCache;
|
|
|
+ /**
|
|
|
+ * 缓存短信模板的最大更新时间,用于后续的增量轮询,判断是否有更新
|
|
|
+ */
|
|
|
+ private volatile Date maxUpdateTime;
|
|
|
+
|
|
|
@Resource
|
|
|
private SysSmsTemplateMapper smsTemplateMapper;
|
|
|
|
|
@@ -55,11 +78,66 @@ public class SysSmsTemplateServiceImpl implements SysSmsTemplateService {
|
|
|
@Resource
|
|
|
private SmsClientFactory smsClientFactory;
|
|
|
|
|
|
+ @Resource
|
|
|
+ private SysSmsProducer smsProducer;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 初始化 {@link #smsTemplateCache} 缓存
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ @PostConstruct
|
|
|
+ public void initLocalCache() {
|
|
|
+ // 获取短信模板列表,如果有更新
|
|
|
+ List<SysSmsTemplateDO> smsTemplateList = this.loadSmsTemplateIfUpdate(maxUpdateTime);
|
|
|
+ if (CollUtil.isEmpty(smsTemplateList)) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 写入缓存
|
|
|
+ ImmutableMap.Builder<String, SysSmsTemplateDO> builder = ImmutableMap.builder();
|
|
|
+ smsTemplateList.forEach(sysSmsTemplateDO -> builder.put(sysSmsTemplateDO.getCode(), sysSmsTemplateDO));
|
|
|
+ smsTemplateCache = builder.build();
|
|
|
+ assert smsTemplateList.size() > 0; // 断言,避免告警
|
|
|
+ maxUpdateTime = smsTemplateList.stream().max(Comparator.comparing(BaseDO::getUpdateTime)).get().getUpdateTime();
|
|
|
+ log.info("[initLocalCache][初始化 SmsTemplate 数量为 {}]", smsTemplateList.size());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Scheduled(fixedDelay = SCHEDULER_PERIOD, initialDelay = SCHEDULER_PERIOD)
|
|
|
+ public void schedulePeriodicRefresh() {
|
|
|
+ initLocalCache();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 如果短信模板发生变化,从数据库中获取最新的全量短信模板。
|
|
|
+ * 如果未发生变化,则返回空
|
|
|
+ *
|
|
|
+ * @param maxUpdateTime 当前短信模板的最大更新时间
|
|
|
+ * @return 短信模板列表
|
|
|
+ */
|
|
|
+ private List<SysSmsTemplateDO> loadSmsTemplateIfUpdate(Date maxUpdateTime) {
|
|
|
+ // 第一步,判断是否要更新。
|
|
|
+ if (maxUpdateTime == null) { // 如果更新时间为空,说明 DB 一定有新数据
|
|
|
+ log.info("[loadSmsTemplateIfUpdate][首次加载全量短信模板]");
|
|
|
+ } else { // 判断数据库中是否有更新的短信模板
|
|
|
+ if (smsTemplateMapper.selectExistsByUpdateTimeAfter(maxUpdateTime) == null) {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+ log.info("[loadSmsTemplateIfUpdate][增量加载全量短信模板]");
|
|
|
+ }
|
|
|
+ // 第二步,如果有更新,则从数据库加载所有短信模板
|
|
|
+ return smsTemplateMapper.selectList();
|
|
|
+ }
|
|
|
+
|
|
|
@Override
|
|
|
public SysSmsTemplateDO getSmsTemplateByCode(String code) {
|
|
|
return smsTemplateMapper.selectByCode(code);
|
|
|
}
|
|
|
|
|
|
+ @Override
|
|
|
+ public SysSmsTemplateDO getSmsTemplateByCodeFromCache(String code) {
|
|
|
+ return smsTemplateCache.get(code);
|
|
|
+ }
|
|
|
+
|
|
|
@Override
|
|
|
public String formatSmsTemplateContent(String content, Map<String, Object> params) {
|
|
|
return StrUtil.format(content, params);
|
|
@@ -84,6 +162,8 @@ public class SysSmsTemplateServiceImpl implements SysSmsTemplateService {
|
|
|
template.setParams(parseTemplateContentParams(template.getContent()));
|
|
|
template.setChannelCode(channelDO.getCode());
|
|
|
smsTemplateMapper.insert(template);
|
|
|
+ // 发送刷新消息
|
|
|
+ smsProducer.sendSmsTemplateRefreshMessage();
|
|
|
// 返回
|
|
|
return template.getId();
|
|
|
}
|
|
@@ -104,6 +184,8 @@ public class SysSmsTemplateServiceImpl implements SysSmsTemplateService {
|
|
|
updateObj.setParams(parseTemplateContentParams(updateObj.getContent()));
|
|
|
updateObj.setChannelCode(channelDO.getCode());
|
|
|
smsTemplateMapper.updateById(updateObj);
|
|
|
+ // 发送刷新消息
|
|
|
+ smsProducer.sendSmsTemplateRefreshMessage();
|
|
|
}
|
|
|
|
|
|
@Override
|
|
@@ -112,6 +194,8 @@ public class SysSmsTemplateServiceImpl implements SysSmsTemplateService {
|
|
|
this.validateSmsTemplateExists(id);
|
|
|
// 更新
|
|
|
smsTemplateMapper.deleteById(id);
|
|
|
+ // 发送刷新消息
|
|
|
+ smsProducer.sendSmsTemplateRefreshMessage();
|
|
|
}
|
|
|
|
|
|
private void validateSmsTemplateExists(Long id) {
|