feat: 集成 dynamicTp

Author: cadecode

README.md

@@ -34,4 +34,4 @@
 - [x] 集成 pagehelper
 - [x] 集成 Spring Boot Admin 监控
 - [x] 集成 TransmittableThreadLocal
-- [ ] 集成 dynamicTp
+- [x] 集成 dynamicTp

application/src/main/java/top/cadecode/uniboot/UniBootApplication.java

@@ -1,5 +1,6 @@
 package top.cadecode.uniboot;
 
+import com.dtp.core.spring.EnableDynamicTp;
 import org.mybatis.spring.annotation.MapperScan;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
@@ -9,8 +10,9 @@
 import org.springframework.context.annotation.EnableAspectJAutoProxy;
 
 /**
- *
+ * 启动类
  */
+@EnableDynamicTp
 @MapperScan("top.cadecode.**.mapper")
 @EnableAspectJAutoProxy(exposeProxy = true)
 @SpringBootApplication(exclude = DataSourceAutoConfiguration.class)

application/src/main/java/top/cadecode/uniboot/demo/controller/DynamicTpController.java

@@ -0,0 +1,41 @@
+package top.cadecode.uniboot.demo.controller;
+
+import com.dtp.core.thread.DtpExecutor;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import top.cadecode.uniboot.common.annotation.ApiFormat;
+
+/**
+ * dynamic-tp 测试
+ *
+ * @author Cade Li
+ * @date 2023/3/15
+ */
+@ApiFormat
+@Slf4j
+@RequiredArgsConstructor
+@Api(tags = "dynamic-tp 测试")
+@RestController
+@RequestMapping("demo/tp")
+public class DynamicTpController {
+
+    private final DtpExecutor asyncExecutor;
+
+    @ApiOperation("测试 reject 通知")
+    @PostMapping("test_reject")
+    public String testReject() {
+        // 提交 1000 个任务，测试 reject
+        for (int i = 0; i < 10000; i++) {
+            int finalI = i;
+            asyncExecutor.execute(() -> {
+                log.info("task {}", finalI);
+            });
+        }
+        return "OK";
+    }
+}

application/src/main/resources/application.yml

@@ -50,6 +50,23 @@ spring:
         wall:
           config:
             multi-statement-allow: true
+  dynamic:
+    tp:
+      collectorTypes: logging
+      logPath: /log/${uni-boot.config.name}
+      monitorInterval: 10
+      executors:
+        - threadPoolName: asyncExecutor
+          thread-name-prefix: asyncExecutor
+          corePoolSize: 6
+          maximumPoolSize: 8
+          queueCapacity: 1000
+          queueType: VariableLinkedBlockingQueue
+          rejectedHandlerType: CallerRunsPolicy
+          keepAliveTime: 50
+          waitForTasksToCompleteOnShutdown: true
+          awaitTerminationSeconds: 5
+          notifyEnabled: false
 
 # pagehelper
 pagehelper:

common/pom.xml

@@ -113,5 +113,14 @@
             <groupId>com.alibaba</groupId>
             <artifactId>transmittable-thread-local</artifactId>
         </dependency>
+        <!--dynamicTp-->
+        <dependency>
+            <groupId>cn.dynamictp</groupId>
+            <artifactId>dynamic-tp-spring-boot-starter-common</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>cn.dynamictp</groupId>
+            <artifactId>dynamic-tp-spring-boot-starter-extension-notify-email</artifactId>
+        </dependency>
     </dependencies>
 </project>

framework/src/main/java/top/cadecode/uniboot/framework/config/ThreadPoolConfig.java

@@ -0,0 +1,67 @@
+package top.cadecode.uniboot.framework.config;
+
+import com.dtp.core.thread.DtpExecutor;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.scheduling.annotation.AsyncConfigurer;
+import org.springframework.scheduling.annotation.SchedulingConfigurer;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
+import org.springframework.scheduling.config.ScheduledTaskRegistrar;
+
+import java.util.concurrent.Executor;
+
+/**
+ * 线程池设置
+ *
+ * @author Cade Li
+ * @date 2023/3/15
+ */
+@Slf4j
+@Configuration
+public class ThreadPoolConfig {
+
+    /**
+     * Spring 定时任务线程池
+     */
+    @Bean(name = "taskScheduler", destroyMethod = "shutdown")
+    public ThreadPoolTaskScheduler taskScheduler() {
+        ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
+        scheduler.setPoolSize(6);
+        scheduler.setThreadNamePrefix("taskScheduler");
+        scheduler.setWaitForTasksToCompleteOnShutdown(true);
+        scheduler.setAwaitTerminationSeconds(5);
+        return scheduler;
+    }
+
+
+    @Slf4j
+    @RequiredArgsConstructor
+    @Configuration
+    public static class ExecutorConfig implements AsyncConfigurer, SchedulingConfigurer {
+
+        private final DtpExecutor asyncExecutor;
+        private final ThreadPoolTaskScheduler taskScheduler;
+
+        @Override
+        public Executor getAsyncExecutor() {
+            return asyncExecutor;
+        }
+
+        @Override
+        public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
+            return (throwable, method, objects) -> {
+                log.error("异步任务执行出现异常, message {}, method {}, params {}",
+                        throwable, method, objects);
+            };
+
+        }
+
+        @Override
+        public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
+            taskRegistrar.setTaskScheduler(taskScheduler);
+        }
+    }
+}

pom.xml

@@ -44,6 +44,7 @@
         <dynamic-ds.version>3.6.0</dynamic-ds.version>
         <sba.version>2.6.11</sba.version>
         <ttl.version>2.12.2</ttl.version>
+        <dynamic-tp.version>1.1.0</dynamic-tp.version>
     </properties>
 
     <dependencies>
@@ -159,6 +160,16 @@
                 <artifactId>transmittable-thread-local</artifactId>
                 <version>${ttl.version}</version>
             </dependency>
+            <dependency>
+                <groupId>cn.dynamictp</groupId>
+                <artifactId>dynamic-tp-spring-boot-starter-common</artifactId>
+                <version>${dynamic-tp.version}</version>
+            </dependency>
+            <dependency>
+                <groupId>cn.dynamictp</groupId>
+                <artifactId>dynamic-tp-spring-boot-starter-extension-notify-email</artifactId>
+                <version>${dynamic-tp.version}</version>
+            </dependency>
         </dependencies>
     </dependencyManagement>