spring boot如何使用自定義的線程池執(zhí)行Async任務(wù)

這篇文章將為大家詳細(xì)講解有關(guān)spring boot如何使用自定義的線程池執(zhí)行Async任務(wù)，小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

一、增加配置屬性類

package com.chhliu.springboot.async.configuration; 
import org.springframework.boot.context.properties.ConfigurationProperties; 
@ConfigurationProperties(prefix = "spring.task.pool") // 該注解的locations已經(jīng)被啟用，現(xiàn)在只要是在環(huán)境中，都會(huì)優(yōu)先加載 
public class TaskThreadPoolConfig { 
 private int corePoolSize; 
 private int maxPoolSize; 
 private int keepAliveSeconds; 
 private int queueCapacity; 
 …………省略getter,setter方法………… 
}

二、創(chuàng)建線程池

package com.chhliu.springboot.async.pool; 
import java.util.concurrent.Executor; 
import java.util.concurrent.ThreadPoolExecutor; 
import org.springframework.beans.factory.annotation.Autowired; 
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 com.chhliu.springboot.async.configuration.TaskThreadPoolConfig; 
@Configuration 
@EnableAsync 
public class TaskExecutePool { 
 @Autowired 
 private TaskThreadPoolConfig config; 
 @Bean 
 public Executor myTaskAsyncPool() { 
 ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); 
 executor.setCorePoolSize(config.getCorePoolSize()); 
 executor.setMaxPoolSize(config.getMaxPoolSize()); 
 executor.setQueueCapacity(config.getQueueCapacity()); 
 executor.setKeepAliveSeconds(config.getKeepAliveSeconds()); 
 executor.setThreadNamePrefix("MyExecutor-"); 
 // rejection-policy：當(dāng)pool已經(jīng)達(dá)到max size的時(shí)候，如何處理新任務(wù) 
 // CALLER_RUNS：不在新線程中執(zhí)行任務(wù)，而是由調(diào)用者所在的線程來執(zhí)行 
 executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); 
 executor.initialize(); 
 return executor; 
 } 
}

三、在主類中開啟配置支持

package com.chhliu.springboot.async; 
import org.springframework.boot.SpringApplication; 
import org.springframework.boot.autoconfigure.SpringBootApplication; 
import org.springframework.boot.context.properties.EnableConfigurationProperties; 
import org.springframework.scheduling.annotation.EnableAsync; 
import com.chhliu.springboot.async.configuration.TaskThreadPoolConfig; 
@SpringBootApplication 
@EnableAsync 
@EnableConfigurationProperties({TaskThreadPoolConfig.class} ) // 開啟配置屬性支持 
public class SpringbootAsyncApplication { 
 public static void main(String[] args) { 
 SpringApplication.run(SpringbootAsyncApplication.class, args); 
 } 
}

四、測(cè)試類

package com.chhliu.springboot.async.pool; 
import org.slf4j.Logger; 
import org.slf4j.LoggerFactory; 
import org.springframework.scheduling.annotation.Async; 
import org.springframework.stereotype.Component; 
@Component 
public class AsyncTask { 
 protected final Logger logger = LoggerFactory.getLogger(this.getClass()); 
 @Async("myTaskAsyncPool") //myTaskAsynPool即配置線程池的方法名，此處如果不寫自定義線程池的方法名，會(huì)使用默認(rèn)的線程池 
 public void doTask1(int i) throws InterruptedException{ 
 logger.info("Task"+i+" started."); 
 } 
}

五、測(cè)試

package com.chhliu.springboot.async; 
import java.util.concurrent.ExecutionException; 
import org.junit.Test; 
import org.junit.runner.RunWith; 
import org.slf4j.Logger; 
import org.slf4j.LoggerFactory; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.boot.test.context.SpringBootTest; 
import org.springframework.test.context.junit4.SpringRunner; 
import com.chhliu.springboot.async.pool.AsyncTask; 
@RunWith(SpringRunner.class) 
@SpringBootTest 
public class SpringbootAsyncApplicationTests { 
 protected final Logger logger = LoggerFactory.getLogger(this.getClass()); 
 @Autowired 
 private AsyncTask asyncTask; 
 @Test 
 public void AsyncTaskTest() throws InterruptedException, ExecutionException { 
 for (int i = 0; i < 100; i++) { 
  asyncTask.doTask1(i); 
 } 
 logger.info("All tasks finished."); 
 } 
}

測(cè)試結(jié)果如下：

2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-10] c.c.springboot.async.pool.AsyncTask      : Task60 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-25] c.c.springboot.async.pool.AsyncTask      : Task61 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [   MyExecutor-6] c.c.springboot.async.pool.AsyncTask      : Task62 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-23] c.c.springboot.async.pool.AsyncTask      : Task63 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-20] c.c.springboot.async.pool.AsyncTask      : Task64 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-19] c.c.springboot.async.pool.AsyncTask      : Task65 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-16] c.c.springboot.async.pool.AsyncTask      : Task66 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-15] c.c.springboot.async.pool.AsyncTask      : Task67 started. 
2017-03-20 20:15:15.208  INFO 4068 --- [  MyExecutor-12] c.c.springboot.async.pool.AsyncTask      : Task68 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [   MyExecutor-1] c.c.springboot.async.pool.AsyncTask      : Task69 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [  MyExecutor-11] c.c.springboot.async.pool.AsyncTask      : Task81 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [   MyExecutor-8] c.c.springboot.async.pool.AsyncTask      : Task82 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [   MyExecutor-7] c.c.springboot.async.pool.AsyncTask      : Task83 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [   MyExecutor-4] c.c.springboot.async.pool.AsyncTask      : Task84 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [  MyExecutor-29] c.c.springboot.async.pool.AsyncTask      : Task85 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [  MyExecutor-21] c.c.springboot.async.pool.AsyncTask      : Task86 started. 
2017-03-20 20:15:15.209  INFO 4068 --- [  MyExecutor-17] c.c.springboot.async.pool.AsyncTask      : Task88 started. 

測(cè)試結(jié)果ok！

六、配置默認(rèn)的線程池

如果我們想使用默認(rèn)的線程池，但是只是想修改默認(rèn)線程池的配置，那怎么做了，此時(shí)我們需要實(shí)現(xiàn)AsyncConfigurer類，示例代碼如下：

import java.lang.reflect.Method; 
import java.util.concurrent.Executor; 
import java.util.concurrent.ThreadPoolExecutor; 
import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.context.annotation.Configuration; 
import org.springframework.scheduling.annotation.AsyncConfigurer; 
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; 
import com.chhliu.cq.emailservice.threadconfiguration.TaskThreadPoolConfig; 
import lombok.extern.slf4j.Slf4j; 
/** 
 * 注意：該線程池被所有的異步任務(wù)共享，而不屬于某一個(gè)異步任務(wù) 
 * 描述：配置異步任務(wù)的線程池 
 * @author chhliu 
 * 創(chuàng)建時(shí)間：2017年5月22日 上午10:20:56 
 * @version 1.2.0 
 */ 
@Slf4j 
@Configuration 
public class AsyncTaskExecutePool implements AsyncConfigurer{ 
 @Autowired 
 private TaskThreadPoolConfig config; // 配置屬性類，見上面的代碼 
 @Override 
 public Executor getAsyncExecutor() { 
 ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); 
 executor.setCorePoolSize(config.getCorePoolSize()); 
 executor.setMaxPoolSize(config.getMaxPoolSize()); 
 executor.setQueueCapacity(config.getQueueCapacity()); 
 executor.setKeepAliveSeconds(config.getKeepAliveSeconds()); 
 executor.setThreadNamePrefix("taskExecutor-"); 
 // rejection-policy：當(dāng)pool已經(jīng)達(dá)到max size的時(shí)候，如何處理新任務(wù) 
 // CALLER_RUNS：不在新線程中執(zhí)行任務(wù)，而是由調(diào)用者所在的線程來執(zhí)行 
 executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); 
 executor.initialize(); 
 return executor; 
 } 
 @Override 
 public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {// 異步任務(wù)中異常處理 
 return new AsyncUncaughtExceptionHandler() { 
  @Override 
  public void handleUncaughtException(Throwable arg0, Method arg1, Object... arg2) { 
  log.error("=========================="+arg0.getMessage()+"=======================", arg0); 
  log.error("exception method:"+arg1.getName()); 
  } 
 }; 
 } 
}

使用的時(shí)候，只需在方法上加上@Async即可。

關(guān)于“spring boot如何使用自定義的線程池執(zhí)行Async任務(wù)”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。