美文网首页
2019-05-23 @Async并行

2019-05-23 @Async并行

作者: 暖熊熊 | 来源:发表于2019-06-05 09:39 被阅读0次

1.常规代码顺序执行

如下定义三个任务,然后在单元测试依次调用三个任务,可以看到执行结果是先后执行三个任务,必须等待前一个任务结束后才能开始下一个任务,总共耗时6000多毫秒
测试类

package com.ghw.async;

import javafx.scene.paint.Stop;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.AutoConfigureOrder;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.util.StopWatch;

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

    @Autowired
    private Task task;

    private static StopWatch stopWatch = new StopWatch();

    @Test
    public void contextLoads() {
    }

    @Test
    public void test() throws Exception {
        task.setStopWatch(stopWatch);
        task.doTaskOne();
        task.doTaskTwo();
        task.doTaskThree();
        System.out.println(stopWatch.prettyPrint());
    }
}

Task类

package com.ghw.async;

import org.springframework.stereotype.Component;
import org.springframework.util.StopWatch;

import java.util.Random;
import java.util.concurrent.TimeUnit;

@Component
public class Task {

    private StopWatch stopWatch;

    public void setStopWatch(StopWatch stopWatch) {
        this.stopWatch = stopWatch;
    }

    public void doTaskOne() throws Exception {
        stopWatch.start("task1");
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
    }

    public void doTaskTwo() throws Exception {
        stopWatch.start("task2");
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
    }

    public void doTaskThree() throws Exception {
        stopWatch.start("task3");
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
    }
}

执行结果

StopWatch '': running time (millis) = 6039
-----------------------------------------
ms     %     Task name
-----------------------------------------
02014  033%  task1
02013  033%  task2
02012  033%  task3

2.采用@Async注解异步执行

Task类

package com.ghw.async;

import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.AsyncResult;
import org.springframework.stereotype.Component;
import org.springframework.util.StopWatch;

import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;

@Component
public class Task {

    @Async
    public Future<String> doTaskOne() throws Exception {
        StopWatch stopWatch = new StopWatch();
        String task = "task1";
        stopWatch.start(task);
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
        System.out.println(task + " spend time:" + stopWatch.getLastTaskTimeMillis());
        return new AsyncResult<>(task + " is finished");
    }

    @Async
    public Future<String> doTaskTwo() throws Exception {
        StopWatch stopWatch = new StopWatch();
        String task = "task2";
        stopWatch.start(task);
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
        System.out.println(task + " spend time:" + stopWatch.getLastTaskTimeMillis());
        return new AsyncResult<>(task + " is finished");
    }

    @Async
    public Future<String> doTaskThree() throws Exception {
        StopWatch stopWatch = new StopWatch();
        String task = "task3";
        stopWatch.start(task);
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
        System.out.println(task + " spend time:" + stopWatch.getLastTaskTimeMillis());
        return new AsyncResult<>(task + " is finished");
    }
}

测试类

package com.ghw.async;

import javafx.scene.paint.Stop;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.util.StopWatch;

import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;

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

    @Autowired
    private Task task;

    private static StopWatch stopWatch = new StopWatch();

    @Test
    public void contextLoads() {
    }

    @Test
    public void test() throws Exception {
        StopWatch stopWatch = new StopWatch();
        stopWatch.start("all");
        Future<String> task1 = task.doTaskOne();
        Future<String> task2 = task.doTaskTwo();
        Future<String> task3 = task.doTaskThree();

        while (true) {
            if (task1.isDone() && task2.isDone() && task3.isDone()) {
                // 三个任务都调用完成,退出循环等待
                break;
            }
        }
        stopWatch.stop();
        System.out.println("all task spend time:" + stopWatch.getLastTaskTimeMillis());
    }
}

执行结果

task3 spend time:2014
task1 spend time:2014
task2 spend time:2014
all task spend time:2060

可以看出只需要2060毫秒,三个任务就全部执行完成了,相比第一种方法,能够提高3倍速度。
其中使用Future<T>来返回异步调用的结果。在while循环中判断,当三个任务全部完成的时候,退出while循环。

采用自己配置的线程池

package com.ghw.async;

import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import org.springframework.util.StopWatch;

import java.util.concurrent.TimeUnit;

@Slf4j
@Component
public class Task {

    @Async(value = "taskExecutor")
    public void doTaskOne() throws Exception {
        log.info("开始做任务一");
        StopWatch stopWatch = new StopWatch();
        String task = "task1";
        stopWatch.start(task);
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
        log.info("完成任务一,耗时:" + (stopWatch.getLastTaskTimeMillis()) + "毫秒");
    }

    @Async(value = "taskExecutor")
    public void doTaskTwo() throws Exception {
        log.info("开始做任务二");
        StopWatch stopWatch = new StopWatch();
        String task = "task2";
        stopWatch.start(task);
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
        log.info("完成任务二,耗时:" + (stopWatch.getLastTaskTimeMillis()) + "毫秒");
    }

    @Async(value = "taskExecutor")
    public void doTaskThree() throws Exception {
        log.info("开始做任务三");
        StopWatch stopWatch = new StopWatch();
        String task = "task3";
        stopWatch.start(task);
        TimeUnit.SECONDS.sleep(2);
        stopWatch.stop();
        log.info("完成任务三,耗时:" + (stopWatch.getLastTaskTimeMillis()) + "毫秒");
    }
}
package com.ghw.async;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

@SpringBootApplication
public class AsyncApplication {

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

    @EnableAsync
    @Configuration
    class TaskPoolConfig {

        @Bean("taskExecutor")
        public Executor taskExecutor() {
            ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
            executor.setCorePoolSize(10);
            executor.setMaxPoolSize(20);
            executor.setQueueCapacity(200);
            executor.setKeepAliveSeconds(60);
            executor.setThreadNamePrefix("taskExecutor-");
            executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
            return executor;
        }
    }
}

单元测试

package com.ghw.async;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.util.StopWatch;

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

    @Autowired
    private Task task;

    @Autowired
    private ThreadPoolTaskExecutor taskExecutor;

    private static StopWatch stopWatch = new StopWatch();

    @Test
    public void contextLoads() {
    }

    @Test
    public void test() throws Exception {
        task.doTaskOne();
        task.doTaskTwo();
        task.doTaskThree();
        Thread.currentThread().join();
    }
}

相关文章

网友评论

      本文标题:2019-05-23 @Async并行

      本文链接:https://www.haomeiwen.com/subject/pmobzqtx.html