lsm/weixin-java-tools
1
0
Fork 0
mirror of https://gitee.com/binary/weixin-java-tools.git synced 2025-04-05 17:38:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🎨 增加switchoverTo方法在mpId不存在时的Storage实例获取获取

Browse Source
This commit is contained in:
zacone 2023-08-24 20:28:33 +08:00 committed by GitHub
parent bcc4839ff0
commit 90e1a9ebc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 34 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
weixin-java-mp/src/main/java/me/chanjar/weixin/mp/api/WxMpService.java
View File

@ -20,6 +20,7 @@ import me.chanjar.weixin.mp.config.WxMpConfigStorage;
import me.chanjar.weixin.mp.enums.WxMpApiUrl;
import java.util.Map;
import java.util.function.Function;
/**
* 微信公众号API的Service.
@ -393,6 +394,8 @@ public interface WxMpService extends WxService {
*/
boolean switchover(String mpId);
boolean switchover(String mpId, Function<String, WxMpConfigStorage> func);
/**
* 进行相应的公众号切换.
*
@ -401,6 +404,8 @@ public interface WxMpService extends WxService {
*/
WxMpService switchoverTo(String mpId);
WxMpService switchoverTo(String mpId, Function<String, WxMpConfigStorage> func);
/**
* 返回客服接口方法实现类，以方便调用其各个接口.
*

31
weixin-java-mp/src/main/java/me/chanjar/weixin/mp/api/impl/BaseWxMpServiceImpl.java
View File

@ -43,6 +43,7 @@ import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.function.Function;
import static me.chanjar.weixin.mp.enums.WxMpApiUrl.Other.*;
@ -156,6 +157,10 @@ public abstract class BaseWxMpServiceImpl<H, P> implements WxMpService, RequestH
@Setter
private WxMpFreePublishService freePublishService = new WxMpFreePublishServiceImpl(this);
@Getter
@Setter
private Function<String, WxMpConfigStorage> configStorageFunction;
private Map<String, WxMpConfigStorage> configStorageMap = new HashMap<>();
private int retrySleepMillis = 1000;
@ -575,21 +580,43 @@ public abstract class BaseWxMpServiceImpl<H, P> implements WxMpService, RequestH
@Override
public WxMpService switchoverTo(String mpId) {
return switchoverTo(mpId, configStorageFunction);
}
@Override
public WxMpService switchoverTo(String mpId, Function<String, WxMpConfigStorage> func) {
if (this.configStorageMap.containsKey(mpId)) {
WxMpConfigStorageHolder.set(mpId);
return this;
}
if (func != null) {
WxMpConfigStorage storage = func.apply(mpId);
if (storage != null) {
this.addConfigStorage(mpId, storage);
return this;
}
}
throw new WxRuntimeException(String.format("无法找到对应【%s】的公众号配置信息，请核实！", mpId));
}
@Override
public boolean switchover(String mpId) {
return switchover(mpId, configStorageFunction);
}
@Override
public boolean switchover(String mpId, Function<String, WxMpConfigStorage> func) {
if (this.configStorageMap.containsKey(mpId)) {
WxMpConfigStorageHolder.set(mpId);
return true;
}
if (func != null) {
WxMpConfigStorage storage = func.apply(mpId);
if (storage != null) {
this.addConfigStorage(mpId, storage);
return true;
}
}
log.error("无法找到对应【{}】的公众号配置信息，请核实！", mpId);
return false;
}