SpringBoot执行有返回值的异步任务

欢迎访问个人网站

Springboot如何使用多线程处理异步任务，并且是代返回值的。比如，我一个Controller层的接口，调用到Service层，在对应的Service方法中有三个方法，这三个方法都是去调Dao层查询数据库数据，每个查询平均耗时5s,最后将这三个查询的结果进行合并计算。如果不采用异步查询的至少需要15s的时间来处理这个请求。此时我们可以使用异步任务方式来操作，可以减少10s左右的时间。

在Springboot中使用异步需要开启异步支持（@EnableAsync），代码如下

@SpringBootApplication
@EnableAsync
public class PublisherApplication {

    public static void main(String[] args) {
        SpringApplication.run(PublisherApplication.class, args);
    }

}

然后增加一个配置类，不过这个Springboot也会默认配置，一般我们使用线程池是都不使用默认的，而是使用自定义的：

/**
 * @author Mr. Zhang
 * @description 异步线程池
 * @date 2019-04-19 14:21
 * @website https://www.zhangguimin.cn
 */
@Configuration
public class AsyncConfig {

    private static final int THREADS = Runtime.getRuntime().availableProcessors();

    @Bean("taskExecutor")
    public Executor execute() {
        System.out.println(THREADS);
        Executor executor = new ThreadPoolExecutor(THREADS, 2 * THREADS, 5, TimeUnit.SECONDS, new LinkedBlockingQueue<>(1024));
        return executor;
    }

}

此配置为IO密集型服务器参数，即最大线程数为CPU核数*2（THREADS为CPU核数），其他拒绝策略默认即可。
在之后的异步任务中使用此出定义的线程池taskExecutor。
然后就是异步任务，在对应的任务方法上标注@Async(“taskExecutor”)即可，表示使用taskExecutor线程池中线程执行异步任务。代码参考：

@Service
public class TaskServer {

    @Async("taskExecutor")
    public Future<Integer> asyncTask(CountDownLatch latch) {
        try {
            Thread.sleep(3000L);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            latch.countDown();
        }
        return new AsyncResult<>(10);
    }

    @Async("taskExecutor")
    public Future<Integer> asyncTask2(CountDownLatch latch) {
        try {
            Thread.sleep(3000L);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            latch.countDown();
        }
        return new AsyncResult<>(20);
    }

    @Async("taskExecutor")
    public Future<Integer> asyncTask3(CountDownLatch latch) {
        try {
            Thread.sleep(3000L);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            latch.countDown();
        }
        return new AsyncResult<>(30);
    }
}

注意的是，需要在类上标注为组件注解，此处为@Service。
在这三个方法中执行sleep，使线程都是sleep 3秒的时间。然后参数为CountDownLatch，在每次执行完就countDown()，然后返回值为Future，如果执行的异步方法没有返回值可以不需要CountDownLatch和返回值就行。
在对应的测试类中执行

@RunWith(SpringRunner.class)
@SpringBootTest
public class PublisherApplicationTests {

    @Autowired
    private TaskServer taskServer;

    @Test
    public void test() {
        CountDownLatch latch = new CountDownLatch(3);
        LocalTime time = LocalTime.now();
        Future<Integer> integerFuture = taskServer.asyncTask(latch);
        Future<Integer> integerFuture1 = taskServer.asyncTask2(latch);
        Future<Integer> integerFuture2 = taskServer.asyncTask3(latch);
        try {
            latch.await();
            LocalTime end =LocalTime.now();
            System.out.println(Duration.between(time, end));

            Integer integer = integerFuture.get();
            Integer integer1 = integerFuture1.get();
            Integer integer2 = integerFuture2.get();
            int i = integer + integer1 + integer2;
            System.out.println(i);
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (ExecutionException e) {
            e.printStackTrace();
        }
    }

}

注意CountDownLatch latch = new CountDownLatch(3);有三个异步方法传入的值为3，在调用的异步方法后面需要latch.await();
一定不要在latch.await();前去get()结果值，这样会导致异步方法强制执行。
另外，有的人也使用while循环，在所有结果出来前一致在循环等待，类似自旋锁的原理，对资源消耗较高，还有，如果一个任务出现错误，就可能会造成一只循环。而CountDownLatch可在await()方法中传入等待时间，如果超过这个时间就会结束等待，直接完成下面的操作。