refactor: 重构拒绝策略 SPI 接口.

dynamic-threadpool-spring-boot-starter/src/main/java/io/dynamic/threadpool/starter/spi/rejected/CustomRejectedExecutionHandler.java

@@ -1,9 +1,5 @@
 package io.dynamic.threadpool.starter.spi.rejected;
 
-import lombok.AllArgsConstructor;
-import lombok.Getter;
-import lombok.Setter;
-
 import java.util.concurrent.RejectedExecutionHandler;
 
 /**
@@ -15,20 +11,17 @@ import java.util.concurrent.RejectedExecutionHandler;
 public interface CustomRejectedExecutionHandler {
 
     /**
-     * 生成拒绝策略
+     * 获取类型
      *
      * @return
      */
-    RejectedExecutionHandlerWrap generateRejected();
+    Integer getType();
-
-    @Getter
-    @Setter
-    @AllArgsConstructor
-    class RejectedExecutionHandlerWrap {
 
-        private Integer type;
+    /**
-
+     * 生成拒绝策略
-        private RejectedExecutionHandler rejectedExecutionHandler;
+     *
+     * @return
+     */
+    RejectedExecutionHandler generateRejected();
 
-    }
 }

dynamic-threadpool-spring-boot-starter/src/main/java/io/dynamic/threadpool/starter/toolkit/thread/RejectedTypeEnum.java

@@ -72,9 +72,8 @@ public enum RejectedTypeEnum {
             Collection<CustomRejectedExecutionHandler> customRejectedExecutionHandlers = DynamicTpServiceLoader
                     .getSingletonServiceInstances(CustomRejectedExecutionHandler.class);
             Optional<RejectedExecutionHandler> customRejected = customRejectedExecutionHandlers.stream()
-                    .map(each -> each.generateRejected())
                     .filter(each -> Objects.equals(type, each.getType()))
-                    .map(each -> each.getRejectedExecutionHandler())
+                    .map(each -> each.generateRejected())
                     .findFirst();
 
             return customRejected.orElse(ABORT_POLICY.rejectedHandler);