Spring Boot 中启用异步调用

在Java中一般开发程序都会同步调用的，程序中代码是一行一行执行下去的，每一行代码需要等待上一行代码执行完成才能开始执行。

在异步编程中，代码执行不是阻塞的，在方法调用中不需要等待所有代码执行完毕就可以返回。在某些场景中，异步调用可以提升用户响应的体验感。

那么如何在Springboot中开启异步调用呢？

@EnableAsync

在启动类上增加@EnableAsync，表示启动异步支持。

@EnableAsync
@SpringBootApplication
public class ABerApplication
{
    public static void main(String[] args)
    {
        SpringApplication.run(ABerApplication.class, args);
        System.out.println("(♥◠‿◠)ﾉﾞ  ABer 启动成功");
    }
}

创建线程池

异步调用不会因为主线程阻塞而阻塞，是因为异步调用每次都会新建一个线程去执行，这样新建线程或销毁线程会有一定的性能影响，我们可以定义一个线程池来管理这些异步线程。

@Configuration
public class ThreadPoolConfig {

    // 核心线程池大小
    private int corePoolSize = 50;

    // 最大可创建的线程数
    private int maxPoolSize = 200;

    // 队列最大长度
    private int queueCapacity = 1000;

    // 线程池维护线程所允许的空闲时间
    private int keepAliveSeconds = 300;

    // 等待终止秒数
    private int awaitTerminationSeconds = 60;

    @Bean(name = "asyncThreadPoolTaskExecutor")
    public ThreadPoolTaskExecutor asyncThreadPoolTaskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(maxPoolSize);
        executor.setMaxPoolSize(maxPoolSize);
        executor.setKeepAliveSeconds(keepAliveSeconds);
        executor.setQueueCapacity(queueCapacity);
        executor.setAwaitTerminationSeconds(awaitTerminationSeconds);
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.setThreadGroupName("asyncThread");
        
        // 线程池对拒绝任务(无线程可用)的处理策略
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());

        executor.initialize();
        return executor;
    }
}

异步方法

在异步方法上增加@Async，并指定刚刚定义的线程池Bean名称。

@Service
public class PullMessageService {

    /**
     * 文件下发
     *
     * @param msg
     */
    @Async("asyncThreadPoolTaskExecutor")
    public void pushFile(Message msg) {
        System.out.println(msg.toJsonString());
    }
}

异步回调

如果异步方法是有返回值的，那么就需要用到Future来接收回调值。再通过Future.get()方法接收异步调用的返回值。
Future.get()方法是阻塞方法，只有异步调用返回了结果，才会往下执行。
Future还有一个get的重载方法Future.get(long timeout, TimeUnit unit)，通过这个重载方法我们可以设置异步调用的超时时间，即如果异步方法在设定时间范围内没有执行完毕的话，会直接抛出java.util.concurrent.TimeoutException异常。

@Service
public class PullMessageService {

    /**
     * 文件下发
     *
     * @param msg
     */
    @Async("asyncThreadPoolTaskExecutor")
    public Future<String> pushFile(Message msg) {
        System.out.println(msg.toJsonString());
        logger.info("异步方法的线程名称：{}", Thread.currentThread().getName()); 
        return new AsyncResult<>(msg.toJsonString());
    }
}