SpringBoot 如何实现定时任务，写得太好了

终码一生
419 阅读4分钟

1、前言

SpringBoot创建定时任务的方式很简单，主要有两种方式：

  • 基于注解的方式（@Scheduled）

  • 数据库动态配置

实际开发中，第一种需要在代码中写死表达式，如果修改起来，又得重启会显示很麻烦；所以我们往往会采取第二种方式，可以直接从数据库中读取定时任务的指定执行时间，无需重启。

2、基于注解（@Scheduled）

基于注解是一种静态的方式，只需要几行代码就可以搞定了

添加一个配置类

1 @Configuration  //标记配置类 2 @EnableScheduling   //开启定时任务 3 public class MyScheduleConfig { 4  5     //添加定时任务 6     @Scheduled(cron = "0/5 * * * * ?") 7     private void myTasks() { 8         System.out.println("执行定时任务 " + LocalDateTime.now()); 9     }10 }

上面代码的cron表达式表示每5秒执行一次，可以通过这个网站（qqe2.com/cron）去生成要的c…

启动应用，控制台看效果

图片

这个方式的确很简单方便，但前面介绍也说到了，有个缺点就是当我们需要去修改定时任务的执行周期或者停止的时候，我们需要到代码层去修改，重启。

3、数据库动态配置

这里使用MySQL数据库

1、表数据添加，资源配置

1.1 添加表

CREATE TABLE `scheduled_job` (  `job_id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键id',  `job_key` varchar(128) NOT NULL COMMENT '定时任务完整类名',  `cron_expression` varchar(20) NOT NULL COMMENT 'cron表达式',  `task_explain` varchar(50) NOT NULL DEFAULT '' COMMENT '任务描述',  `status` tinyint(4) NOT NULL DEFAULT '1' COMMENT '状态,1:正常;-1:停用',  PRIMARY KEY (`job_id`),  UNIQUE KEY `job_key` (`job_key`),  UNIQUE KEY `cron_key_unique_idx` (`job_key`)) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8 COLLATE=utf8_bin COMMENT='定时任务表';

1.2 插入两条数据，job_key根据是完整的类名

图片

1.3 引入依赖

<dependency>            <groupId>org.springframework.boot</groupId>            <artifactId>spring-boot-starter-web</artifactId>        </dependency>        <!--mysql-->        <dependency>            <groupId>mysql</groupId>            <artifactId>mysql-connector-java</artifactId>            <version>5.1.49</version>            <scope>runtime</scope>        </dependency>        <!--mybatis-plus-->        <dependency>            <groupId>com.baomidou</groupId>            <artifactId>mybatis-plus-boot-starter</artifactId>            <version>3.3.1.tmp</version>        </dependency>        <!--lombok简化代码-->        <dependency>            <groupId>org.projectlombok</groupId>            <artifactId>lombok</artifactId>            <version>1.18.20</version>            <scope>provided</scope>        </dependency>

1.4 配置application.yml

spring:  datasource:    url: jdbc:mysql://127.0.0.1:3306/test?userUnicode=true&characterEncoding=UTF8&useSSL=false    username: root    password: 123    driver-class-name: com.mysql.jdbc.Driverserver:  servlet:    context-path: /demo  port: 8888

2、疯狂贴代码

2.1 创建定时任务线程池

@Configuration@Slf4jpublic class ScheduledConfig {    @Bean    public ThreadPoolTaskScheduler threadPoolTaskScheduler() {        log.info("创建定时任务调度线程池 start");        ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();        threadPoolTaskScheduler.setPoolSize(20);        threadPoolTaskScheduler.setThreadNamePrefix("taskExecutor-");        threadPoolTaskScheduler.setWaitForTasksToCompleteOnShutdown(true);        threadPoolTaskScheduler.setAwaitTerminationSeconds(60);        log.info("创建定时任务调度线程池 end");        return threadPoolTaskScheduler;    }}

2.2 项目启动时初始化定时任务

@Slf4j@Componentpublic class ScheduledTaskRunner implements ApplicationRunner {    @Autowired    private ScheduledTaskService scheduledTaskService;    @Override    public void run(ApplicationArguments args) throws Exception {        log.info("----初始化定时任务开始----");        scheduledTaskService.initTask();        log.info("----初始化定时任务完成----");    }}

2.3 定时任务公共接口

public interface ScheduledOfTask extends Runnable{    void execute();    @Override    default void run() {        execute();    }}

2.4 创建两个定时任务实现类

@Component@Slf4jpublic class TaskJob1 implements ScheduledOfTask{    @Override    public void execute() {        log.info("执行任务1 "+ LocalDateTime.now());    }}

@Component@Slf4jpublic class TaskJob2 implements ScheduledOfTask{    @Override    public void execute() {        log.info("执行任务2 "+ LocalDateTime.now());    }}

2.5 定时任务管理接口

public interface ScheduledTaskService{    Boolean start(ScheduledJob scheduledJob);    Boolean stop(String jobKey);    Boolean restart(ScheduledJob scheduledJob);    void initTask();}

2.6 定时任务管理实现类

@Slf4j@Servicepublic class ScheduledTaskServiceImpl implements ScheduledTaskService {    /**     * 可重入锁     */    private ReentrantLock lock = new ReentrantLock();    /**     * 定时任务线程池     */    @Autowired    private ThreadPoolTaskScheduler threadPoolTaskScheduler;    /**     * 启动状态的定时任务集合     */    public Map<String, ScheduledFuture> scheduledFutureMap = new ConcurrentHashMap<>();    @Autowired    private ScheduledJobService scheduledJobService;    @Override    public Boolean start(ScheduledJob scheduledJob) {        String jobKey = scheduledJob.getJobKey();        log.info("启动定时任务"+jobKey);        //添加锁放一个线程启动，防止多人启动多次        lock.lock();        log.info("加锁完成");        try {            if(this.isStart(jobKey)){                log.info("当前任务在启动状态中");                return false;            }            //任务启动            this.doStartTask(scheduledJob);        } finally {            lock.unlock();            log.info("解锁完毕");        }        return true;    }    /**     * 任务是否已经启动     */    private Boolean isStart(String taskKey) {        //校验是否已经启动        if (scheduledFutureMap.containsKey(taskKey)) {            if (!scheduledFutureMap.get(taskKey).isCancelled()) {                return true;            }        }        return false;    }    @Override    public Boolean stop(String jobKey) {        log.info("停止任务 "+jobKey);        boolean flag = scheduledFutureMap.containsKey(jobKey);        log.info("当前实例是否存在 "+flag);        if(flag){            ScheduledFuture scheduledFuture = scheduledFutureMap.get(jobKey);            scheduledFuture.cancel(true);            scheduledFutureMap.remove(jobKey);        }        return flag;    }    @Override    public Boolean restart(ScheduledJob scheduledJob) {        log.info("重启定时任务"+scheduledJob.getJobKey());        //停止        this.stop(scheduledJob.getJobKey());        return this.start(scheduledJob);    }    /**     * 执行启动任务     */    public void doStartTask(ScheduledJob sj){        log.info(sj.getJobKey());        if(sj.getStatus().intValue() != 1)            return;        Class<?> clazz;        ScheduledOfTask task;        try {            clazz = Class.forName(sj.getJobKey());            task = (ScheduledOfTask) SpringContextUtil.getBean(clazz);        } catch (ClassNotFoundException e) {            throw new IllegalArgumentException("spring_scheduled_cron表数据" + sj.getJobKey() + "有误", e);        }        Assert.isAssignable(ScheduledOfTask.class, task.getClass(), "定时任务类必须实现ScheduledOfTask接口");        ScheduledFuture scheduledFuture = threadPoolTaskScheduler.schedule(task,(triggerContext -> new CronTrigger(sj.getCronExpression()).nextExecutionTime(triggerContext)));        scheduledFutureMap.put(sj.getJobKey(),scheduledFuture);    }    @Override    public void initTask() {        List<ScheduledJob> list = scheduledJobService.list();        for (ScheduledJob sj : list) {            if(sj.getStatus().intValue() == -1) //未启用                continue;            doStartTask(sj);        }    }}

2.7上面用到的获取Bean的工具类SpringContextUtil

@Componentpublic class SpringContextUtil implements ApplicationContextAware {    private static ApplicationContext applicationContext = null;    @Override    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {        if(SpringContextUtil.applicationContext == null){            SpringContextUtil.applicationContext  = applicationContext;        }    }    public static ApplicationContext getApplicationContext() {        return applicationContext;    }    public static Object getBean(String name){        return getApplicationContext().getBean(name);    }    public static <T> T getBean(Class<T> clazz){        return getApplicationContext().getBean(clazz);    }    public static <T> T getBean(String name,Class<T> clazz){        return getApplicationContext().getBean(name, clazz);    }}

2.8 表操作对应的一些类

Pojo

@Data@TableName("scheduled_job")public class ScheduledJob {    @TableId(value = "job_id",type = IdType.AUTO)    private Integer jobId;    private String jobKey;    private String cronExpression;    private String taskExplain;    private Integer status;}

ScheduledJobMapper

1 public interface ScheduledJobMapper extends BaseMapper<ScheduledJob> {2 }

ScheduledJobService

public interface ScheduledJobService extends IService<ScheduledJob> {    /**     * 修改定时任务，并重新启动     * @param scheduledJob     * @return     */    boolean updateOne(ScheduledJob scheduledJob);}

@Service@Slf4jpublic class ScheduledJobServiceImpl extends ServiceImpl<ScheduledJobMapper, ScheduledJob> implements ScheduledJobService{    @Autowired    private ScheduledTaskService scheduledTaskService;    @Override    public boolean updateOne(ScheduledJob scheduledJob) {        if(updateById(scheduledJob))            scheduledTaskService.restart(getById(scheduledJob.getJobId()));        return true;    }}

2.9 修改定时任务的接口

@RestController@RequestMapping("/job")public class ScheduledJobController {    @Autowired    private ScheduledJobService scheduledJobService;    @PostMapping(value = "/update")    public CallBackResult update(HttpServletRequest request, ScheduledJob scheduledJob){         if(scheduledJobService.updateOne(scheduledJob))             return new CallBackResult(true,"修改成功");         return new CallBackResult(false,"修改失败");    }}

3、测试结果

3.1 启动项目，看下定时任务的执行结果，控制台输出结果

图片

我们可以看到任务1是每5秒执行一次，任务2是12秒执行一次

3.2 修改任务1的cron参数或者状态

3.2.1 修改cron，执行周期改为20秒执行一次，状态不变

图片

图片

图片

再看控制台输出结果，任务2没变化，任务1由5秒一次变成了20秒一次了

图片

3.2.1 修改状态

图片

再看控制台输出结果，任务2没变化，任务1已经不再执行了

图片

4、最后

第二种方式支持通过接口的方式去改动，并且不需要重启，当然啦，也可以直接在数据库中添加或修改数据后重启项目，配置更加灵活一点。

如果是一个固定的需求，执行周期一定不会变的了，推荐还是第一种写法，毕竟简单嘛。

PS：防止找不到本篇文章，可以收藏点赞，方便翻阅查找哦

avatar
文章
阅读
粉丝