springboot如何初始化通用線程池

這篇文章主要講解了“springboot如何初始化通用線程池”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“springboot如何初始化通用線程池”吧！

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 java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

/**
* @description : 通用線程池,用于異步執行寫操作不影響主線程
*/
@Configuration
@EnableAsync
public class InitThread {

    //線程池維護線程的最少數量
    private static final int CORE_POOL_SIZE = 10;

    //線程池維護線程的最大數量
    private static final int MAX_POOL_SIZE = 50;

    //緩存隊列
    private static final int QUEUE_CAPACITY = 10;

    //允許的空閑時間
    private static final int KEEP_ALIVE = 60;

    @Bean
    public Executor myExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(CORE_POOL_SIZE);
        executor.setMaxPoolSize(MAX_POOL_SIZE);
        executor.setQueueCapacity(QUEUE_CAPACITY);
        executor.setThreadNamePrefix("executor-");
        /*
         *  使用此策略，如果添加到線程池失敗，那么主線程會自己去執行該任務，不會等待線程池中的線程去執行
         */
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.setKeepAliveSeconds(KEEP_ALIVE);
        executor.initialize();
        return executor;
    }

}

拒絕策略RejectedExecutionHandler

AbortPolicy：該策略是線程池的默認策略。使用該策略時，如果線程池隊列滿了丟掉這個任務并且拋出RejectedExecutionException異常
public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {  
            //不做任何處理，直接拋出異常  
            throw new RejectedExecutionException("Task" + r.toString() +  
                                                 " rejected from " +  
                                                 e.toString());  
}

DiscardPolicy ：這個策略和AbortPolicy的slient版本，如果線程池隊列滿了，會直接丟掉這個任務并且不會有任何異常。 
public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {  
         //就是一個空的方法  
      }  

DiscardOldestPolicy ：這個策略從字面上也很好理解，丟棄最老的。也就是說如果隊列滿了，會將最早進入隊列的任務刪掉騰出空間，再嘗試加入隊列。 
因為隊列是隊尾進，隊頭出，所以隊頭元素是最老的，因此每次都是移除對頭元素后再嘗試入隊。
 public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {  
     if (!e.isShutdown()) {  
         //移除隊頭元素  
         e.getQueue().poll();  
         //再嘗試入隊  
         e.execute(r);  
     }  
 }

CallerRunsPolicy ：使用此策略，如果添加到線程池失敗，那么主線程會自己去執行該任務，不會等待線程池中的線程去執行。就像是個急脾氣的人，我等不到別人來做這件事就干脆自己干。
 public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {  
             if (!e.isShutdown()) {  
                 //直接執行run方法  
                 r.run();  
             }  
         }


自定義：只要實現RejectedExecutionHandler接口，并且實現rejectedExecution方法就可以了。具體的邏輯就在rejectedExecution方法里去定義就OK了
 public class MyRejectPolicy implements RejectedExecutionHandler{  
     public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {  
         //Sender是我的Runnable類，里面有message字段  
         if (r instanceof Sender) {  
             Sender sender = (Sender) r;  
             //直接打印  
             System.out.println(sender.getMessage());  
         }  
     }  
 }

感謝各位的閱讀，以上就是“springboot如何初始化通用線程池”的內容了，經過本文的學習后，相信大家對springboot如何初始化通用線程池這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！