spring boot @Async异步注解上下文透传

方案一：

继承线程池，重写相应的方法，透传上下文。

方案二：（推荐）

线程池ThreadPoolTaskExecutor，有一个TaskDecorator装饰器，实现这个接口，透传上下文。

方案一：继承线程池，重写相应的方法，透传上下文。

1、ThreadPoolTaskExecutor spring封装的线程池

ThreadPoolTaskExecutor 线程池代码如下：

https://www.jianshu.com/p/2a58e1df2bbb

@Bean(ExecutorConstant.simpleExecutor_3)

public Executor asyncExecutor3() {

MyThreadPoolTaskExecutor taskExecutor = new MyThreadPoolTaskExecutor();

taskExecutor.setCorePoolSize(corePoolSize);

taskExecutor.setMaxPoolSize(maxPoolSize);

taskExecutor.setQueueCapacity(queueCapacity);

taskExecutor.setThreadNamePrefix(threadNamePrefix_3);

taskExecutor.initialize();

return taskExecutor;

}

//------- 继承父类 重写对应的方法 start

class MyCallable implements Callable {

private Callable task;

private RequestAttributes context;

public MyCallable(Callable task, RequestAttributes context) {

this.task = task;

this.context = context;

}

@Override

public T call() throws Exception {

if (context != null) {

RequestContextHolder.setRequestAttributes(context);

}

try {

return task.call();

} finally {

RequestContextHolder.resetRequestAttributes();

}

}

}

class MyThreadPoolTaskExecutor extends ThreadPoolTaskExecutor{

@Override

public Future submit(Callable task) {

return super.submit(new MyCallable(task, RequestContextHolder.currentRequestAttributes()));

}

@Override

public ListenableFuture submitListenable(Callable task) {

return super.submitListenable(new MyCallable(task, RequestContextHolder.currentRequestAttributes()));

}

}

//------- 继承父类 重写对应的方法 end

1、MyCallable是继承Callable，创建MyCallable对象的时候已经把Attributes对象赋值给属性context了（创建MyCallable对象的时候因为实在当前主线程创建的，所以是能获取到请求的Attributes），在执行call方法前，先执行了RequestContextHolder.setRequestAttributes(context); 【把这个MyCallable对象的属性context 设置到setRequestAttributes中】 所以在执行具体业务时，当前线程（子线程）就能取得主线程的Attributes

2、MyThreadPoolTaskExecutor类是继承了ThreadPoolTaskExecutor 重写了submit和submitListenable方法

为什么是重写submit和submitListenable这两个方法了？

@Async AOP源码的方法位置是在：AsyncExecutionInterceptor.invoke

doSubmit方法能看出来

无返回值调用的是线程池方法：submit()

有返回值，根据不同的返回类型也知道：

1. 返回值类型是：Future.class 调用的是方法：submit()

2. 返回值类型是：ListenableFuture.class 调用的方法是：submitListenable(task)

3. 返回值类型是：CompletableFuture.class调用的是CompletableFuture.supplyAsync这个在异步注解中暂时用不上的，就不考虑重写了。

public Object invoke(final MethodInvocation invocation) throws Throwable {

Class<?> targetClass = (invocation.getThis() != null ? AopUtils.getTargetClass(invocation.getThis()) : null);

Method specificMethod = ClassUtils.getMostSpecificMethod(invocation.getMethod(), targetClass);

final Method userDeclaredMethod = BridgeMethodResolver.findBridgedMethod(specificMethod);

AsyncTaskExecutor executor = determineAsyncExecutor(userDeclaredMethod);

if (executor == null) {

throw new IllegalStateException(

“No executor specified and no default executor set on AsyncExecutionInterceptor either”);

}

Callable task = () -> {

try {

Object result = invocation.proceed();

if (result instanceof Future) {

return ((Future<?>) result).get();

}

}

catch (ExecutionException ex) {

handleError(ex.getCause(), userDeclaredMethod, invocation.getArguments());

}

catch (Throwable ex) {

handleError(ex, userDeclaredMethod, invocation.getArguments());

}

return null;

};

return doSubmit(task, executor, invocation.getMethod().getReturnType());

}

@Nullable

protected Object doSubmit(Callable task, AsyncTaskExecutor executor, Class<?> returnType) {

if (CompletableFuture.class.isAssignableFrom(returnType)) {

return CompletableFuture.supplyAsync(() -> {

try {

return task.call();

}

catch (Throwable ex) {

throw new CompletionException(ex);

}

}, executor);

}

else if (ListenableFuture.class.isAssignableFrom(returnType)) {

return ((AsyncListenableTaskExecutor) executor).submitListenable(task);

}

else if (Future.class.isAssignableFrom(returnType)) {

return executor.submit(task);

}

else {

executor.submit(task);

return null;

}

}

---

2、ThreadPoolExecutor 原生线程池https://www.jianshu.com/p/2a58e1df2bbb

ThreadPoolExecutor线程池代码如下：

//------- ThreadPoolExecutor 继承父类 重写对应的方法 start

class MyRunnable implements Runnable {

private Runnable runnable;

private RequestAttributes context;

public MyRunnable(Runnable runnable, RequestAttributes context) {

this.runnable = runnable;

this.context = context;

}

@Override

public void run() {

if (context != null) {

RequestContextHolder.setRequestAttributes(context);

}

try {

runnable.run();

} finally {

RequestContextHolder.resetRequestAttributes();

}

}

}

class MyThreadPoolExecutor extends ThreadPoolExecutor{

@Override

public void execute(Runnable command) {

if(!(command instanceof MyRunnable)){

command = new MyRunnable(command,RequestContextHolder.currentRequestAttributes())

}

super.execute(command);

}

public MyThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue);

}

public MyThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, ThreadFactory threadFactory) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory);

}

public MyThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, RejectedExecutionHandler handler) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, handler);

}

public MyThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, ThreadFactory threadFactory, RejectedExecutionHandler handler) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory, handler);

}

}

//------- ThreadPoolExecutor 继承父类 重写对应的方法 end

像ThreadPoolExecutor主要重写execute方法，在启动新线程的时候先把Attributes取到放到MyRunnable对象的一个属性中，MyRunnable在具体执行run方法的时候，把属性Attributes赋值到子线程中，当run方法执行完了在把Attributes清空掉。

为什么只要重写了execute方法就可以了？

ThreadPoolExecutor大家都知道主要是由submit和execute方法来执行的。

看ThreadPoolExecutor类的submit具体执行方法是由父类AbstractExecutorService#submit来实现。

具体代码在下面贴出来了，可以看到submit实际上最后调用的还是execute方法，所以我们重写execute方法就好了。

submit方法路径及源码：

java.util.concurrent.AbstractExecutorService#submit(java.lang.Runnable)

public Future<?> submit(Runnable task) {

if (task == null) throw new NullPointerException();

RunnableFuture ftask = newTaskFor(task, null);

execute(ftask);

return ftask;

}

/**

• @throws RejectedExecutionException {@inheritDoc}

• @throws NullPointerException {@inheritDoc}

*/

public Future submit(Runnable task, T result) {

if (task == null) throw new NullPointerException();

RunnableFuture ftask = newTaskFor(task, result);

execute(ftask);

return ftask;

}

/**

• @throws RejectedExecutionException {@inheritDoc}

• @throws NullPointerException {@inheritDoc}

*/

public Future submit(Callable task) {

if (task == null) throw new NullPointerException();

RunnableFuture ftask = newTaskFor(task);

execute(ftask);

return ftask;
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

言尽于此，完结

无论是一个初级的 coder，高级的程序员，还是顶级的系统架构师，应该都有深刻的领会到设计模式的重要性。

• 第一，设计模式能让专业人之间交流方便，如下：

程序员A：这里我用了XXX设计模式

程序员B：那我大致了解你程序的设计思路了

• 第二，易维护

项目经理：今天客户有这样一个需求…

程序员：明白了，这里我使用了XXX设计模式，所以改起来很快

• 第三，设计模式是编程经验的总结

程序员A：B，你怎么想到要这样去构建你的代码

程序员B：在我学习了XXX设计模式之后，好像自然而然就感觉这样写能避免一些问题

• 第四，学习设计模式并不是必须的

程序员A：B，你这段代码使用的是XXX设计模式对吗？

程序员B：不好意思，我没有学习过设计模式，但是我的经验告诉我是这样写的

从设计思想解读开源框架，一步一步到Spring、Spring5、SpringMVC、MyBatis等源码解读，我都已收集整理全套，篇幅有限，这块只是详细的解说了23种设计模式，整理的文件如下图一览无余！

搜集费时费力，能看到此处的都是真爱！
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
所以改起来很快

• 第三，设计模式是编程经验的总结

程序员A：B，你怎么想到要这样去构建你的代码

程序员B：在我学习了XXX设计模式之后，好像自然而然就感觉这样写能避免一些问题

• 第四，学习设计模式并不是必须的

程序员A：B，你这段代码使用的是XXX设计模式对吗？

程序员B：不好意思，我没有学习过设计模式，但是我的经验告诉我是这样写的

[外链图片转存中…(img-CK5CFfMC-1713291770522)]

从设计思想解读开源框架，一步一步到Spring、Spring5、SpringMVC、MyBatis等源码解读，我都已收集整理全套，篇幅有限，这块只是详细的解说了23种设计模式，整理的文件如下图一览无余！

[外链图片转存中…(img-DZ7oEfcA-1713291770522)]

搜集费时费力，能看到此处的都是真爱！
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！