网站内容不收录,中国建设规划采购网站,经过开发建设 网站上线了,黄冈市建设局网站Java-12.Spring 中通过 ThreadPoolTaskExecutor 和 AsyncConfigurerSupport 配置默认异步线程池
前言
在Spring框架中#xff0c;TaskDecorator 是一个接口#xff0c;它可以用来自定义由 ThreadPoolTaskExecutor 或其他任务执行器管理的任务的装饰行为。这通常用于在执行任…Java-12.Spring 中通过 ThreadPoolTaskExecutor 和 AsyncConfigurerSupport 配置默认异步线程池
前言
在Spring框架中TaskDecorator 是一个接口它可以用来自定义由 ThreadPoolTaskExecutor 或其他任务执行器管理的任务的装饰行为。这通常用于在执行任务之前和之后添加某些上下文相关的行为比如设置线程上下文或者清理资源。
例如在执行异步操作时你可能需要将主线程的一些上下文信息比如用户身份验证令牌或请求上下文信息传递给执行异步操作的线程。TaskDecorator 就可以在这种场景下发挥作用。
自定义 TaskDecorator
public class ThreadPoolContextTaskDecorator implements TaskDecorator { Override public Runnable decorate(Runnable runnable) { return TtlRunnable.get(ThreadMdcUtil.wrap(runnable, MDC.getCopyOfContextMap())); } }可以继承 AsyncConfigurerSupport 实现默认的异步线程池
EnableAsync
SpringBootConfiguration
EnableConfigurationProperties(TaskExecutionProperties.class)
public class ThreadPoolConfig extends AsyncConfigurerSupport { Resource private TaskExecutionProperties properties; /** * 重写默认线程池配置Async异步会使用这个线程池 */ Override public Executor getAsyncExecutor() { TaskExecutionProperties.Pool pool properties.getPool(); TaskExecutorBuilder builder new TaskExecutorBuilder(); builder builder.queueCapacity(pool.getQueueCapacity()); builder builder.corePoolSize(pool.getCoreSize()); builder builder.maxPoolSize(pool.getMaxSize()); builder builder.allowCoreThreadTimeOut(pool.isAllowCoreThreadTimeout()); builder builder.keepAlive(pool.getKeepAlive()); Shutdown shutdown properties.getShutdown(); builder builder.awaitTermination(shutdown.isAwaitTermination()); builder builder.awaitTerminationPeriod(shutdown.getAwaitTerminationPeriod()); builder builder.threadNamePrefix(properties.getThreadNamePrefix()); builder.taskDecorator(new ThreadPoolContextTaskDecorator()); // 重点是这里ThreadPoolTaskExecutor executor builder.build(); executor.initialize(); return executor; } Override public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() { return new SimpleAsyncUncaughtExceptionHandler(); } }也可以直接定义一个 TaskDecorator Bean
查看源码 TaskExecutionAutoConfiguration
ConditionalOnClass(ThreadPoolTaskExecutor.class)
AutoConfiguration
EnableConfigurationProperties(TaskExecutionProperties.class)
public class TaskExecutionAutoConfiguration { /** * Bean name of the application {link TaskExecutor}. */ public static final String APPLICATION_TASK_EXECUTOR_BEAN_NAME applicationTaskExecutor; Bean ConditionalOnMissingBean public TaskExecutorBuilder taskExecutorBuilder(TaskExecutionProperties properties, ObjectProviderTaskExecutorCustomizer taskExecutorCustomizers, ObjectProviderTaskDecorator taskDecorator) { TaskExecutionProperties.Pool pool properties.getPool(); TaskExecutorBuilder builder new TaskExecutorBuilder(); builder builder.queueCapacity(pool.getQueueCapacity()); builder builder.corePoolSize(pool.getCoreSize()); builder builder.maxPoolSize(pool.getMaxSize()); builder builder.allowCoreThreadTimeOut(pool.isAllowCoreThreadTimeout()); builder builder.keepAlive(pool.getKeepAlive()); Shutdown shutdown properties.getShutdown(); builder builder.awaitTermination(shutdown.isAwaitTermination()); builder builder.awaitTerminationPeriod(shutdown.getAwaitTerminationPeriod()); builder builder.threadNamePrefix(properties.getThreadNamePrefix()); builder builder.customizers(taskExecutorCustomizers.orderedStream()::iterator); // spring-boot-autoconfigure 这里提供了 TaskDecorator 的入口// 只需要提供一个 TaskDecorator 的Bean 就可以复用 Spring Boot 中的异步线程池啦builder builder.taskDecorator(taskDecorator.getIfUnique());return builder; } Lazy Bean(name { APPLICATION_TASK_EXECUTOR_BEAN_NAME, AsyncAnnotationBeanPostProcessor.DEFAULT_TASK_EXECUTOR_BEAN_NAME }) ConditionalOnMissingBean(Executor.class) public ThreadPoolTaskExecutor applicationTaskExecutor(TaskExecutorBuilder builder) { return builder.build(); } }
