package com.qianwen.core.notify.provider.wechat.mp;
|
|
import java.time.Duration;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.atomic.AtomicReference;
|
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.qianwen.core.http.HttpRequest;
|
import com.qianwen.core.log.exception.BizServiceException;
|
import com.qianwen.core.notify.DefaultNotifyType;
|
import com.qianwen.core.notify.NotifyType;
|
import com.qianwen.core.notify.Provider;
|
import com.qianwen.core.notify.notifier.AbstractNotifier;
|
import com.qianwen.core.notify.notifier.NotifierProperties;
|
import com.qianwen.core.notify.provider.wechat.WechatProvider;
|
import com.qianwen.core.notify.template.TemplateManager;
|
import com.qianwen.core.tool.api.BizMessage;
|
import com.qianwen.core.tool.metadata.Values;
|
import com.qianwen.core.tool.utils.Func;
|
import com.qianwen.core.tool.utils.StringUtil;
|
|
import me.chanjar.weixin.mp.bean.template.WxMpTemplateData;
|
import me.chanjar.weixin.mp.bean.template.WxMpTemplateMessage;
|
|
public class WeixinMpNotifier extends AbstractNotifier<WechatMpMessageTemplate> {
|
private static final Logger log = LoggerFactory.getLogger(WeixinMpNotifier.class);
|
private final AtomicReference<String> accessToken;
|
private long refreshTokenTime;
|
private final long tokenTimeOut;
|
private static final String tokenApi = "https://api.weixin.qq.com/cgi-bin/token";
|
private static final String notify = "https://api.weixin.qq.com/cgi-bin/message/template/send?access_token=";
|
private final WechatMpProperties properties;
|
private final String notifierId;
|
private NotifierProperties notifierProperties;
|
|
|
@Override // com.qianwen.core.notify.notifier.Notifier
|
public String getNotifierId() {
|
return this.notifierId;
|
}
|
|
@Override // com.qianwen.core.notify.notifier.Notifier
|
public NotifierProperties getNotifierProperties() {
|
return this.notifierProperties;
|
}
|
|
public WeixinMpNotifier(NotifierProperties properties, TemplateManager templateManager) {
|
super(templateManager);
|
this.accessToken = new AtomicReference<>();
|
this.tokenTimeOut = Duration.ofSeconds(7000L).toMillis();
|
WechatMpProperties wechatMpProperties = (WechatMpProperties) new JSONObject(properties.getConfiguration()).toJavaObject(WechatMpProperties.class);
|
this.properties = wechatMpProperties;
|
this.notifierProperties = properties;
|
this.notifierId = properties.getId();
|
}
|
|
@Override // com.qianwen.core.notify.notifier.Notifier
|
public NotifyType getType() {
|
return DefaultNotifyType.weiXinMp;
|
}
|
|
@Override // com.qianwen.core.notify.notifier.Notifier
|
public Provider getProvider() {
|
return WechatProvider.mpMessage;
|
}
|
|
public void send(WechatMpMessageTemplate template, String traceId, Values context, List<String> notifiedParty) {
|
String access_token = getToken();
|
Map<String, Object> maps = context.getAllValues();
|
List<WxMpTemplateData> templateDataList = new ArrayList<>();
|
if (Func.isNotEmpty(maps)) {
|
maps.forEach((k, v) -> {
|
WxMpTemplateData temp = new WxMpTemplateData();
|
temp.setValue(v.toString());
|
temp.setName(k);
|
templateDataList.add(temp);
|
});
|
}
|
HttpRequest httpRequest = HttpRequest.post(notify.concat(access_token));
|
if (Func.isNotEmpty(notifiedParty)) {
|
notifiedParty.forEach(x -> {
|
WxMpTemplateMessage wxMpTemplateMessage = new WxMpTemplateMessage();
|
wxMpTemplateMessage.setTemplateId(template.getTemplateId());
|
wxMpTemplateMessage.setToUser(x);
|
wxMpTemplateMessage.setUrl(template.getUrl());
|
wxMpTemplateMessage.setMiniProgram(template.getMiniProgram());
|
wxMpTemplateMessage.setData(templateDataList);
|
Map<String, Object> result = httpRequest.bodyString(wxMpTemplateMessage.toJson()).execute().onFailed((k2, v2) -> {
|
log.info(StringUtil.format("发送微信公众号模板消息{}失败,原因{}", new Object[]{k2.url(), v2.getMessage()}));
|
}).asMap(Object.class);
|
if (result.containsKey("errcode") && !result.get("errcode").equals(0)) {
|
BizMessage bizMessage = new BizMessage();
|
bizMessage.setMessage(result.get("errcode").toString());
|
bizMessage.setMessage(result.get("errmsg").toString());
|
throw new BizServiceException(bizMessage);
|
}
|
});
|
}
|
}
|
|
private String getToken() {
|
if (System.currentTimeMillis() - this.refreshTokenTime > this.tokenTimeOut || this.accessToken.get() == null) {
|
return requestToken();
|
}
|
return this.accessToken.get();
|
}
|
|
private String requestToken() {
|
HttpRequest httpRequest = HttpRequest.get(tokenApi);
|
Map<String, Object> queries = new HashMap<>();
|
queries.put("grant_type", "client_credential");
|
queries.put("appid", this.properties.getAppid());
|
queries.put("secret", this.properties.getSecret());
|
httpRequest.queryMap(queries);
|
Map<String, Object> result = httpRequest.execute().onFailed((k, v) -> {
|
log.info(StringUtil.format("发送微信公众号模板消息{}失败,原因{}", new Object[]{k.url(), v.getMessage()}));
|
}).asMap(String.class);
|
if (result.containsKey("access_token")) {
|
this.refreshTokenTime = System.currentTimeMillis();
|
this.accessToken.set(result.get("access_token").toString());
|
return result.get("access_token").toString();
|
}
|
BizMessage bizMessage = new BizMessage();
|
bizMessage.setMessage(result.get("errcode").toString());
|
bizMessage.setMessage(result.get("errmsg").toString());
|
throw new BizServiceException(bizMessage);
|
}
|
|
@Override // com.qianwen.core.notify.notifier.Notifier
|
public void close() {
|
this.accessToken.set(null);
|
this.refreshTokenTime = 0L;
|
}
|
}
|
