Spring Boot项目中TaskDecorator的应用实践

一、前言

TaskDecorator是一个执行回调方法的装饰器，主要应用于传递上下文，或者提供任务的监控/统计信息，可以用于处理子线程与主线程间数据传递的问题。

二、开发示例

1.自定义TaskDecorator

import org.springframework.core.task.TaskDecorator;

public class MyTaskDecorator implements TaskDecorator {
   
    @Override
    public Runnable decorate(Runnable runnable) {
   
        return ()->{
   
            System.out.println("线程执行前，资源设置。。。。");
            runnable.run();
            System.out.println("线程执行后，资源清理。。。。");
        };
    }
}

2. 自定义线程池，设置TaskDecorator

ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setTaskDecorator(new MyTaskDecorator());
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
executor.setWaitForTasksToCompleteOnShutdown(true);
executor.initialize();

executor.execute(()->{
   
    System.out.println("我是一个测试线程。。。");
});

executor.shutdown();

3. 测试