推薦 Spring Boot/Cloud 視訊：

在之前的Spring Boot基礎教程系列中，已經通過《Spring Boot中使用@Async實現非同步呼叫》一文介紹過如何使用@Async註解來實現非同步呼叫了。但是，對於這些非同步執行的控制是我們保障自身應用健康的基本技能。本文我們就來學習一下，如果通過自定義執行緒池的方式來控制非同步呼叫的併發。

定義執行緒池

第一步，先在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建立了一個執行緒池，同時設定了以下這些引數：

• 核心執行緒數10：執行緒池建立時候初始化的執行緒數
• 最大執行緒數20：執行緒池最大的執行緒數，只有在緩衝佇列滿了之後才會申請超過核心執行緒數的執行緒
• 緩衝佇列200：用來緩衝執行任務的佇列
• 允許執行緒的空閒時間60秒：當超過了核心執行緒出之外的執行緒在空閒時間到達之後會被銷燬
• 執行緒池名的字首：設定好了之後可以方便我們定位處理任務所在的執行緒池
• 執行緒池對拒絕任務的處理策略：這裡採用了CallerRunsPolicy策略，當執行緒池沒有處理能力的時候，該策略會直接在 execute 方法的呼叫執行緒中執行被拒絕的任務；如果執行程式已關閉，則會丟棄該任務

使用執行緒池

在定義了執行緒池之後，我們如何讓非同步呼叫的執行任務使用這個執行緒池中的資源來執行呢？方法非常簡單，我們只需要在@Async註解中指定執行緒池名即可，比如：

@Slf4j
@Component
public class Task {

    public static Random random = new Random();

    @Async("taskExecutor")
    public void doTaskOne() throws Exception {
        log.info("開始做任務一");
        long start = System.currentTimeMillis();
        Thread.sleep(random.nextInt(10000));
        long end = System.currentTimeMillis();
        log.info("完成任務一，耗時：" + (end - start) + "毫秒");
    }

    @Async("taskExecutor")
    public void doTaskTwo() throws Exception {
        log.info("開始做任務二");
        long start = System.currentTimeMillis();
        Thread.sleep(random.nextInt(10000));
        long end = System.currentTimeMillis();
        log.info("完成任務二，耗時：" + (end - start) + "毫秒");
    }

    @Async("taskExecutor")
    public void doTaskThree() throws Exception {
        log.info("開始做任務三");
        long start = System.currentTimeMillis();
        Thread.sleep(random.nextInt(10000));
        long end = System.currentTimeMillis();
        log.info("完成任務三，耗時：" + (end - start) + "毫秒");
    }

}
 

單元測試

最後，我們來寫個單元測試來驗證一下

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

}

執行上面的單元測試，我們可以在控制檯中看到所有輸出的執行緒名前都是之前我們定義的執行緒池字首名開始的，說明我們使用執行緒池來執行非同步任務的試驗成功了！

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

專家推薦

“隨著微服務架構的發展，Spring Cloud 使用得越來越廣泛。馳狼課堂 Spring Boot 快速入門，Spring Boot 與Spring Cloud 整合，docker+k8s，大型電商商城等多套免費實戰教程可以幫您真正做到快速上手，將技術點切實運用到微服務專案中。”