Merge pull request #408 from dromara/dev

内存泄漏：默认配置下 DtpExecutor 在 AbortPolicy 拒绝策略下，拒绝时产生内存泄露 #402
dromara · Feb 25, 2024 · 00b5291 · 00b5291
2 parents d1bbc96 + 34d433d
commit 00b5291
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/core/src/main/java/org/dromara/dynamictp/core/reject/RejectedInvocationHandler.java b/core/src/main/java/org/dromara/dynamictp/core/reject/RejectedInvocationHandler.java
@@ -42,13 +42,13 @@ public RejectedInvocationHandler(Object target) {
 
     @Override
     public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
+        beforeReject((Runnable) args[0], (Executor) args[1]);
         try {
-            beforeReject((Runnable) args[0], (Executor) args[1]);
-            Object result = method.invoke(target, args);
-            afterReject((Runnable) args[0], (Executor) args[1]);
-            return result;
+            return method.invoke(target, args);
         } catch (InvocationTargetException ex) {
             throw ex.getCause();
+        } finally {
+            afterReject((Runnable) args[0], (Executor) args[1]);
         }
     }