TaskDecorator实现 @Async 上下文传递

最新推荐文章于 2024-10-25 11:23:51 发布

编码行者

最新推荐文章于 2024-10-25 11:23:51 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： spring 文章标签： java 开发语言

本文链接：https://blog.csdn.net/mbh12333/article/details/126629311

spring 专栏收录该内容

24 篇文章

订阅专栏

该博客详细介绍了如何在Spring Boot中配置并实现异步任务，包括创建ThreadPoolTaskExecutor，设置线程池参数，以及自定义AsyncUncaughtExceptionHandler来处理未捕获的异常。此外，还提供了一个AsyncTaskContextDecorator类，用于在异步任务中保持请求上下文。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

@Configuration
@EnableAsync
public class ListenerAsyncConfiguration implements AsyncConfigurer {
    private static final Logger log = Logger.getLogger(ListenerAsyncConfiguration.class);

    /**
     * 获取异步线程池执行对象
     *
     * @return
     */
    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
        taskExecutor.setCorePoolSize(10);
        taskExecutor.setMaxPoolSize(500);
        taskExecutor.setQueueCapacity(100000);
        taskExecutor.setThreadNamePrefix("task-async");
        taskExecutor.setTaskDecorator(new AsyncTaskContextDecorator());
        taskExecutor.initialize();
        return taskExecutor;
    }

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

    static class AsyncExceptionHandler implements AsyncUncaughtExceptionHandler {
        //手动处理捕获的异常
        @Override
        public void handleUncaughtException(Throwable throwable, Method method, Object... obj) {
            log.info("Exception message - " + throwable.getMessage());
            log.info("Method name - " + method.getName());
            for (Object param : obj) {
                log.info("Parameter value - " + param);
            }
        }
    }
}

public class AsyncTaskContextDecorator implements TaskDecorator {
    @Override
    public Runnable decorate(Runnable runnable) {
        RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
        Map<String, Object> container = CurrentContext.getContainer();
        try {
            return () -> {
                try {
                    CurrentContext.setContainer(container);
                    RequestContextHolder.setRequestAttributes(attributes);
                    runnable.run();
                } finally {
                    RequestContextHolder.resetRequestAttributes();
                    CurrentContext.clear();
                }
            };
        } catch (Exception e) {
            return runnable;
        }
    }
}