SpringBoot中怎么動態控制定時任務

SpringBoot中怎么動態控制定時任務，相信很多沒有經驗的人對此束手無策，為此本文總結了問題出現的原因和解決方法，通過這篇文章希望你能解決這個問題。

1.定時任務的配置類：SchedulingConfig

import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.scheduling.TaskScheduler;import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;/** * @program: simple-demo * @description: 定時任務配置類 * @author: CaoTing * @date: 2019/5/23 **/@Configurationpublic class SchedulingConfig {  @Bean  public TaskScheduler taskScheduler() {    ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();    // 定時任務執行線程池核心線程數    taskScheduler.setPoolSize(4);    taskScheduler.setRemoveOnCancelPolicy(true);    taskScheduler.setThreadNamePrefix("TaskSchedulerThreadPool-");    return taskScheduler;  }}

2.定時任務注冊類：CronTaskRegistrar

這個類包含了新增定時任務，移除定時任務等等核心功能方法

import com.caotinging.demo.task.ScheduledTask;import org.springframework.beans.factory.DisposableBean;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.scheduling.TaskScheduler;import org.springframework.scheduling.config.CronTask;import org.springframework.stereotype.Component;import java.util.Map;import java.util.concurrent.ConcurrentHashMap;/** * @program: simple-demo * @description: 添加定時任務注冊類，用來增加、刪除定時任務。 * @author: CaoTing * @date: 2019/5/23 **/@Componentpublic class CronTaskRegistrar implements DisposableBean {  private final Map<Runnable, ScheduledTask> scheduledTasks = new ConcurrentHashMap<>(16);  @Autowired  private TaskScheduler taskScheduler;  public TaskScheduler getScheduler() {    return this.taskScheduler;  }  /**   * 新增定時任務   * @param task   * @param cronExpression   */  public void addCronTask(Runnable task, String cronExpression) {    addCronTask(new CronTask(task, cronExpression));  }  public void addCronTask(CronTask cronTask) {    if (cronTask != null) {      Runnable task = cronTask.getRunnable();      if (this.scheduledTasks.containsKey(task)) {        removeCronTask(task);      }      this.scheduledTasks.put(task, scheduleCronTask(cronTask));    }  }  /**   * 移除定時任務   * @param task   */  public void removeCronTask(Runnable task) {    ScheduledTask scheduledTask = this.scheduledTasks.remove(task);    if (scheduledTask != null)      scheduledTask.cancel();  }  public ScheduledTask scheduleCronTask(CronTask cronTask) {    ScheduledTask scheduledTask = new ScheduledTask();    scheduledTask.future = this.taskScheduler.schedule(cronTask.getRunnable(), cronTask.getTrigger());    return scheduledTask;  }  @Override  public void destroy() {    for (ScheduledTask task : this.scheduledTasks.values()) {      task.cancel();    }    this.scheduledTasks.clear();  }}

3.定時任務執行類：SchedulingRunnable

import com.caotinging.demo.utils.SpringContextUtils;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.util.ReflectionUtils;import java.lang.reflect.Method;import java.util.Objects;/** * @program: simple-demo * @description: 定時任務運行類 * @author: CaoTing * @date: 2019/5/23 **/public class SchedulingRunnable implements Runnable {  private static final Logger logger = LoggerFactory.getLogger(SchedulingRunnable.class);  private String beanName;  private String methodName;  private Object[] params;  public SchedulingRunnable(String beanName, String methodName) {    this(beanName, methodName, null);  }  public SchedulingRunnable(String beanName, String methodName, Object...params ) {    this.beanName = beanName;    this.methodName = methodName;    this.params = params;  }  @Override  public void run() {    logger.info("定時任務開始執行 - bean：{}，方法：{}，參數：{}", beanName, methodName, params);    long startTime = System.currentTimeMillis();    try {      Object target = SpringContextUtils.getBean(beanName);      Method method = null;      if (null != params && params.length > 0) {        Class<?>[] paramCls = new Class[params.length];        for (int i = 0; i < params.length; i++) {          paramCls[i] = params[i].getClass();        }        method = target.getClass().getDeclaredMethod(methodName, paramCls);      } else {        method = target.getClass().getDeclaredMethod(methodName);      }      ReflectionUtils.makeAccessible(method);      if (null != params && params.length > 0) {        method.invoke(target, params);      } else {        method.invoke(target);      }    } catch (Exception ex) {      logger.error(String.format("定時任務執行異常 - bean：%s，方法：%s，參數：%s ", beanName, methodName, params), ex);    }    long times = System.currentTimeMillis() - startTime;    logger.info("定時任務執行結束 - bean：{}，方法：{}，參數：{}，耗時：{} 毫秒", beanName, methodName, params, times);  }  @Override  public boolean equals(Object o) {    if (this == o) return true;    if (o == null || getClass() != o.getClass()) return false;    SchedulingRunnable that = (SchedulingRunnable) o;    if (params == null) {      return beanName.equals(that.beanName) &&          methodName.equals(that.methodName) &&          that.params == null;    }    return beanName.equals(that.beanName) &&        methodName.equals(that.methodName) &&        params.equals(that.params);  }  @Override  public int hashCode() {    if (params == null) {      return Objects.hash(beanName, methodName);    }    return Objects.hash(beanName, methodName, params);  }}

4.定時任務控制類：ScheduledTask

import java.util.concurrent.ScheduledFuture;/** * @program: simple-demo * @description: 定時任務控制類 * @author: CaoTing * @date: 2019/5/23 **/public final class ScheduledTask {  public volatile ScheduledFuture<?> future;  /**   * 取消定時任務   */  public void cancel() {    ScheduledFuture<?> future = this.future;    if (future != null) {      future.cancel(true);    }  }}

5.定時任務的測試

編寫一個需要用于測試的任務類

import org.springframework.stereotype.Component;/** * @program: simple-demo * @description: * @author: CaoTing * @date: 2019/5/23 **/@Component("demoTask")public class DemoTask {  public void taskWithParams(String param1, Integer param2) {    System.out.println("這是有參示例任務：" + param1 + param2);  }  public void taskNoParams() {    System.out.println("這是無參示例任務");  }}

進行單元測試

import com.caotinging.demo.application.DynamicTaskApplication;import com.caotinging.demo.application.SchedulingRunnable;import com.caotinging.demo.config.CronTaskRegistrar;import org.junit.Test;import org.junit.runner.RunWith;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.boot.test.context.SpringBootTest;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;/** * @program: simple-demo * @description: 測試定時任務 * @author: CaoTing * @date: 2019/5/23 **/@RunWith(SpringJUnit4ClassRunner.class)@SpringBootTest(classes = DynamicTaskApplication.class)public class TaskTest {  @Autowired  CronTaskRegistrar cronTaskRegistrar;  @Test  public void testTask() throws InterruptedException {    SchedulingRunnable task = new SchedulingRunnable("demoTask", "taskNoParams", null);    cronTaskRegistrar.addCronTask(task, "0/10 * * * * ?");    // 便于觀察    Thread.sleep(3000000);  }  @Test  public void testHaveParamsTask() throws InterruptedException {    SchedulingRunnable task = new SchedulingRunnable("demoTask", "taskWithParams", "haha", 23);    cronTaskRegistrar.addCronTask(task, "0/10 * * * * ?");    // 便于觀察    Thread.sleep(3000000);  }}

6.工具類：SpringContextUtils

import org.springframework.beans.BeansException;import org.springframework.context.ApplicationContext;import org.springframework.context.ApplicationContextAware;import org.springframework.stereotype.Component;/** * @program: simple-demo * @description: spring獲取bean工具類 * @author: CaoTing * @date: 2019/5/23 **/@Componentpublic class SpringContextUtils implements ApplicationContextAware {  private static ApplicationContext applicationContext = null;  @Override  public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {    if (SpringContextUtils.applicationContext == null) {      SpringContextUtils.applicationContext = applicationContext;    }  }  //獲取applicationContext  public static ApplicationContext getApplicationContext() {    return applicationContext;  }  //通過name獲取 Bean.  public static Object getBean(String name) {    return getApplicationContext().getBean(name);  }  //通過class獲取Bean.  public static <T> T getBean(Class<T> clazz) {    return getApplicationContext().getBean(clazz);  }  //通過name,以及Clazz返回指定的Bean  public static <T> T getBean(String name, Class<T> clazz) {    return getApplicationContext().getBean(name, clazz);  }}

7.我的pom依賴

<dependencies>    <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-jdbc</artifactId>    </dependency>    <dependency>      <groupId>com.baomidou</groupId>      <artifactId>mybatisplus-spring-boot-starter</artifactId>      <version>1.0.5</version>    </dependency>    <dependency>      <groupId>com.baomidou</groupId>      <artifactId>mybatis-plus</artifactId>      <version>2.1.9</version>    </dependency>    <dependency>      <groupId>mysql</groupId>      <artifactId>mysql-connector-java</artifactId>      <scope>runtime</scope>    </dependency>    <dependency>      <groupId>com.alibaba</groupId>      <artifactId>druid-spring-boot-starter</artifactId>      <version>1.1.9</version>    </dependency>    <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-aop</artifactId>    </dependency>    <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-web</artifactId>    </dependency>    <!-- 數據庫-->    <!--<dependency>      <groupId>mysql</groupId>      <artifactId>mysql-connector-java</artifactId>      <scope>runtime</scope>    </dependency>-->    <!-- https://mvnrepository.com/artifact/com.hynnet/oracle-driver-ojdbc -->    <!--<dependency>      <groupId>com.oracle</groupId>      <artifactId>ojdbc6</artifactId>      <version>11.2.0.1.0</version>    </dependency>-->    <!-- 單元測試 -->    <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-test</artifactId>      <scope>provided</scope>    </dependency>    <!--redisTemplate -->    <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-data-redis</artifactId>    </dependency>    <dependency>      <groupId>redis.clients</groupId>      <artifactId>jedis</artifactId>      <version>2.7.3</version>    </dependency>    <!-- http連接 restTemplate -->    <dependency>      <groupId>org.apache.httpcomponents</groupId>      <artifactId>httpclient</artifactId>    </dependency>    <dependency>      <groupId>org.apache.httpcomponents</groupId>      <artifactId>httpclient-cache</artifactId>    </dependency>    <!-- 工具-->    <dependency>      <groupId>org.projectlombok</groupId>      <artifactId>lombok</artifactId>      <optional>true</optional>    </dependency>    <dependency>      <groupId>com.alibaba</groupId>      <artifactId>fastjson</artifactId>      <version>1.2.31</version>    </dependency>    <dependency>      <groupId>org.apache.commons</groupId>      <artifactId>commons-lang3</artifactId>    </dependency>    <dependency>      <groupId>commons-lang</groupId>      <artifactId>commons-lang</artifactId>      <version>2.6</version>    </dependency>    <!-- https://mvnrepository.com/artifact/com.google/guava -->    <dependency>      <groupId>com.google.guava</groupId>      <artifactId>guava</artifactId>      <version>10.0.1</version>    </dependency>    <!-- pinyin4j -->    <dependency>      <groupId>com.belerweb</groupId>      <artifactId>pinyin4j</artifactId>      <version>2.5.0</version>    </dependency>  </dependencies>

看完上述內容，你們掌握SpringBoot中怎么動態控制定時任務的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！