快捷導(dǎo)航

springboot2.0以上調(diào)度器配置線程池的實(shí)現(xiàn)

更新時間：2019年12月26日 08:29:26 作者：爾過

這篇文章主要介紹了springboot2.0以上調(diào)度器配置線程池的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

一我們使用@EnableScheduling 開啟spring task 調(diào)度器的時候，發(fā)現(xiàn)此調(diào)度器默認(rèn)配置為單線程的。

二打開注解發(fā)現(xiàn)其配置信息在此SchedulingConfiguration類中。發(fā)現(xiàn)其創(chuàng)建了ScheduledTaskRegistrar類
研讀代碼不難發(fā)現(xiàn)調(diào)度器默認(rèn)配置是如下代碼，線程池為單線程的。

protected void scheduleTasks() {
 if (this.taskScheduler == null) {
  this.localExecutor = Executors.newSingleThreadScheduledExecutor();
  this.taskScheduler = new ConcurrentTaskScheduler(this.localExecutor);
 }
 if (this.triggerTasks != null) {
  for (TriggerTask task : this.triggerTasks) {
  addScheduledTask(scheduleTriggerTask(task));
  }
 }
 if (this.cronTasks != null) {
  for (CronTask task : this.cronTasks) {
  addScheduledTask(scheduleCronTask(task));
  }
 }
 if (this.fixedRateTasks != null) {
  for (IntervalTask task : this.fixedRateTasks) {
  addScheduledTask(scheduleFixedRateTask(task));
  }
 }
 if (this.fixedDelayTasks != null) {
  for (IntervalTask task : this.fixedDelayTasks) {
  addScheduledTask(scheduleFixedDelayTask(task));
  }
 }
 }

如何改變此配置呢？

如果想改變其中配置則只需要如下核心代碼

package com.ccbobe.common.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;

import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledThreadPoolExecutor;

@EnableScheduling
@Configuration
public class SchedulerConfig implements SchedulingConfigurer {
  @Bean
  public ScheduledExecutorService concurrentTaskScheduler(){
    ScheduledThreadPoolExecutor executorService = new ScheduledThreadPoolExecutor(20);
    executorService.setMaximumPoolSize(20);
    executorService.setRejectedExecutionHandler(new ScheduledThreadPoolExecutor.CallerRunsPolicy());
    return executorService;
  }
  @Override
  public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
    taskRegistrar.setScheduler(concurrentTaskScheduler());
  }
}

其中Scheduler 支持兩種，種分別是：TaskScheduler 和 ScheduledExecutorService

/**
 * Set the {@link TaskScheduler} to register scheduled tasks with, or a
 * {@link java.util.concurrent.ScheduledExecutorService} to be wrapped as a
 * {@code TaskScheduler}.
 */
 public void setScheduler(@Nullable Object scheduler) {
 if (scheduler == null) {
  this.taskScheduler = null;
 }
 else if (scheduler instanceof TaskScheduler) {
  this.taskScheduler = (TaskScheduler) scheduler;
 }
 else if (scheduler instanceof ScheduledExecutorService) {
  this.taskScheduler = new ConcurrentTaskScheduler(((ScheduledExecutorService) scheduler));
 }
 else {
  throw new IllegalArgumentException("Unsupported scheduler type: " + scheduler.getClass());
 }
 }

完成以上配置，即可讓spring task 運(yùn)行在多線程環(huán)境中。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: