From a88619a7dc2509798b9448bbbadf79f5f864ff73 Mon Sep 17 00:00:00 2001
From: nadirvishun <nadirvishun@gmail.com>
Date: Thu, 14 Apr 2022 11:52:49 +0800
Subject: [PATCH] =?UTF-8?q?:art:=20#2587=20=E3=80=90=E5=B0=8F=E7=A8=8B?=
 =?UTF-8?q?=E5=BA=8F=E3=80=91=E5=A2=9E=E5=8A=A0=E8=B7=AF=E7=94=B1=E7=BA=BF?=
 =?UTF-8?q?=E7=A8=8B=E6=B1=A0=E5=85=B3=E9=97=AD=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../wx/miniapp/message/WxMaMessageRouter.java | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/message/WxMaMessageRouter.java b/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/message/WxMaMessageRouter.java
index 031c688c5..e2c497e13 100644
--- a/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/message/WxMaMessageRouter.java
+++ b/weixin-java-miniapp/src/main/java/cn/binarywang/wx/miniapp/message/WxMaMessageRouter.java
@@ -51,6 +51,51 @@ public class WxMaMessageRouter {
     this.messageDuplicateChecker = new WxMessageInMemoryDuplicateChecker();
   }
 
+  /**
+   * 使用自定义的 {@link ExecutorService}.
+   */
+  public WxMaMessageRouter(WxMaService wxMaService, ExecutorService executorService) {
+    this.wxMaService = wxMaService;
+    this.executorService = executorService;
+    this.sessionManager = new StandardSessionManager();
+    this.exceptionHandler = new LogExceptionHandler();
+    this.messageDuplicateChecker = new WxMessageInMemoryDuplicateChecker();
+  }
+
+  /**
+   * 系统退出前，应该调用该方法
+   */
+  public void shutDownExecutorService() {
+    this.executorService.shutdown();
+  }
+
+  /**
+   * 系统退出前，应该调用该方法，增加了超时时间检测
+   */
+  public void shutDownExecutorService(Integer second) {
+    this.executorService.shutdown();
+    try {
+      if (!this.executorService.awaitTermination(second, TimeUnit.SECONDS)) {
+        this.executorService.shutdownNow();
+        if (!this.executorService.awaitTermination(second, TimeUnit.SECONDS))
+          log.error("线程池未关闭！");
+      }
+    } catch (InterruptedException ie) {
+      this.executorService.shutdownNow();
+      Thread.currentThread().interrupt();
+    }
+  }
+
+  /**
+   * <pre>
+   * 设置自定义的 {@link ExecutorService}
+   * 如果不调用该方法，默认使用内置的
+   * </pre>
+   */
+  public void setExecutorService(ExecutorService executorService) {
+    this.executorService = executorService;
+  }
+
   /**
    * 开始一个新的Route规则.
    */