通過(guò)上一篇：配置@Async異步任務(wù)的線程池的介紹睹耐，你應(yīng)該已經(jīng)了解到異步任務(wù)的執(zhí)行背后有一個(gè)線程池來(lái)管理執(zhí)行任務(wù)辐赞。為了控制異步任務(wù)的并發(fā)不影響到應(yīng)用的正常運(yùn)作，我們必須要對(duì)線程池做好相應(yīng)的配置疏橄，防止資源的過(guò)渡使用占拍。除了默認(rèn)線程池的配置之外略就，還有一類場(chǎng)景，也是很常見(jiàn)的晃酒，那就是多任務(wù)情況下的線程池隔離表牢。
什么是線程池的隔離，為什么要隔離
可能有的小伙伴還不太了解什么是線程池的隔離贝次，為什么要隔離崔兴？。所以蛔翅，我們先來(lái)看看下面的場(chǎng)景案例：
@RestController
public class HelloController {
@Autowired
private AsyncTasks asyncTasks;
@GetMapping("/api-1")
public String taskOne() {
CompletableFuture<String> task1 = asyncTasks.doTaskOne("1");
CompletableFuture<String> task2 = asyncTasks.doTaskOne("2");
CompletableFuture<String> task3 = asyncTasks.doTaskOne("3");
CompletableFuture.allOf(task1, task2, task3).join();
return "";
}
@GetMapping("/api-2")
public String taskTwo() {
CompletableFuture<String> task1 = asyncTasks.doTaskTwo("1");
CompletableFuture<String> task2 = asyncTasks.doTaskTwo("2");
CompletableFuture<String> task3 = asyncTasks.doTaskTwo("3");
CompletableFuture.allOf(task1, task2, task3).join();
return "";
}
}
上面的代碼中敲茄，有兩個(gè)API接口，這兩個(gè)接口的具體執(zhí)行邏輯中都會(huì)把執(zhí)行過(guò)程拆分為三個(gè)異步任務(wù)來(lái)實(shí)現(xiàn)山析。
好了堰燎，思考一分鐘，想一下笋轨。如果這樣實(shí)現(xiàn)秆剪，會(huì)有什么問(wèn)題嗎？
上面這段代碼爵政，在API請(qǐng)求并發(fā)不高仅讽，同時(shí)如果每個(gè)任務(wù)的處理速度也夠快的時(shí)候，是沒(méi)有問(wèn)題的钾挟。但如果并發(fā)上來(lái)或其中某幾個(gè)處理過(guò)程扯后腿了的時(shí)候洁灵。這兩個(gè)提供不相干服務(wù)的接口可能會(huì)互相影響。比如：假設(shè)當(dāng)前線程池配置的最大線程數(shù)有2個(gè)掺出，這個(gè)時(shí)候/api-1接口中task1和task2處理速度很慢徽千，阻塞了；那么此時(shí)蛛砰，當(dāng)用戶調(diào)用api-2接口的時(shí)候罐栈，這個(gè)服務(wù)也會(huì)阻塞！
造成這種現(xiàn)場(chǎng)的原因是：默認(rèn)情況下泥畅，所有用@Async創(chuàng)建的異步任務(wù)都是共用的一個(gè)線程池荠诬，所以當(dāng)有一些異步任務(wù)碰到性能問(wèn)題的時(shí)候，是會(huì)直接影響其他異步任務(wù)的位仁。
為了解決這個(gè)問(wèn)題柑贞，我們就需要對(duì)異步任務(wù)做一定的線程池隔離，讓不同的異步任務(wù)互不影響聂抢。
不同異步任務(wù)配置不同線程池
下面钧嘶，我們就來(lái)實(shí)際操作一下！
第一步：初始化多個(gè)線程池琳疏，比如下面這樣：
@EnableAsync
@Configuration
public class TaskPoolConfig {
@Bean
public Executor taskExecutor1() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(2);
executor.setMaxPoolSize(2);
executor.setQueueCapacity(10);
executor.setKeepAliveSeconds(60);
executor.setThreadNamePrefix("executor-1-");
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
return executor;
}
@Bean
public Executor taskExecutor2() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(2);
executor.setMaxPoolSize(2);
executor.setQueueCapacity(10);
executor.setKeepAliveSeconds(60);
executor.setThreadNamePrefix("executor-2-");
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
return executor;
}
}
注意：這里特地用
executor.setThreadNamePrefix設(shè)置了線程名的前綴有决，這樣可以方便觀察后面具體執(zhí)行的順序闸拿。
第二步：創(chuàng)建異步任務(wù)，并指定要使用的線程池名稱
@Slf4j
@Component
public class AsyncTasks {
public static Random random = new Random();
@Async("taskExecutor1")
public CompletableFuture<String> doTaskOne(String taskNo) throws Exception {
log.info("開(kāi)始任務(wù)：{}", taskNo);
long start = System.currentTimeMillis();
Thread.sleep(random.nextInt(10000));
long end = System.currentTimeMillis();
log.info("完成任務(wù)：{}书幕，耗時(shí)：{} 毫秒", taskNo, end - start);
return CompletableFuture.completedFuture("任務(wù)完成");
}
@Async("taskExecutor2")
public CompletableFuture<String> doTaskTwo(String taskNo) throws Exception {
log.info("開(kāi)始任務(wù)：{}", taskNo);
long start = System.currentTimeMillis();
Thread.sleep(random.nextInt(10000));
long end = System.currentTimeMillis();
log.info("完成任務(wù)：{}新荤，耗時(shí)：{} 毫秒", taskNo, end - start);
return CompletableFuture.completedFuture("任務(wù)完成");
}
}
這里@Async注解中定義的taskExecutor1和taskExecutor2就是線程池的名字。由于在第一步中台汇，我們沒(méi)有具體寫(xiě)兩個(gè)線程池Bean的名稱苛骨，所以默認(rèn)會(huì)使用方法名，也就是taskExecutor1和taskExecutor2苟呐。
第三步：寫(xiě)個(gè)單元測(cè)試來(lái)驗(yàn)證下痒芝，比如下面這樣：
@Slf4j
@SpringBootTest
public class Chapter77ApplicationTests {
@Autowired
private AsyncTasks asyncTasks;
@Test
public void test() throws Exception {
long start = System.currentTimeMillis();
// 線程池1
CompletableFuture<String> task1 = asyncTasks.doTaskOne("1");
CompletableFuture<String> task2 = asyncTasks.doTaskOne("2");
CompletableFuture<String> task3 = asyncTasks.doTaskOne("3");
// 線程池2
CompletableFuture<String> task4 = asyncTasks.doTaskTwo("4");
CompletableFuture<String> task5 = asyncTasks.doTaskTwo("5");
CompletableFuture<String> task6 = asyncTasks.doTaskTwo("6");
// 一起執(zhí)行
CompletableFuture.allOf(task1, task2, task3, task4, task5, task6).join();
long end = System.currentTimeMillis();
log.info("任務(wù)全部完成，總耗時(shí)：" + (end - start) + "毫秒");
}
}
在上面的單元測(cè)試中牵素，一共啟動(dòng)了6個(gè)異步任務(wù)严衬，前三個(gè)用的是線程池1，后三個(gè)用的是線程池2两波。
先不執(zhí)行瞳步，根據(jù)設(shè)置的核心線程2和最大線程數(shù)2闷哆，來(lái)分析一下腰奋，大概會(huì)是怎么樣的執(zhí)行情況？
- 線程池1的三個(gè)任務(wù)抱怔，task1和task2會(huì)先獲得執(zhí)行線程劣坊，然后task3因?yàn)闆](méi)有可分配線程進(jìn)入緩沖隊(duì)列
- 線程池2的三個(gè)任務(wù)，task4和task5會(huì)先獲得執(zhí)行線程屈留，然后task6因?yàn)闆](méi)有可分配線程進(jìn)入緩沖隊(duì)列
- 任務(wù)task3會(huì)在task1或task2完成之后局冰，開(kāi)始執(zhí)行
- 任務(wù)task6會(huì)在task4或task5完成之后，開(kāi)始執(zhí)行
分析好之后灌危，執(zhí)行下單元測(cè)試康二，看看是否是這樣的：
2021-09-15 23:45:11.369 INFO 61670 --- [ executor-1-1] com.didispace.chapter77.AsyncTasks : 開(kāi)始任務(wù)：1
2021-09-15 23:45:11.369 INFO 61670 --- [ executor-2-2] com.didispace.chapter77.AsyncTasks : 開(kāi)始任務(wù)：5
2021-09-15 23:45:11.369 INFO 61670 --- [ executor-2-1] com.didispace.chapter77.AsyncTasks : 開(kāi)始任務(wù)：4
2021-09-15 23:45:11.369 INFO 61670 --- [ executor-1-2] com.didispace.chapter77.AsyncTasks : 開(kāi)始任務(wù)：2
2021-09-15 23:45:15.905 INFO 61670 --- [ executor-2-1] com.didispace.chapter77.AsyncTasks : 完成任務(wù)：4，耗時(shí)：4532 毫秒
2021-09-15 23:45:15.905 INFO 61670 --- [ executor-2-1] com.didispace.chapter77.AsyncTasks : 開(kāi)始任務(wù)：6
2021-09-15 23:45:18.263 INFO 61670 --- [ executor-1-2] com.didispace.chapter77.AsyncTasks : 完成任務(wù)：2勇蝙，耗時(shí)：6890 毫秒
2021-09-15 23:45:18.263 INFO 61670 --- [ executor-1-2] com.didispace.chapter77.AsyncTasks : 開(kāi)始任務(wù)：3
2021-09-15 23:45:18.896 INFO 61670 --- [ executor-2-2] com.didispace.chapter77.AsyncTasks : 完成任務(wù)：5沫勿，耗時(shí)：7523 毫秒
2021-09-15 23:45:19.842 INFO 61670 --- [ executor-1-2] com.didispace.chapter77.AsyncTasks : 完成任務(wù)：3，耗時(shí)：1579 毫秒
2021-09-15 23:45:20.551 INFO 61670 --- [ executor-1-1] com.didispace.chapter77.AsyncTasks : 完成任務(wù)：1味混，耗時(shí)：9178 毫秒
2021-09-15 23:45:24.117 INFO 61670 --- [ executor-2-1] com.didispace.chapter77.AsyncTasks : 完成任務(wù)：6产雹，耗時(shí)：8212 毫秒
2021-09-15 23:45:24.117 INFO 61670 --- [ main] c.d.chapter77.Chapter77ApplicationTests : 任務(wù)全部完成，總耗時(shí)：12762毫秒
好了翁锡，今天的學(xué)習(xí)就到這里蔓挖！如果您學(xué)習(xí)過(guò)程中如遇困難？可以加入我們超高質(zhì)量的Spring技術(shù)交流群馆衔，參與交流與討論瘟判，更好的學(xué)習(xí)與進(jìn)步怨绣！更多Spring Boot教程可以點(diǎn)擊直達(dá)！拷获，歡迎收藏與轉(zhuǎn)發(fā)支持梨熙！
代碼示例
本文的完整工程可以查看下面?zhèn)}庫(kù)中2.x目錄下的chapter7-7工程：
- Github：https://github.com/dyc87112/SpringBoot-Learning/
- Gitee：https://gitee.com/didispace/SpringBoot-Learning/
如果您覺(jué)得本文不錯(cuò)，歡迎Star支持刀诬，您的關(guān)注是我堅(jiān)持的動(dòng)力咽扇！
歡迎關(guān)注我的公眾號(hào)：程序猿DD，分享外面看不到的干貨陕壹！
