[ISSUE #402] 修复异常拒绝任务时跳过afterReject执行问题

dromara · Feb 22, 2024 · 9638935 · 9638935
1 parent 12b672a
commit 9638935
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]);
         }
     }