多租户，接入 Spring Async 机制

YunaiV
1 parent cc78025f
Showing 11 changed files with 48 additions and 30 deletions
yudao-admin-server/src/main/java/cn/iocoder/yudao/adminserver/framework/async/config/AsyncConfiguration.java
yudao-admin-server/src/main/java/cn/iocoder/yudao/adminserver/framework/async/package-info.java
yudao-framework/yudao-spring-boot-starter-job/pom.xml
yudao-framework/yudao-spring-boot-starter-job/src/main/java/cn/iocoder/yudao/framework/quartz/config/YudaoAsyncAutoConfiguration.java
yudao-framework/yudao-spring-boot-starter-job/src/main/java/cn/iocoder/yudao/framework/quartz/config/YudaoQuartzAutoConfiguration.java
yudao-framework/yudao-spring-boot-starter-job/src/main/java/cn/iocoder/yudao/framework/quartz/package-info.java
yudao-framework/yudao-spring-boot-starter-job/src/main/resources/META-INF/spring.factories
yudao-user-server/src/main/java/cn/iocoder/yudao/userserver/framework/async/config/AsyncConfiguration.java
yudao-user-server/src/main/java/cn/iocoder/yudao/userserver/framework/async/package-info.java
yudao-user-server/src/main/java/cn/iocoder/yudao/userserver/framework/async/《芋道 Spring Boot 异步任务入门》.md
-package cn.iocoder.yudao.adminserver.framework.async.config;
-
-import org.springframework.context.annotation.Configuration;
-import org.springframework.scheduling.annotation.EnableAsync;
-
-@Configuration
-@EnableAsync
-public class AsyncConfiguration {
-}
-/**
- * 异步执行，基于 Spring @Async 实现
- */
-package cn.iocoder.yudao.adminserver.framework.async;
@@ -12,7 +12,10 @@
     <packaging>jar</packaging>
  
     <name>${artifactId}</name>
-    <description>定时任务，基于 Quartz 拓展</description>
+    <description>任务拓展
+        1. 定时任务，基于 Quartz 拓展
+        2. 异步任务，基于 Spring Async 拓展
+    </description>
     <url>https://github.com/YunaiV/ruoyi-vue-pro</url>
  
     <dependencies>
+package cn.iocoder.yudao.framework.quartz.config;
+
+import com.alibaba.ttl.TtlRunnable;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.config.BeanPostProcessor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.scheduling.annotation.EnableAsync;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+
+/**
+ * 异步任务 Configuration
+ */
+@Configuration
+@EnableAsync
+public class YudaoAsyncAutoConfiguration {
+
+    @Bean
+    public BeanPostProcessor threadPoolTaskExecutorBeanPostProcessor() {
+        return new BeanPostProcessor() {
+
+            @Override
+            public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
+                if (!(bean instanceof ThreadPoolTaskExecutor)) {
+                    return bean;
+                }
+                // 修改提交的任务，接入 TransmittableThreadLocal
+                ThreadPoolTaskExecutor executor = (ThreadPoolTaskExecutor) bean;
+                executor.setTaskDecorator(TtlRunnable::get);
+                return executor;
+            }
+
+        };
+    }
+
+}
@@ -6,6 +6,9 @@ import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.scheduling.annotation.EnableScheduling;
  
+/**
+ * 定时任务 Configuration
+ */
 @Configuration
 @EnableScheduling // 开启 Spring 自带的定时任务
 public class YudaoQuartzAutoConfiguration {
 /**
- * 定时任务，采用 Quartz 实现进程内的任务执行。
+ * 1. 定时任务，采用 Quartz 实现进程内的任务执行。
  * 考虑到高可用，使用 Quartz 自带的 MySQL 集群方案。
+ *
+ * 2. 异步任务，采用 Spring Async 异步执行。
  */
 package cn.iocoder.yudao.framework.quartz;
 org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
-  cn.iocoder.yudao.framework.quartz.config.YudaoQuartzAutoConfiguration
+  cn.iocoder.yudao.framework.quartz.config.YudaoQuartzAutoConfiguration,\
+  cn.iocoder.yudao.framework.quartz.config.YudaoAsyncAutoConfiguration
-package cn.iocoder.yudao.userserver.framework.async.config;
-
-import org.springframework.context.annotation.Configuration;
-import org.springframework.scheduling.annotation.EnableAsync;
-
-@Configuration
-@EnableAsync
-public class AsyncConfiguration {
-}
-/**
- * 异步执行，基于 Spring @Async 实现
- */
-package cn.iocoder.yudao.userserver.framework.async;
-<http://www.iocoder.cn/Spring-Boot/Async-Job/?yudao>