quartz与Springboot实现分布式自定义定时任务

悠悠 2023-06-22 04:25 31阅读 0赞

**一、导入POM文件**

<dependency>
                <groupId>org.quartz-scheduler</groupId>
                <artifactId>quartz</artifactId>
                <version>2.2.1</version>
            </dependency>
            <dependency>
                <groupId>org.quartz-scheduler</groupId>
                <artifactId>quartz-jobs</artifactId>
                <version>2.2.1</version>
            </dependency>
             <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter</artifactId>
                <version>2.1.8.RELEASE</version>
            </dependency>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-data-redis</artifactId>
                <version>2.1.8.RELEASE</version>
            </dependency>

**二、实现分布式锁**

public abstract class TestTask implements Job { 
    
        @Resource
        private StringRedisTemplate stringRedisTemplate;
    
        @Override
        @Transactional
        public void execute(JobExecutionContext context) { 
            log.info("执行调度任务：{}", context.getJobDetail().getKey());
            // TODO 此处阻止任务继续运行
            if (context.getScheduler() != null) { 
                return;
            }
            String taskId = getTaskId();
            MDC.put("REQIDS", taskId);
            String value = Utils.generateOpenId();
            if (Boolean.TRUE.equals(
                    stringRedisTemplate.opsForValue().setIfAbsent(taskId, value, 1200, TimeUnit.SECONDS))) { 
                log.info("任务锁: {} 成功加锁!", taskId);
                try { 
                    doTask(taskId);
                } catch (Exception e) { 
                    log.info(e.getMessage());
                } finally { 
                    if (value.equals(stringRedisTemplate.opsForValue().get(taskId))) { 
                        stringRedisTemplate.delete(taskId);
                        log.info("任务锁: {} 被成功释放!", taskId);
                    }
                }
            } else { 
                log.info("任务锁: {} 加锁失败!", taskId);
            }
        }
    
        private void doTask(String taskId) { 
            log.info("任务开始运行:{}", taskId);
        }
    
        //任务id,该任务id会作为分布式锁，保证只有一个节点执行任务
    	//可以通过继承方式运行
        protected abstract String getTaskId();
        //参数运行
        protected abstract void doTaskInternal(String taskId);
        //定时任务
        public abstract String cron();
    }

**三、设置定时任务**

@Configuration
    public class TaskConfig implements ApplicationListener<ApplicationReadyEvent> { 
    
        public JobFactory jobFactory(ApplicationContext applicationContext) { 
            return new AdaptableJobFactory() { 
                @Override
                protected Object createJobInstance(TriggerFiredBundle bundle) throws Exception { 
                    Class<?> clazz = bundle.getJobDetail().getJobClass();
                    return applicationContext.getBean(clazz);
                }
            };
        }
        private void addJob(Scheduler scheduler, String beanName, ReportTask reportTask) { 
            try { 
                String jobName = beanName + "-task";
                JobDetail jobDetail = JobBuilder.newJob(reportTask.getClass())
                        .withIdentity(jobName, "job-group-name").build();
                CronTrigger cronTrigger = TriggerBuilder.newTrigger()
                        .withIdentity(jobName, "task-group-name")
                        .withSchedule(CronScheduleBuilder.cronSchedule(reportTask.cron())).build();
                scheduler.scheduleJob(jobDetail, cronTrigger);
                if (!scheduler.isShutdown()) { 
                    scheduler.start();
                }
            } catch (Exception e) { 
                log.error(e.getMessage());
            }
        }
        @Override
        public void onApplicationEvent(ApplicationReadyEvent event) { 
            ConfigurableApplicationContext applicationContext = event.getApplicationContext();
            Map<String, ReportTask> reportTaskMap = applicationContext.getBeansOfType(ReportTask.class);
            Scheduler scheduler = applicationContext.getBean(SchedulerFactoryBean.class).getScheduler();
            try { 
                scheduler.setJobFactory(jobFactory(applicationContext));
            } catch (SchedulerException e) { 
                log.error(e.getMessage());
            }
            reportTaskMap.forEach((k, v) -> addJob(scheduler, k, v));
        }
    }

作者：叶飞