package com.ruoyi.message.sms;
|
|
import com.ruoyi.message.SendException;
|
import com.ruoyi.message.SendResult;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.stereotype.Component;
|
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
import java.util.TreeSet;
|
|
/**
|
* 短信发送方工厂：按 key 路由到一个具体实现，调用方不再需要 {@code if (active.equals("xx"))} 硬分支。
|
* <p>
|
* Spring 会自动把所有 {@link SmsSender} 注入进来，新增一个站点只需新增一个实现类。
|
*/
|
@Slf4j
|
@Component
|
public class SmsSenderFactory {
|
|
/**
|
* 站点 profile → 短信通道 key 的映射。
|
* <p>
|
* 目前两边取值完全一致（{@code spring.profiles.active} 直接就是渠道名），
|
* 单独留出映射表是为了将来 profile 名与渠道名解耦时不改调用方。
|
*/
|
private static final Map<String, String> PROFILE_ALIAS = new LinkedHashMap<>();
|
|
static {
|
PROFILE_ALIAS.put("xh", "xh");
|
PROFILE_ALIAS.put("ls", "ls");
|
PROFILE_ALIAS.put("sltd", "sltd");
|
PROFILE_ALIAS.put("nhfy", "nhfy");
|
PROFILE_ALIAS.put("hzszlyy", "hzszlyy");
|
}
|
|
private final Map<String, SmsSender> senders = new LinkedHashMap<>();
|
|
public SmsSenderFactory(List<SmsSender> senderList) {
|
for (SmsSender sender : senderList) {
|
SmsSender exists = senders.put(sender.key(), sender);
|
if (exists != null) {
|
throw new IllegalStateException("短信发送方 key 重复：" + sender.key()
|
+ "，冲突实现：" + exists.getClass().getName() + " 与 " + sender.getClass().getName());
|
}
|
}
|
log.info("【SmsSenderFactory】已注册短信发送方：{}", senders.keySet());
|
}
|
|
/**
|
* 按 key 获取发送方
|
*
|
* @param key 渠道 key（即 {@code spring.profiles.active} 取值）
|
* @return 发送方实现
|
* @throws SendException 找不到对应实现时抛出
|
*/
|
public SmsSender get(String key) {
|
String resolved = resolveSmsKey(key);
|
SmsSender sender = resolved == null ? null : senders.get(resolved);
|
if (sender == null) {
|
throw new SendException("未找到短信发送方，key=" + key + "，已注册：" + senders.keySet());
|
}
|
return sender;
|
}
|
|
/**
|
* 按 key 获取发送方，找不到返回 {@code null}
|
*/
|
public SmsSender getOrNull(String key) {
|
String resolved = resolveSmsKey(key);
|
return resolved == null ? null : senders.get(resolved);
|
}
|
|
/**
|
* 按 key 发送短信（等价于旧代码里那 200 行 if-else 的结果）
|
*/
|
public SendResult send(String key, SmsSendRequest request) {
|
return get(key).send(request);
|
}
|
|
/**
|
* 判断是否存在该 key 的发送方
|
*/
|
public boolean supports(String key) {
|
return getOrNull(key) != null;
|
}
|
|
/**
|
* 已注册的渠道 key
|
*/
|
public Set<String> keys() {
|
return new TreeSet<>(senders.keySet());
|
}
|
|
/**
|
* 站点 profile → 渠道 key
|
*/
|
public static String resolveSmsKey(String profile) {
|
if (StringUtils.isEmpty(profile)) {
|
return null;
|
}
|
String alias = PROFILE_ALIAS.get(profile);
|
return alias == null ? profile : alias;
|
}
|
}
|
