问题
在我们的日常开发中，可以通过 @Async 注解，很方便地启动一个异步线程。
比如现在有一个用户注册成功后，发送欢迎邮件的需求，在用户注册成功以后，便可以启动一个异步线程，在这个线程中调用邮件服务给用户发消息。
这样，即使邮件服务出了问题，也不会影响到当前用户的注册体验。
问题在于，异步线程无法获取原线程的数据信息，如果每次通过手写参数传递又会比较麻烦，所以我们希望通过某种形式，让数据可以自动传递给子线程。
解决方案
1，新建一个类，重写 TaskDecorator 类的 decorate 的方法
public class MDCContextDecorator implements TaskDecorator {
@Override public Runnable decorate(Runnable runnable) { //RequestAttributes context = RequestContextHolder.currentRequestAttributes(); //这里获取是mdc的上下文，也可以获取RequestContextHolder，具体根据你的业务需要操作即可 Map<String,String> previous = MDC.getCopyOfContextMap(); return () -> { try { if (previous != null) { MDC.setContextMap(previous); } runnable.run(); } finally { //务必记得clear，否则可能会产生内存泄露 MDC.clear(); } }; }}
复制代码
2，在自定义的线程池中设置我们的自定义装饰器
@Bean public Executor taskExecutor() { ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); // 设置核心线程数 executor.setCorePoolSize(20); // 设置最大线程数 executor.setMaxPoolSize(30); // 设置队列容量 executor.setQueueCapacity(1000); // 设置线程活跃时间（秒） executor.setKeepAliveSeconds(60); // 设置默认线程名称 executor.setThreadNamePrefix("job-"); // 设置拒绝策略 executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); // 等待所有任务结束后再关闭线程池 全局线程池不能关闭 executor.setWaitForTasksToCompleteOnShutdown(true); //设置我们自定义的Decorator executor.setTaskDecorator(new MDCContextDecorator()); return executor; }
复制代码
原理探究
Spring 给我们预留一个任务装饰器 TaskDecorator，通过这个任务装饰器，可以像 AOP 一样，对线程做一些功能增强。
在 ThreadPoolTaskExecutor 的源码中，initializeExecutor 方法对线程池进行初始化，会判断是否有装饰器的实现。
protected ExecutorService initializeExecutor( ThreadFactory threadFactory, RejectedExecutionHandler rejectedExecutionHandler) { BlockingQueue<Runnable> queue = createQueue(this.queueCapacity); ThreadPoolExecutor executor; if (this.taskDecorator != null) { //如果进行了装饰，就转而去执行自定义的装饰方法 executor = new ThreadPoolExecutor(this.corePoolSize, this.maxPoolSize, this.keepAliveSeconds, TimeUnit.SECONDS, queue, threadFactory, rejectedExecutionHandler) { @Override public void execute(Runnable command) { Runnable decorated = taskDecorator.decorate(command); if (decorated != command) { decoratedTaskMap.put(decorated, command); } super.execute(decorated); } }; } else { executor = new ThreadPoolExecutor( this.corePoolSize, this.maxPoolSize, this.keepAliveSeconds, TimeUnit.SECONDS, queue, threadFactory, rejectedExecutionHandler);
} if (this.allowCoreThreadTimeOut) { executor.allowCoreThreadTimeOut(true); } this.threadPoolExecutor = executor; return executor; }
复制代码
总结
利用 ThreadPoolTaskExecutor 的 TaskDecorator，动态的给一个对象添加一些额外的功能，比生成子类会更加灵活。在我们平常的编码过程中，也建议大家尝试使用装饰模式优化我们的代码。
评论