网站推广托管,营销网站建设步骤,深圳东门步行街在哪个区,网站建设 qq业务网制作概述
需求是想在线程池执行任务的时候#xff0c;在开始前将调用线程的信息传到子线程中#xff0c;在子线程完成后#xff0c;再清除传入的数据。
下面使用了spring的ThreadPoolTaskExecutor来实现这个需求.
ThreadPoolTaskExecutor
在jdk中使用的是ThreadPoolExecutor…概述
需求是想在线程池执行任务的时候在开始前将调用线程的信息传到子线程中在子线程完成后再清除传入的数据。
下面使用了spring的ThreadPoolTaskExecutor来实现这个需求.
ThreadPoolTaskExecutor
在jdk中使用的是ThreadPoolExecutor用于自定义线程池。 在spring中则是对ThreadPoolExecutor又包了一层加了一些参数进去ThreadPoolTaskExecutor,然后作为bean注入到spring的ioc容器中.
通常在使用线程池的时候想把调用线程的一些信息传递给子线程(线程池中的线程)一般都是要自己写一个装饰器然后把装饰器传递给线程池的execute方法。
不过spring中已经有现成的方法了就在ThreadPoolTaskExecutor中可以给定自定义的装饰器。
org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor#initializeExecutor 可以看到在初始化的时候会判断是否存在装饰器
ThreadPoolTaskExecutor使用装饰器传递调用线程信息
这样线程池中的线程在执行的时候都会经过装饰器处理要注意的是在线程执行完成之后需要把信息清理不然信息会串的
package org.xxx.common.core.executor.decorator;import org.slf4j.MDC;
import org.springframework.core.task.TaskDecorator;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;import java.util.Map;/*** 对spring的线程中的线程进行装饰*/public class ContextCopyingDecorator implements TaskDecorator {Overridepublic Runnable decorate(Runnable runnable) {try {//当前请求上下文RequestAttributes context RequestContextHolder.currentRequestAttributes();//copy当前调用线程的 ThreadLocalMap 中保存的信息MapString,String previous MDC.getCopyOfContextMap();return () - {try {//http request上下文塞到当前线程中RequestContextHolder.setRequestAttributes(context);//将调用线程的 ThreadLocalMap 塞到当前线程MDC.setContextMap(previous);runnable.run();} finally {//clearRequestContextHolder.resetRequestAttributes();MDC.clear();}};} catch (IllegalStateException e) {return runnable;}}
}//线程池配置/*** 核心线程数 cpu 核心数 1*/
private final int core Runtime.getRuntime().availableProcessors() 1;private ScheduledExecutorService scheduledExecutorService;Bean(name threadPoolTaskExecutor)
ConditionalOnProperty(prefix thread-pool, name enabled, havingValue true)
public ThreadPoolTaskExecutor threadPoolTaskExecutor(ThreadPoolProperties threadPoolProperties) {ThreadPoolTaskExecutor executor new ThreadPoolTaskExecutor();executor.setCorePoolSize(threadPoolProperties.getCoreSize());executor.setMaxPoolSize(threadPoolProperties.getMaxCoreSize());if(threadPoolProperties.getCoreSize() 0) executor.setCorePoolSize(core);if(threadPoolProperties.getMaxCoreSize() 0) executor.setMaxPoolSize(core * 2);//线程池队列大小executor.setQueueCapacity(threadPoolProperties.getQueueCapacity());//线程空闲存活时间executor.setKeepAliveSeconds(threadPoolProperties.getKeepAliveSeconds());//线程池拒绝时交由调用线程执行executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());//装饰线程池中的线程executor.setTaskDecorator(new ContextCopyingDecorator());return executor;
}
