系列文章推薦閱讀順序：
對于異步執(zhí)行的控制是我們保障自身應(yīng)用健康的基本技能句旱。本文通過自定義線程池的方式來控制異步調(diào)用的并發(fā)。
一、定義線程池
第一步蛤迎，先在Spring Boot主類中定義一個線程池薯嗤，比如：
package com.erbadagang.springboot.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
@EnableAsync
public class AsyncApplication {
public static void main(String[] args) {
SpringApplication.run(AsyncApplication.class, args);
}
@Configuration
@EnableAsync
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("guoxiuzhiTE-");
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
return executor;
}
}
}
上面我們通過使用ThreadPoolTaskExecutor創(chuàng)建了一個線程池买窟，同時設(shè)置了以下這些參數(shù)：
- 核心線程數(shù)10：線程池創(chuàng)建時候初始化的線程數(shù)
- 最大線程數(shù)20：線程池最大的線程數(shù)登渣，只有在緩沖隊列滿了之后才會申請超過核心線程數(shù)的線程
- 緩沖隊列200：用來緩沖執(zhí)行任務(wù)的隊列
- 允許線程的空閑時間60秒：當超過了核心線程出之外的線程在空閑時間到達之后會被銷毀
- 線程池名的前綴：設(shè)置好了之后可以方便我們定位處理任務(wù)所在的線程池
- 線程池對拒絕任務(wù)的處理策略：這里采用了CallerRunsPolicy策略摊求，當線程池沒有處理能力的時候嘉抓，該策略會直接在 execute 方法的調(diào)用線程中運行被拒絕的任務(wù)索守；如果執(zhí)行程序已關(guān)閉，則會丟棄該任務(wù)
二抑片、使用線程池
在定義了線程池之后卵佛，我們?nèi)绾巫尞惒秸{(diào)用的執(zhí)行任務(wù)使用這個線程池中的資源來運行呢？方法非常簡單，我們只需要在@Async注解中指定線程池名即可截汪，比如：
package com.erbadagang.springboot.async.task;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import java.util.Random;
/**
* @description 通過線程池異步調(diào)用
* @ClassName: AsyncTask
* @author: 郭秀志 jbcode@126.com
* @date: 2020/7/24 11:08
* @Copyright:
*/
@Component
@Slf4j
public class AsyncTaskWithThreadPool {
public static Random random = new Random();
@Async("taskExecutor")
public void doTaskOne() throws Exception {
log.info("開始做任務(wù)一");
long start = System.currentTimeMillis();
Thread.sleep(random.nextInt(5000));
long end = System.currentTimeMillis();
log.info("完成任務(wù)一疾牲，耗時：" + (end - start) + "毫秒");
}
@Async("taskExecutor")
public void doTaskTwo() throws Exception {
log.info("開始做任務(wù)二");
long start = System.currentTimeMillis();
Thread.sleep(random.nextInt(5000));
long end = System.currentTimeMillis();
log.info("完成任務(wù)二，耗時：" + (end - start) + "毫秒");
}
@Async("taskExecutor")
public void doTaskThree() throws Exception {
log.info("開始做任務(wù)三");
long start = System.currentTimeMillis();
Thread.sleep(random.nextInt(5000));
long end = System.currentTimeMillis();
log.info("完成任務(wù)三衙解，耗時：" + (end - start) + "毫秒");
}
}
為了日志輸出線程名阳柔，使用了@Slf4j,需要引入依賴
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.12</version>
</dependency>
我們來寫個單元測試來驗證一下
@Autowired
private AsyncTaskWithThreadPool asyncTaskWithThreadPool;
@Test
public void test() throws Exception {
asyncTaskWithThreadPool.doTaskOne();
asyncTaskWithThreadPool.doTaskTwo();
asyncTaskWithThreadPool.doTaskThree();
Thread.currentThread().join();
}
執(zhí)行上面的單元測試，我們可以在控制臺中看到所有輸出的線程名前都是之前我們定義的線程池前綴名開始的蚓峦，說明我們使用線程池來執(zhí)行異步任務(wù)的試驗成功了舌剂！
2020-07-24 11:29:08.783 INFO 11588 --- [ guoxiuzhiTE-1] c.e.s.a.task.AsyncTaskWithThreadPool : 開始做任務(wù)一
2020-07-24 11:29:08.783 INFO 11588 --- [ guoxiuzhiTE-3] c.e.s.a.task.AsyncTaskWithThreadPool : 開始做任務(wù)三
2020-07-24 11:29:08.783 INFO 11588 --- [ guoxiuzhiTE-2] c.e.s.a.task.AsyncTaskWithThreadPool : 開始做任務(wù)二
底線
本文源代碼使用 Apache License 2.0開源許可協(xié)議，可從如下Gitee地址免費獲取代碼通過git clone命令下載到本地或者直接點擊鏈接通過瀏覽器方式查看源代碼枫匾。
