package com.walker.wxtools.support;
|
|
import com.fasterxml.jackson.databind.node.ObjectNode;
|
import com.walker.infrastructure.utils.JsonUtils;
|
import com.walker.wxtools.AbstractOfficialAccountEngine;
|
import com.walker.wxtools.BasicAccessToken;
|
import com.walker.wxtools.Constants;
|
import com.walker.wxtools.JsTicket;
|
import com.walker.wxtools.WxRemoteException;
|
import com.walker.wxtools.WxUserInfo;
|
import org.springframework.web.client.RestTemplate;
|
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
* 微信公众号引擎默认实现。
|
* <p>子类可继承该对象，重写方法，以满足自己应用，例如：缓存'Redis'缓存等。</p>
|
* @author 时克英
|
* @date 2023-04-20
|
*/
|
public class DefaultOfficialAccountEngine extends AbstractOfficialAccountEngine {
|
|
@Override
|
protected JsTicket acquireCachedJsTicket(String cacheKey) {
|
return this.innerJsTicketCache.get(cacheKey);
|
}
|
|
@Override
|
protected void writeJsTicketToCache(String cacheKey, JsTicket jsTicket) {
|
if(this.innerJsTicketCache.get(cacheKey) != null){
|
this.innerJsTicketCache.remove(cacheKey);
|
}
|
this.innerJsTicketCache.put(cacheKey, jsTicket);
|
logger.info("写入'jsTicket'缓存");
|
}
|
|
@Override
|
protected JsTicket acquireRemoteJsTicket(String basicAccessToken) throws WxRemoteException {
|
this.checkRestTemplate();
|
Map<String, Object> paramMap = new HashMap<>(4);
|
paramMap.put("type", "jsapi");
|
paramMap.put("access_token", basicAccessToken);
|
|
try {
|
String entity = restTemplate.getForObject(Constants.URL_JS_TICKET, String.class, paramMap);
|
if(entity == null){
|
throw new RuntimeException("调用'jsTicket'返回空数据, basicAccessToken = " + basicAccessToken);
|
}
|
if(this.logger.isDebugEnabled()){
|
this.logger.debug(entity);
|
}
|
ObjectNode objectNode = JsonUtils.jsonStringToObjectNode(entity);
|
if(objectNode.get(Constants.NAME_ERROR_CODE).intValue() != 0){
|
throw new WxRemoteException(objectNode.get(Constants.NAME_ERROR_MESSAGE).asText(), "微信获取'jsTicket'返回错误", null);
|
}
|
|
JsTicket jsTicket = new JsTicket();
|
jsTicket.setTicket(objectNode.get("ticket").asText());
|
jsTicket.setExpireSeconds(objectNode.get(Constants.NAME_EXPIRES_IN).longValue());
|
jsTicket.setCreateTime(System.currentTimeMillis());
|
return jsTicket;
|
|
} catch (Exception ex){
|
StringBuilder error = new StringBuilder("微信远程获取'jsTicket'失败, basicAccessToken=").append(basicAccessToken);
|
if(ex instanceof WxRemoteException){
|
WxRemoteException wex = (WxRemoteException) ex;
|
throw wex;
|
} else {
|
error.append(", error = ").append(ex.getMessage());
|
}
|
throw new WxRemoteException(error.toString(), ex.getMessage(), ex);
|
}
|
}
|
|
@Override
|
protected BasicAccessToken acquireCachedBasicAccessToken(String cacheKey) {
|
// 默认实现，基于内存，获取时，需要看是否超时
|
return this.innerAccessTokenCache.get(cacheKey);
|
}
|
|
@Override
|
protected void writeBasicAccessTokenToCache(String cacheKey, BasicAccessToken basicAccessToken) {
|
if(this.innerAccessTokenCache.get(cacheKey) != null){
|
this.innerAccessTokenCache.remove(cacheKey);
|
}
|
this.innerAccessTokenCache.put(cacheKey, basicAccessToken);
|
logger.info("写入'basicAccessToken'缓存");
|
}
|
|
@Override
|
protected BasicAccessToken acquireRemoteBasicAccessToken(String appId, String secret) throws WxRemoteException {
|
this.checkRestTemplate();
|
Map<String, Object> paramMap = new HashMap<>(4);
|
paramMap.put("appid", appId);
|
paramMap.put("secret", secret);
|
paramMap.put("grant_type", "client_credential");
|
try {
|
// 必须用：get 方式
|
// ResponseEntity<String> entity = restTemplate.postForEntity(Constants.URL_BASIC_ACCESS_TOKEN, paramMap, String.class);
|
String entity = restTemplate.getForObject(Constants.URL_BASIC_ACCESS_TOKEN, String.class, paramMap);
|
if(entity == null){
|
throw new RuntimeException("调用'BasicAccessToken'返回空数据, appId = " + appId);
|
}
|
if(this.logger.isDebugEnabled()){
|
// this.logger.debug(entity.getBody());
|
this.logger.debug(entity);
|
}
|
// ObjectNode objectNode = JsonUtils.jsonStringToObjectNode(entity.getBody());
|
ObjectNode objectNode = JsonUtils.jsonStringToObjectNode(entity);
|
if(objectNode.has(Constants.NAME_ERROR_CODE)){
|
throw new WxRemoteException(objectNode.get(Constants.NAME_ERROR_MESSAGE).asText(), "微信获取'BasicAccessToken'返回错误", null);
|
}
|
BasicAccessToken basicAccessToken = new BasicAccessToken();
|
basicAccessToken.setAccessToken(objectNode.get(Constants.NAME_ACCESS_TOKEN).asText());
|
basicAccessToken.setExpireSeconds(objectNode.get(Constants.NAME_EXPIRES_IN).longValue());
|
basicAccessToken.setCreateTime(System.currentTimeMillis());
|
return basicAccessToken;
|
|
} catch (Exception ex){
|
StringBuilder error = new StringBuilder("微信远程获取'BasicAccessToken'失败, appId=");
|
error.append(appId);
|
if(ex instanceof WxRemoteException){
|
WxRemoteException wex = (WxRemoteException) ex;
|
error.append(", error = ").append(wex.getError());
|
}
|
throw new WxRemoteException(error.toString(), ex.getMessage(), ex);
|
}
|
}
|
|
@Override
|
protected WxUserInfo acquireRemoteUserInfoSub(String openId, String basicAccessToken) throws WxRemoteException {
|
this.checkRestTemplate();
|
Map<String, Object> paramMap = new HashMap<>(4);
|
paramMap.put("access_token", basicAccessToken);
|
paramMap.put("openid", openId);
|
paramMap.put("lang", "zh_CN");
|
|
try {
|
String entity = restTemplate.getForObject(Constants.URL_USER_INFO_SUB, String.class, paramMap);
|
if(entity == null){
|
throw new RuntimeException("调用'userInfoSub'返回空数据, openId = " + openId + ", access_token=" + basicAccessToken);
|
}
|
ObjectNode objectNode = JsonUtils.jsonStringToObjectNode(entity);
|
if(objectNode.has(Constants.NAME_ERROR_CODE)){
|
throw new WxRemoteException(objectNode.get(Constants.NAME_ERROR_MESSAGE).asText(), "微信获取'userInfoSub'返回错误", null);
|
}
|
WxUserInfo userInfo = JsonUtils.jsonStringToObject(entity, WxUserInfo.class);
|
return userInfo;
|
|
} catch (Exception ex){
|
StringBuilder error = new StringBuilder("微信远程获取'userInfoSub'失败, openId=");
|
error.append(openId);
|
if(ex instanceof WxRemoteException){
|
WxRemoteException wex = (WxRemoteException) ex;
|
// error.append(", error = ").append(wex.getError());
|
throw wex;
|
} else {
|
error.append(", error = ").append(ex.getMessage());
|
}
|
throw new WxRemoteException(error.toString(), ex.getMessage(), ex);
|
}
|
}
|
|
private void checkRestTemplate(){
|
if(this.restTemplate == null){
|
throw new IllegalArgumentException("RestTemplate未配置!");
|
}
|
}
|
|
public void setRestTemplate(RestTemplate restTemplate) {
|
this.restTemplate = restTemplate;
|
}
|
|
private final Map<String, JsTicket> innerJsTicketCache = new ConcurrentHashMap<>(8);
|
private final Map<String, BasicAccessToken> innerAccessTokenCache = new ConcurrentHashMap<>(8);
|
private RestTemplate restTemplate;
|
}
|
