一、Spring Boot任務(wù)線程池

線程池的作用

• 防止資源占用無(wú)限的擴(kuò)張
• 調(diào)用過(guò)程省去資源的創(chuàng)建和銷(xiāo)毀所占用的時(shí)間

在高并發(fā)環(huán)境下，不斷的分配新資源，可能導(dǎo)致系統(tǒng)資源耗盡。所以為了避免這個(gè)問(wèn)題，我們?yōu)楫惒饺蝿?wù)規(guī)劃一個(gè)線程池。當(dāng)然，如果沒(méi)有配置線程池的話，springboot會(huì)自動(dòng)配置一個(gè)ThreadPoolTaskExecutor 線程池到bean當(dāng)中。

# 核心線程數(shù)
spring.task.execution.pool.core-size=8  
# 最大線程數(shù)
spring.task.execution.pool.max-size=16
# 空閑線程存活時(shí)間
spring.task.execution.pool.keep-alive=60s
# 是否允許核心線程超時(shí)
spring.task.execution.pool.allow-core-thread-timeout=true
# 線程隊(duì)列數(shù)量
spring.task.execution.pool.queue-capacity=100
# 線程關(guān)閉等待
spring.task.execution.shutdown.await-termination=false
spring.task.execution.shutdown.await-termination-period=
# 線程名稱前綴
spring.task.execution.thread-name-prefix=task-

在springboot配置文件中加入上面的配置，即可實(shí)現(xiàn)ThreadPoolTaskExecutor 線程池。

二、自定義線程池

有的時(shí)候，我們希望將系統(tǒng)內(nèi)的一類(lèi)任務(wù)放到一個(gè)線程池，另一類(lèi)任務(wù)放到另外一個(gè)線程池，所以使用Spring Boot自帶的任務(wù)線程池就捉襟見(jiàn)肘了。下面介紹自定義線程池的方法。

創(chuàng)建一個(gè) 線程池配置類(lèi) TaskConfiguration ，并配置一個(gè) 任務(wù)線程池對(duì)象 taskExecutor。

@Configuration
public class TaskConfiguration {
    @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 CallerRunsPolicy());
        return executor;
    }
}

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

Reject策略預(yù)定義有四種：

• AbortPolicy，用于被拒絕任務(wù)的處理程序，它將拋出RejectedExecutionException。
• CallerRunsPolicy，用于被拒絕任務(wù)的處理程序，它直接在execute方法的調(diào)用線程中運(yùn)行被拒絕的任務(wù)。
• DiscardOldestPolicy，用于被拒絕任務(wù)的處理程序，它放棄最舊的未處理請(qǐng)求，然后重試execute。
• DiscardPolicy，用于被拒絕任務(wù)的處理程序，默認(rèn)情況下它將丟棄被拒絕的任務(wù)。

創(chuàng)建 AsyncExecutorTask類(lèi)，三個(gè)任務(wù)的配置和 AsyncTask 一樣，不同的是 @Async 注解需要指定前面配置的 線程池的名稱 taskExecutor。

@Component
public class AsyncExecutorTask extends AbstractTask {
? ? @Async("taskExecutor")
? ? public Future<String> doTaskOneCallback() throws Exception {
? ? ? ? super.doTaskOne();
? ? ? ? System.out.println("任務(wù)一，當(dāng)前線程：" + Thread.currentThread().getName());
? ? ? ? return new AsyncResult<>("任務(wù)一完成");
? ? }

? ? @Async("taskExecutor")
? ? public Future<String> doTaskTwoCallback() throws Exception {
? ? ? ? super.doTaskTwo();
? ? ? ? System.out.println("任務(wù)二，當(dāng)前線程：" + Thread.currentThread().getName());
? ? ? ? return new AsyncResult<>("任務(wù)二完成");
? ? }

? ? @Async("taskExecutor")
? ? public Future<String> doTaskThreeCallback() throws Exception {
? ? ? ? super.doTaskThree();
? ? ? ? System.out.println("任務(wù)三，當(dāng)前線程：" + Thread.currentThread().getName());
? ? ? ? return new AsyncResult<>("任務(wù)三完成");
? ? }
}

在 單元測(cè)試 用例中，注入 AsyncExecutorTask 對(duì)象，并在測(cè)試用例中執(zhí)行 doTaskOne()，doTaskTwo()，doTaskThree() 三個(gè)方法。

@SpringBootTest
public class AsyncExecutorTaskTest {
? ? @Autowired
? ? private AsyncExecutorTask task;

? ? @Test
? ? public void testAsyncExecutorTask() throws Exception {
? ? ? ? task.doTaskOneCallback();
? ? ? ? task.doTaskTwoCallback();
? ? ? ? task.doTaskThreeCallback();

? ? ? ? sleep(30 * 1000L);
? ? }
}

執(zhí)行一下上述的 單元測(cè)試，可以看到如下結(jié)果：

開(kāi)始做任務(wù)一
開(kāi)始做任務(wù)三
開(kāi)始做任務(wù)二
完成任務(wù)二，耗時(shí)：3905毫秒
任務(wù)二，當(dāng)前線程：taskExecutor-2
完成任務(wù)一，耗時(shí)：6184毫秒
任務(wù)一，當(dāng)前線程：taskExecutor-1
完成任務(wù)三，耗時(shí)：9737毫秒
任務(wù)三，當(dāng)前線程：taskExecutor-3

執(zhí)行上面的單元測(cè)試，觀察到 任務(wù)線程池 的 線程池名的前綴 被打印，說(shuō)明 線程池 成功執(zhí)行 異步任務(wù)！

三、優(yōu)雅地關(guān)閉線程池

由于在應(yīng)用關(guān)閉的時(shí)候異步任務(wù)還在執(zhí)行，導(dǎo)致類(lèi)似 數(shù)據(jù)庫(kù)連接池 這樣的對(duì)象一并被 銷(xiāo)毀了，當(dāng) 異步任務(wù) 中對(duì) 數(shù)據(jù)庫(kù) 進(jìn)行操作就會(huì)出錯(cuò)。

解決方案如下，重新設(shè)置線程池配置對(duì)象，新增線程池 setWaitForTasksToCompleteOnShutdown() 和 setAwaitTerminationSeconds() 配置：

@Bean("taskExecutor")
public Executor taskExecutor() {
    ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
    executor.setPoolSize(20);
    executor.setThreadNamePrefix("taskExecutor-");
    executor.setWaitForTasksToCompleteOnShutdown(true);
    executor.setAwaitTerminationSeconds(60);
    return executor;
}

• setWaitForTasksToCompleteOnShutdown(true): 該方法用來(lái)設(shè)置 線程池關(guān)閉 的時(shí)候 等待 所有任務(wù)都完成后，再繼續(xù) 銷(xiāo)毀 其他的 Bean，這樣這些 異步任務(wù) 的 銷(xiāo)毀 就會(huì)先于 數(shù)據(jù)庫(kù)連接池對(duì)象 的銷(xiāo)毀。
• setAwaitTerminationSeconds(60): 該方法用來(lái)設(shè)置線程池中 任務(wù)的等待時(shí)間，如果超過(guò)這個(gè)時(shí)間還沒(méi)有銷(xiāo)毀就 強(qiáng)制銷(xiāo)毀，以確保應(yīng)用最后能夠被關(guān)閉，而不是阻塞住。

異步任務(wù)** 的 銷(xiāo)毀 就會(huì)先于 數(shù)據(jù)庫(kù)連接池對(duì)象 的銷(xiāo)毀。

• setAwaitTerminationSeconds(60): 該方法用來(lái)設(shè)置線程池中 任務(wù)的等待時(shí)間，如果超過(guò)這個(gè)時(shí)間還沒(méi)有銷(xiāo)毀就 強(qiáng)制銷(xiāo)毀，以確保應(yīng)用最后能夠被關(guān)閉，而不是阻塞住。

 到此這篇關(guān)于springboot為異步任務(wù)規(guī)劃自定義線程池的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)springboot異步自定義線程池內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論