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

好的，下面是一个简单的示例：

假设我们有一个 UserService，其中有一个方法 sendEmail，它需要异步地发送电子邮件。现在我们想要将当前用户的信息透传给异步任务中使用的线程。

首先，我们需要在异步方法上添加 @Async 注解，并在配置类中启用异步支持：

@Configuration
@EnableAsync
public class AppConfig implements AsyncConfigurer {

    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(100);
        executor.setQueueCapacity(10);
        executor.initialize();
        return executor;
    }

    @Override
    public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
        return new CustomAsyncExceptionHandler();
    }
}

在上面的示例中，我们创建了一个 ThreadPoolTaskExecutor，它将用于执行异步任务。我们还实现了 AsyncConfigurer 接口，并覆盖了 getAsyncExecutor 和 getAsyncUncaughtExceptionHandler 方法，以提供自定义的 Executor 和异常处理程序。

现在我们需要将当前用户信息存储在一个 ThreadLocal 对象中。这可以通过一个拦截器来实现：

public class UserContextInterceptor extends HandlerInterceptorAdapter {

    private final ThreadLocal<String> userThreadLocal = new ThreadLocal<>();

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String currentUser = request.getHeader("X-User");
        userThreadLocal.set(currentUser);
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        userThreadLocal.remove();
    }

    public String getCurrentUser() {
        return userThreadLocal.get();
    }
}

在上面的示例中，我们创建了一个 UserContextInterceptor，它将在每个请求的开始和结束时执行。在 preHandle 方法中，我们从请求头中获取当前用户信息，并将其存储在一个 ThreadLocal 对象中。在 afterCompletion 方法中，我们将删除该信息，以避免内存泄漏。

现在，我们可以在 UserService 的 sendEmail 方法中使用 UserContextInterceptor 中存储的当前用户信息：

@Service
public class UserService {

    @Autowired
    private JavaMailSender mailSender;

    @Autowired
    private UserContextInterceptor userContextInterceptor;

    @Async
    public void sendEmail(String to, String subject, String text) {
        String currentUser = userContextInterceptor.getCurrentUser();
        // 使用当前用户信息发送电子邮件
        // ...
    }
}

在上面的示例中，我们使用 @Autowired 注解将 UserContextInterceptor 注入到 UserService 中。在 sendEmail 方法中，我们从 UserContextInterceptor 中获取当前用户信息，并在发送电子邮件时使用它。

通过这种方式，我们可以将当前用户信息透传给异步任务中使用的线程。