本文中的例子我們可以在之前的例子基礎(chǔ)上修改，也可以創(chuàng)建一個全新的Spring Boot項(xiàng)目來嘗試。

[[233638]]

定義線程池

***步，先在Spring Boot主類中定義一個線程池，比如：

@SpringBootApplication 
public class Application { 
 
    public static void main(String[] args) { 
        SpringApplication.run(Application.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; 
        } 
    } 
 
} 

上面我們通過使用ThreadPoolTaskExecutor創(chuàng)建了一個線程池，同時(shí)設(shè)置了以下這些參數(shù)：

• 核心線程數(shù)10：線程池創(chuàng)建時(shí)候初始化的線程數(shù)
• ***線程數(shù)20：線程池***的線程數(shù)，只有在緩沖隊(duì)列滿了之后才會申請超過核心線程數(shù)的線程
• 緩沖隊(duì)列200：用來緩沖執(zhí)行任務(wù)的隊(duì)列
• 允許線程的空閑時(shí)間60秒：當(dāng)超過了核心線程出之外的線程在空閑時(shí)間到達(dá)之后會被銷毀
• 線程池名的前綴：設(shè)置好了之后可以方便我們定位處理任務(wù)所在的線程池
• 線程池對拒絕任務(wù)的處理策略：這里采用了CallerRunsPolicy策略，當(dāng)線程池沒有處理能力的時(shí)候，該策略會直接在 execute 方法的調(diào)用線程中運(yùn)行被拒絕的任務(wù)；如果執(zhí)行程序已關(guān)閉，則會丟棄該任務(wù)

使用線程池

在定義了線程池之后，我們?nèi)绾巫尞惒秸{(diào)用的執(zhí)行任務(wù)使用這個線程池中的資源來運(yùn)行呢？方法非常簡單，我們只需要在@Async注解中指定線程池名即可，比如：

@Slf4j 
@Component 
public class Task { 
 
    public static Random random = new Random(); 
 
    @Async("taskExecutor") 
    public void doTaskOne() throws Exception { 
        log.info("開始做任務(wù)一"); 
        long start = System.currentTimeMillis(); 
        Thread.sleep(random.nextInt(10000)); 
        long end = System.currentTimeMillis(); 
        log.info("完成任務(wù)一，耗時(shí)：" + (end - start) + "毫秒"); 
    } 
 
    @Async("taskExecutor") 
    public void doTaskTwo() throws Exception { 
        log.info("開始做任務(wù)二"); 
        long start = System.currentTimeMillis(); 
        Thread.sleep(random.nextInt(10000)); 
        long end = System.currentTimeMillis(); 
        log.info("完成任務(wù)二，耗時(shí)：" + (end - start) + "毫秒"); 
    } 
 
    @Async("taskExecutor") 
    public void doTaskThree() throws Exception { 
        log.info("開始做任務(wù)三"); 
        long start = System.currentTimeMillis(); 
        Thread.sleep(random.nextInt(10000)); 
        long end = System.currentTimeMillis(); 
        log.info("完成任務(wù)三，耗時(shí)：" + (end - start) + "毫秒"); 
    } 
 
} 

單元測試

***，我們來寫個單元測試來驗(yàn)證一下

@RunWith(SpringJUnit4ClassRunner.class) 
@SpringBootTest 
public class ApplicationTests { 
 
    @Autowired 
    private Task task; 
 
    @Test 
    public void test() throws Exception { 
 
        task.doTaskOne(); 
        task.doTaskTwo(); 
        task.doTaskThree(); 
 
        Thread.currentThread().join(); 
    } 
 
} 

執(zhí)行上面的單元測試，我們可以在控制臺中看到所有輸出的線程名前都是之前我們定義的線程池前綴名開始的，說明我們使用線程池來執(zhí)行異步任務(wù)的試驗(yàn)成功了！

2018-03-27 22:01:15.620  INFO 73703 --- [ taskExecutor-1] com.didispace.async.Task                 : 開始做任務(wù)一 
2018-03-27 22:01:15.620  INFO 73703 --- [ taskExecutor-2] com.didispace.async.Task                 : 開始做任務(wù)二 
2018-03-27 22:01:15.620  INFO 73703 --- [ taskExecutor-3] com.didispace.async.Task                 : 開始做任務(wù)三 
2018-03-27 22:01:18.165  INFO 73703 --- [ taskExecutor-2] com.didispace.async.Task                 : 完成任務(wù)二，耗時(shí)：2545毫秒 
2018-03-27 22:01:22.149  INFO 73703 --- [ taskExecutor-3] com.didispace.async.Task                 : 完成任務(wù)三，耗時(shí)：6529毫秒 
2018-03-27 22:01:23.912  INFO 73703 --- [ taskExecutor-1] com.didispace.async.Task                 : 完成任務(wù)一，耗時(shí)：8292毫秒 

完整示例：

讀者可以根據(jù)喜好選擇下面的兩個倉庫中查看Chapter4-1-3項(xiàng)目：

Github：https://github.com/dyc87112/SpringBoot-Learning/

Gitee：https://gitee.com/didispace/SpringBoot-Learning/

【本文為51CTO專欄作者“翟永超”的原創(chuàng)稿件，轉(zhuǎn)載請通過51CTO聯(lián)系作者獲取授權(quán)】

戳這里，看該作者更多好文