🆕 binarywang#3000 【小程序/公众号】提供了更新access_token的消费接口

FreeOfYou · May 5, 2023 · 6221707 · 6221707
1 parent 24c18b8
commit 6221707
Show file tree

Hide file tree

Showing 5 changed files with 72 additions and 8 deletions.
diff --git a/weixin-java-common/src/main/java/me/chanjar/weixin/common/bean/WxAccessTokenEntity.java b/weixin-java-common/src/main/java/me/chanjar/weixin/common/bean/WxAccessTokenEntity.java
@@ -0,0 +1,20 @@
+package me.chanjar.weixin.common.bean;
+
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+/**
+ * @author cn
+ * @version 1.0
+ * @description
+ * @date 2023/3/27 18:34
+ */
+@Getter
+@Setter
+@ToString(callSuper = true)
+@EqualsAndHashCode(callSuper = true)
+public class WxAccessTokenEntity extends WxAccessToken {
+  private String appid;
+}
diff --git a/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/api/impl/BaseWxMaServiceImpl.java b/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/api/impl/BaseWxMaServiceImpl.java
@@ -324,7 +324,7 @@ protected String extractAccessToken(String resultContent) throws WxErrorExceptio
       throw new WxErrorException(error);
     }
     WxAccessToken accessToken = WxAccessToken.fromJson(resultContent);
-    config.updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn());
+    config.updateAccessTokenProcessor(accessToken.getAccessToken(), accessToken.getExpiresIn());
     return accessToken.getAccessToken();
   }
 

diff --git a/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/config/WxMaConfig.java b/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/config/WxMaConfig.java
@@ -1,6 +1,7 @@
 package cn.binarywang.wx.miniapp.config;
 
 import me.chanjar.weixin.common.bean.WxAccessToken;
+import me.chanjar.weixin.common.bean.WxAccessTokenEntity;
 import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder;
 
 import java.util.concurrent.locks.Lock;
@@ -43,7 +44,9 @@ public interface WxMaConfig {
    *
    * @param accessToken 要更新的WxAccessToken对象
    */
-  void updateAccessToken(WxAccessToken accessToken);
+  default void updateAccessToken(WxAccessToken accessToken) {
+    updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn());
+  }
 
   /**
    * 应该是线程安全的
@@ -53,6 +56,20 @@ public interface WxMaConfig {
    */
   void updateAccessToken(String accessToken, int expiresInSeconds);
 
+  default void updateAccessTokenProcessor(String accessToken, int expiresInSeconds) {
+    WxAccessTokenEntity wxAccessTokenEntity = new WxAccessTokenEntity();
+    wxAccessTokenEntity.setAppid(getAppid());
+    wxAccessTokenEntity.setAccessToken(accessToken);
+    wxAccessTokenEntity.setExpiresIn(expiresInSeconds);
+    updateAccessTokenBefore(wxAccessTokenEntity);
+    updateAccessToken(accessToken, expiresInSeconds);
+  }
+
+  default void updateAccessTokenBefore(WxAccessTokenEntity wxAccessTokenEntity) {
+
+  }
+
+
   /**
    * Gets jsapi ticket.
    *

diff --git a/...ava-miniapp/src/main/java/cn/binarywang/wx/miniapp/config/impl/WxMaDefaultConfigImpl.java b/...ava-miniapp/src/main/java/cn/binarywang/wx/miniapp/config/impl/WxMaDefaultConfigImpl.java
@@ -2,13 +2,16 @@
 
 import cn.binarywang.wx.miniapp.config.WxMaConfig;
 import cn.binarywang.wx.miniapp.json.WxMaGsonBuilder;
+import lombok.AccessLevel;
 import lombok.Getter;
-import me.chanjar.weixin.common.bean.WxAccessToken;
+import lombok.Setter;
+import me.chanjar.weixin.common.bean.WxAccessTokenEntity;
 import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder;
 
 import java.io.File;
 import java.util.concurrent.locks.Lock;
 import java.util.concurrent.locks.ReentrantLock;
+import java.util.function.Consumer;
 
 /**
  * 基于内存的微信配置provider，在实际生产环境中应该将这些配置持久化
@@ -58,6 +61,23 @@ public class WxMaDefaultConfigImpl implements WxMaConfig {
   private String apiHostUrl;
   private String accessTokenUrl;
 
+  /**
+   * 自定义配置token的消费者
+   */
+  @Setter
+  private Consumer<WxAccessTokenEntity> updateAccessTokenBefore;
+  @Getter(AccessLevel.NONE)
+  private boolean enableUpdateAccessTokenBefore = true;
+
+  /**
+   * 可临时关闭更新token回调，主要用于其他介质初始化数据时，可不进行回调
+   *
+   * @param enableUpdateAccessTokenBefore
+   */
+  public void enableUpdateAccessTokenBefore(boolean enableUpdateAccessTokenBefore) {
+    this.enableUpdateAccessTokenBefore = enableUpdateAccessTokenBefore;
+  }
+
   /**
    * 会过期的数据提前过期时间，默认预留200秒的时间
    */
@@ -95,17 +115,24 @@ public boolean isAccessTokenExpired() {
     return isExpired(this.expiresTime);
   }
 
-  @Override
-  public synchronized void updateAccessToken(WxAccessToken accessToken) {
-    updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn());
-  }
+//  @Override
+//  public synchronized void updateAccessToken(WxAccessToken accessToken) {
+//    updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn());
+//  }
 
   @Override
   public synchronized void updateAccessToken(String accessToken, int expiresInSeconds) {
     setAccessToken(accessToken);
     setExpiresTime(expiresAheadInMillis(expiresInSeconds));
   }
 
+  @Override
+  public void updateAccessTokenBefore(WxAccessTokenEntity wxAccessTokenEntity) {
+    if (updateAccessTokenBefore != null && enableUpdateAccessTokenBefore) {
+      updateAccessTokenBefore.accept(wxAccessTokenEntity);
+    }
+  }
+
   @Override
   public String getJsapiTicket() {
     return this.jsapiTicket;

diff --git a/...va-miniapp/src/main/java/cn/binarywang/wx/miniapp/config/impl/WxMaRedissonConfigImpl.java b/...va-miniapp/src/main/java/cn/binarywang/wx/miniapp/config/impl/WxMaRedissonConfigImpl.java
@@ -91,7 +91,7 @@ public boolean isAccessTokenExpired() {
 
   @Override
   public void updateAccessToken(WxAccessToken accessToken) {
-    redisOps.setValue(this.accessTokenKey, accessToken.getAccessToken(), accessToken.getExpiresIn(), TimeUnit.SECONDS);
+    updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn());
   }
 
   @Override