package myschedule.service.quartz;import static myschedule.service.ErrorCode.SCHEDULER_PROBLEM;import static org.quartz.CalendarIntervalScheduleBuilder.calendarIntervalSchedule;import static org.quartz.CronScheduleBuilder.cronSchedule;import static org.quartz.JobBuilder.newJob;import static org.quartz.SimpleScheduleBuilder.simpleSchedule;import static org.quartz.TriggerBuilder.newTrigger;import java.io.ByteArrayInputStream;import java.io.InputStream;import java.text.ParseException;import java.util.ArrayList;import java.util.Date;import java.util.List;import java.util.Map;import java.util.Set;import myschedule.service.ErrorCodeException;import org.quartz.Calendar;import org.quartz.CalendarIntervalTrigger;import org.quartz.CronTrigger;import org.quartz.DateBuilder.IntervalUnit;import org.quartz.Job;import org.quartz.JobDetail;import org.quartz.JobExecutionContext;import org.quartz.JobKey;import org.quartz.Scheduler;import org.quartz.SchedulerException;import org.quartz.SchedulerMetaData;import org.quartz.SimpleTrigger;import org.quartz.Trigger;import org.quartz.TriggerKey;import org.quartz.impl.matchers.GroupMatcher;import org.quartz.spi.MutableTrigger;import org.quartz.spi.OperableTrigger;/** * A template to simplify quartz Scheduler class. It provides some more convenient methods to schedule * jobs and convert Quartz's "checked" SchedulerException into an "unchecked" ErrorCodeException. * * @author Zemian Deng */public class SchedulerTemplate {protected Scheduler scheduler;public SchedulerTemplate(Scheduler scheduler) {this.scheduler = scheduler;}public void setScheduler(Scheduler scheduler) {this.scheduler = scheduler;}public Scheduler getScheduler() {return scheduler;}public List<String> getCalendarNames() {try {return scheduler.getCalendarNames();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public boolean isShutdown() {try {return scheduler.isShutdown();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public boolean isInStandbyMode() {try {return scheduler.isInStandbyMode();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public boolean isStarted() {try {return scheduler.isStarted();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public List<JobExecutionContext> getCurrentlyExecutingJobs() {try {List<JobExecutionContext> result = new ArrayList<JobExecutionContext>();List<?> jobs = scheduler.getCurrentlyExecutingJobs();for (Object job : jobs) {JobExecutionContext jobec = (JobExecutionContext)job;result.add(jobec);}return result;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public SchedulerMetaData getSchedulerMetaData() {try {return scheduler.getMetaData();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Get all the JobDetails in the scheduler. */public List<JobDetail> getJobDetails() {try {List<JobDetail> jobs = new ArrayList<JobDetail>();List<String> groups = scheduler.getJobGroupNames();for (String group : groups) {Set<JobKey> keys = scheduler.getJobKeys(GroupMatcher.jobGroupEquals(group));for (JobKey key : keys) {JobDetail jobDetail = scheduler.getJobDetail(key);jobs.add(jobDetail);}}return jobs;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Get all the triggers in the scheduler. */public List<? extends Trigger> getTriggers(JobDetail jobDetail) {try {List<? extends Trigger> triggers = scheduler.getTriggersOfJob(jobDetail.getKey());return triggers;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public JobDetail getJobDetail(String jobName, String group) {try {JobKey key = JobKey.jobKey(jobName, group);return scheduler.getJobDetail(key);} catch (Exception e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public Trigger getTrigger(String triggerName, String group) {try {TriggerKey key = TriggerKey.triggerKey(triggerName, group);return scheduler.getTrigger(key);} catch (Exception e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public Calendar getCalendar(String calName) {try {return scheduler.getCalendar(calName);} catch (Exception e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** * Get a list of next fire time dates up to maxCount time. If next fire time needed * before maxCount, then there should be a null object in the last element of the list. */public List<Date> getNextFireTimes(Trigger trigger, Date startTime, int maxCount) {// We will clone the original trigger so we may call its triggered() to get a proper fireTime.OperableTrigger clonedTrigger = (OperableTrigger)((OperableTrigger)trigger).clone();Calendar cal = null; if (clonedTrigger.getNextFireTime() == null) { clonedTrigger.computeFirstFireTime(cal); } List<Date> list = new ArrayList<Date>();Date nextDate = startTime;int count = 0;while(count++ < maxCount) {nextDate = clonedTrigger.getFireTimeAfter(nextDate);if (nextDate == null)break;list.add(nextDate);clonedTrigger.triggered(cal);}return list;}/** * Get a list of next fire time dates up to maxCount time. If next fire time needed * before maxCount, then there should be a null object in the last element of the list. */public List<Date> getNextFireTimesWithCalendar(Trigger trigger, Date startTime, int maxCount) {List<Date> dates = getNextFireTimes(trigger, startTime, maxCount);String calName = trigger.getCalendarName();if (calName == null) {return dates;}// Else check if dates has excluded by calendar or not.Calendar cal = getCalendar(calName);List<Date> result = new ArrayList<Date>();for (Date dt : dates) {if (cal.isTimeIncluded(dt.getTime())) {result.add(dt);}}return result;}/** Update existing job with newJobDetail and return the old one. */public JobDetail updateJobDetail(JobDetail newJobDetail) {try {JobDetail oldJob = scheduler.getJobDetail(newJobDetail.getKey());scheduler.addJob(newJobDetail, true);return oldJob;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Update existing trigger with newTrigger and return the old one. */public Trigger updateTrigger(Trigger newTrigger) {try {Trigger oldTrigger = scheduler.getTrigger(newTrigger.getKey());scheduler.rescheduleJob(oldTrigger.getKey(), newTrigger);return oldTrigger;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Add a JobDetail with a trigger schedule when to fire. */public Date scheduleJob(JobDetail jobDetail, Trigger trigger) {try {Date nextFireTime = scheduler.scheduleJob(jobDetail, trigger);return nextFireTime;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Schedule new trigger to an existing JobDetail. You need to set "trigger.setJobName()". */public void scheduleJob(Trigger trigger) {try {scheduler.scheduleJob(trigger);} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Remove a trigger and its JobDetail if it's not set durable. */public Trigger uncheduleJob(String triggerName, String triggerGroup) {try {Trigger trigger = scheduler.getTrigger(TriggerKey.triggerKey(triggerName, triggerGroup));boolean success = scheduler.unscheduleJob(trigger.getKey());if (!success)throw new SchedulerException("Failed to unschedule job. Trigger name=" + triggerName + ", group=" + triggerGroup);return trigger;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public void addJob(JobDetail job, boolean replace) {try {scheduler.addJob(job, replace);} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Remove a JobDetail and all the triggers associated with it. */public List<? extends Trigger> deleteJob(String jobName, String group) {try {JobKey key = JobKey.jobKey(jobName, group);List<? extends Trigger> triggers = scheduler.getTriggersOfJob(key);boolean success = scheduler.deleteJob(key);if (!success)throw new SchedulerException("Unable to delete job " + key);return triggers;} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** * Load job scheduling data xml using XMLSchedulingDataProcessor. * @param xml - xml content of job_scheduling_data xml. * @return XMLSchedulingDataProcessor instance will contain all the jobs parse in xml. */public XmlJobLoader scheduleXmlSchedulingData(String xml) {try {// XmlJobLoader is not only just a loader, but also use to store what's loaded!XmlJobLoader xmlJobLoader = XmlJobLoader.newInstance(); String systemId = XmlJobLoader.XML_SYSTEM_ID;InputStream istream = new ByteArrayInputStream(xml.getBytes());xmlJobLoader.processStreamAndScheduleJobs(istream, systemId, scheduler);return xmlJobLoader;} catch (Exception e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}/** Run a job immediately with a non-volatile trigger (remove as soon as it's finished.) */public void runJobNow(String name, String group) {try {JobKey key = JobKey.jobKey(name, group);scheduler.triggerJob(key);} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public static JobDetail createJobDetail(String name, Class<? extends Job> jobClass) {return createJobDetail(name, null, true, jobClass, null);}public static JobDetail createJobDetail(String name, String group, boolean durable, Class<? extends Job> jobClass, Map<String, Object> data) {JobDetail jobDetail = newJob(jobClass).withIdentity(name, group).storeDurably(durable).build();if (data != null)jobDetail.getJobDataMap().putAll(data);return jobDetail;}public static MutableTrigger createCalendarIntervalTrigger(String name, int interval, IntervalUnit intervalUnit) {return createCalendarIntervalTrigger(name, null, interval, intervalUnit, null, null);}public static MutableTrigger createCalendarIntervalTrigger(String name, String group,int interval, IntervalUnit intervalUnit,Date startTime, Date endTime) {if (group == null)group = Scheduler.DEFAULT_GROUP;if (startTime == null)startTime = new Date();CalendarIntervalTrigger trigger = newTrigger().withIdentity(name, group).startAt(startTime).endAt(endTime).withSchedule(calendarIntervalSchedule().withInterval(interval, intervalUnit)).build();return (MutableTrigger)trigger;}public static MutableTrigger createCronTrigger(String name, String cron) {return createCronTrigger(name, null, cron, null, null);}public static MutableTrigger createCronTrigger(String name, String group, String cron, Date startTime, Date endTime) {if (group == null)group = Scheduler.DEFAULT_GROUP;if (startTime == null)startTime = new Date();try {CronTrigger trigger = newTrigger().withIdentity(name, group).startAt(startTime).endAt(endTime).withSchedule(cronSchedule(cron)).build();return (MutableTrigger)trigger;} catch (ParseException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public static MutableTrigger createSimpleTrigger(String name, String group, int repeatTotalCount, long repeatInterval, Date startTime, Date endTime) {if (group == null)group = Scheduler.DEFAULT_GROUP;if (startTime == null)startTime = new Date();// Quartz's SimpleTrigger's repeatCount is one less than repeatTotalCount, so we need to adjust.int repeatCount = repeatTotalCount - 1;if (repeatTotalCount < 0)repeatCount = SimpleTrigger.REPEAT_INDEFINITELY;SimpleTrigger trigger = newTrigger().withIdentity(name, group).startAt(startTime).endAt(endTime).withSchedule(simpleSchedule().withRepeatCount(repeatCount).withIntervalInMilliseconds(repeatInterval)).build();return (MutableTrigger)trigger;}public Date scheduleCronJob(String name, String group, String cron, Date startTime, Date endTime,Class<? extends Job> jobClass, Map<String, Object> data) {JobDetail job = createJobDetail(name, group, false, jobClass, data);Trigger trigger = createCronTrigger(name, group, cron, startTime, endTime);return scheduleJob(job, trigger);}public Date scheduleRepeatForeverJob(String name, long repeatInterval, Class<? extends Job> jobClass) {return scheduleRepeatableJob(name, null, null, null, -1, repeatInterval, jobClass, null);}public Date scheduleRepeatableJob(String name, int repeatCount, long repeatInterval, Class<? extends Job> jobClass) {return scheduleRepeatableJob(name, null, null, null, repeatCount, repeatInterval, jobClass, null);}public Date scheduleRepeatableJob(String name, String group, Date startTime, Date endTime,int repeatTotalCount, long repeatInterval,Class<? extends Job> jobClass, Map<String, Object> data) {JobDetail job = createJobDetail(name, group, false, jobClass, data);Trigger trigger = createSimpleTrigger(name, group, repeatTotalCount, repeatInterval, startTime, endTime);return scheduleJob(job, trigger);}public Date scheduleOnetimeJob(String name, Class<? extends Job> jobClass) {return scheduleRepeatableJob(name, 1, 0, jobClass);}public Date scheduleOnetimeJob(String name, String group, Date startTime, Date endTime, Class<? extends Job> jobClass, Map<String, Object> data) {return scheduleRepeatableJob(name, group, startTime, endTime, 1, 0, jobClass, data);}public String getSchedulerName() {try {return scheduler.getSchedulerName();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public String getSchedulerInstanceId() {try {return scheduler.getSchedulerInstanceId();} catch (SchedulerException e) {throw new ErrorCodeException(SCHEDULER_PROBLEM, e);}}public String getSchedulerNameAndId() {return getSchedulerName() + "_$_" + getSchedulerInstanceId();}@Overridepublic String toString() {return "QuartzScheduler[" + getSchedulerNameAndId() + "]";}}
