@Async 注解自定义线程池进行使用

@Async 自定义线程池

自定义线程池逻辑

/**
 * @author Greenarrow
 * @date 2023/9/22
 * @description @Async 自定义线程池
 */
@Component
@Slf4j
public class ThreadPoolConfig implements AsyncConfigurer {

    /**
     * 阻塞系数 0-1 越大线程数越多
     */
    private static final float BLOCKING_COEFFICIENT = 0.85F;
    /**
     * 默认线程池配置 TODO 可改为配置项
     * @return
     */
    @Override
    @Bean(CommonConstant.ASYNC_EXECUTOR_CUSTOM)
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor threadPool = new ThreadPoolTaskExecutor();
        int poolSize = (int) (Runtime.getRuntime().availableProcessors() / (1 - BLOCKING_COEFFICIENT));
        threadPool.setCorePoolSize(poolSize);
        threadPool.setMaxPoolSize(poolSize);
        threadPool.setQueueCapacity(poolSize*2);
        threadPool.setKeepAliveSeconds(60);
        threadPool.setWaitForTasksToCompleteOnShutdown(true);
        threadPool.setAwaitTerminationSeconds(60 * 15);
        threadPool.setThreadNamePrefix("async-task-thread-pool-");
        threadPool.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        threadPool.initialize();
       log.info(">>>>>>>>>>>>>>> 开启自定义异步线程池");
        return threadPool;
    }

    @Override
    public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
        return new MyAsyncExceptionHandler();
    }
    /**
     * 自定义异常处理类
     *
     *
     */
    static class MyAsyncExceptionHandler implements AsyncUncaughtExceptionHandler {

        //手动处理捕获的异常
        @Override
        public void handleUncaughtException(Throwable throwable, Method method, Object... obj) {
            log.error(">>>>>>>>>>>>>>> 自定义线程池捕获线程异常信息 {}",throwable.toString());
            log.error("Exception message - " + throwable.getMessage());
            log.error("Method name - " + method.getName());
            for (Object param : obj) {
                log.info("Parameter value - " + param);
            }
        }

    }
}

使用

@Async(CommonConstant.ASYNC_EXECUTOR_CUSTOM)

其他

    /**
     * 自定义线程池名称
     */
    public static final String ASYNC_EXECUTOR_CUSTOM = "async-executor-custom";