Spring Boot利用@Async怎樣實現異步調用：自定義線程池

小編給大家分享一下Spring Boot利用@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 Boot利用@Async怎樣實現異步調用：自定義線程池”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！