Java教程之Spring的Async注解线程池扩展方案

目录

• Spring的Async注解线程池扩展方案
  • 目录
  • 1. 扩展目的
  • 2. 扩展实现
    • 2.1 扩展Async注解的执行拦截器AnnotationAsyncExecutionInterceptor
    • 2.2 扩展Async注解的Spring代理顾问AsyncAnnotationAdvisor
    • 2.3 扩展Async注解的 Spring Bean 后置处理器AsyncAnnotationBeanPostProcessor
    • 2.4 扩展代理异步配置类ProxyAsyncConfiguration
    • 2.5 扩展异步代理配置选择器AsyncConfigurationSelector
    • 2.6 扩展异步启动注解@EnableAsync
  • 3. 额外扩展：给@Async注解代理指定线程池

1. 扩展目的

1. 异步调用，改用Spring提供的@Aysnc注解实现，代替手写线程池执行。
2. 在实际场景中，可能会遇到需要将主线程的一些个性化参数、变量、数据传递到子线程中使用的需求。
3. InheritableThreadLocal可以解决子线程继承父线程值的需求，但是它存在一些问题。
   1. SessionUser.SESSION_USER是中台提供，无法修改。
   2. InheritableThreadLocal在线程池机制应用中并不友好，不及时在子线程中清除的话，会造成线程安全问题。

实现思路有两种：

1. 针对ThreadLocal进行扩展，并说服中台统一改用扩展后的ThreadLocal。
2. 针对@EnableAsync和@Async注解进行扩展，将手动copy的代码写入到Spring代理类中。

第一种要跟中台打交道，就很烦，能够天平自己独立解决，就自己解决。第二种会是一个不错的选择，扩展实现也并不困难。

2. 扩展实现

2.1 扩展Async注解的执行拦截器AnnotationAsyncExecutionInterceptor

类全名：org.springframework.scheduling.annotation.AnnotationAsyncExecutionInterceptor

从调试记录可以分析得出AnnotationAsyncExecutionInterceptor#invoke方法，正是创建异步任务并且执行异步任务的核心代码所在，我们要做的就是重写这个方法，将父线程的运行参数手动copy到子线程任务体中。

2.2 扩展Async注解的Spring代理顾问AsyncAnnotationAdvisor