package com.ruoyi.message.wechat;
|
|
import com.ruoyi.message.SendException;
|
import com.ruoyi.message.SendResult;
|
import lombok.extern.slf4j.Slf4j;
|
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 路由到一个具体实现。
|
* <p>
|
* 注意：公众号通道的 key 与短信通道的 key 可以同名（例如 {@code hzszlyy} 既有短信发送方
|
* 也有公众号发送方），因此两个工厂各自维护自己的注册表，互不干扰。
|
*/
|
@Slf4j
|
@Component
|
public class WechatSenderFactory {
|
|
private final Map<String, WechatSender> senders = new LinkedHashMap<>();
|
|
public WechatSenderFactory(List<WechatSender> senderList) {
|
for (WechatSender sender : senderList) {
|
WechatSender exists = senders.put(sender.key(), sender);
|
if (exists != null) {
|
throw new IllegalStateException("公众号发送方 key 重复：" + sender.key()
|
+ "，冲突实现：" + exists.getClass().getName() + " 与 " + sender.getClass().getName());
|
}
|
}
|
log.info("【WechatSenderFactory】已注册公众号发送方：{}", senders.keySet());
|
}
|
|
/**
|
* 按 key 获取发送方
|
*
|
* @param key 渠道 key（即 {@code spring.profiles.active} 取值）
|
* @throws SendException 找不到对应实现时抛出
|
*/
|
public WechatSender get(String key) {
|
WechatSender sender = key == null ? null : senders.get(key);
|
if (sender == null) {
|
throw new SendException("未找到公众号发送方，key=" + key + "，已注册：" + senders.keySet());
|
}
|
return sender;
|
}
|
|
/**
|
* 按 key 获取发送方，找不到返回 {@code null}
|
*/
|
public WechatSender getOrNull(String key) {
|
return key == null ? null : senders.get(key);
|
}
|
|
/**
|
* 按 key 发送公众号消息
|
*/
|
public SendResult send(String key, WechatSendRequest 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());
|
}
|
}
|
