feature: update concurrent module

flowerfine · Nov 22, 2024 · 6fa75cb · 6fa75cb
1 parent b04b43f
commit 6fa75cb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...urrent-thread/src/main/java/cn/sliew/milky/concurrent/thread/MilkyThreadPoolExecutor.java b/...urrent-thread/src/main/java/cn/sliew/milky/concurrent/thread/MilkyThreadPoolExecutor.java
@@ -1,6 +1,8 @@
 package cn.sliew.milky.concurrent.thread;
 
 import cn.sliew.milky.common.collect.ConcurrentReferenceHashMap;
+import cn.sliew.milky.common.concurrent.RunnableWrapper;
+import cn.sliew.milky.common.exception.Rethrower;
 import cn.sliew.milky.concurrent.thread.policy.AbortPolicyWithReport;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
@@ -104,7 +106,11 @@ public void execute(Runnable command) {
                 try {
                     ((RunnableWrapper) command).onRejection(ex);
                 } finally {
-                    ((RunnableWrapper) command).onAfter();
+                    try {
+                        ((RunnableWrapper) command).onAfter();
+                    } catch (Exception e) {
+                        Rethrower.throwAs(e);
+                    }
                 }
             } else {
                 throw ex;