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();
}
}