定时任务发展史(一)

public class TimerFactory implements BeanFactoryAware {

	private BeanFactory beanFactory;

	public Object getTask(String taskCode) {
		return beanFactory.getBean(taskCode+"Task");
	}
	
	public Object getTrigger(String taskCode) {
		return beanFactory.getBean(taskCode+"Trigger");
	}

	public void setBeanFactory(BeanFactory beanFactory) {
		this.beanFactory = beanFactory;
	}

	public BeanFactory getBeanFactory() {
		return beanFactory;
	}
}

public interface Task {
	public void executeTask();
}

public class ResendSmsAndEmailTask extends BaseTask{
	private static final String TASK_CODE = "resendSmsAndEmail";
	AtomicInteger ai = new AtomicInteger(0);
	
	public void execute(){
		try {
			ai = new AtomicInteger(0);
			// todo
		}catch (Exception e) {
			String exception = ExceptionUtils.getStackTrace(e);
			logger.error("stat error with exception[{}].", exception);
			this.recordTaskErrorDetail(this.taskRecordId, "ResendSmsAndEmailTask-" + e.getMessage(), exception);
		}finally{
			this.modifyTaskRecord(ai.get(), taskRecordId);
		}
	}
	
	public String getTaskNo() {
		return TASK_CODE;
	}
}

public class DynamicJobAssembler {

	private static Logger logger = LoggerFactory.getLogger(DynamicJobAssembler.class);

	@Resource
	Scheduler scheduler;

	@Resource
	TimerFactory timerFactory;

	@Resource
	TaskDao taskDao;

	public void init() {
		logger.info("start to assemble task from db.");
		List<TaskEntity> tasks = this.taskDao.getAllTask();
		if (tasks == null || tasks.size() <= 0) {
			return;
		}

		Map<String, String> jobNameMap = this.getAllJobNames();
		for (TaskEntity task : tasks) {
			logger.debug(task.toString());
			CronTriggerBean taskTrigger = (CronTriggerBean) timerFactory.getTrigger(task.getTaskNo());
			if (taskTrigger != null) {
				if (!task.getSchedulerRule().equals(taskTrigger.getCronExpression())) {
					try {
						taskTrigger.setCronExpression(task.getSchedulerRule());
					} catch (ParseException e) {
						logger.error("db task's cronExpression parse error:{}", e.getMessage());
					}
					try {
						logger.info("rescheduleJob jobName:{}",task.getTaskNo());
						scheduler.rescheduleJob(task.getTaskNo() + "Trigger", Scheduler.DEFAULT_GROUP, taskTrigger);
					} catch (SchedulerException e) {
						logger.error("revieved task[{},{}] reschedule error:{}", task.getTaskNo(), task.getSchedulerRule(), e.getMessage());
					}
				}
				jobNameMap.remove(task.getTaskNo() + "Job");
			}
		}

		if (jobNameMap != null) {
			logger.info("=====================================");
			logger.info("Jobs need to be removed:" + Arrays.toString(jobNameMap.keySet().toArray()));
			logger.info("=====================================");
			for (String jobName : jobNameMap.keySet()) {
				try {
					scheduler.deleteJob(jobName, jobNameMap.get(jobName));
				} catch (SchedulerException e) {
					logger.error("Error occured when deleting Job[{}] with Exception:{}", jobName, e.getMessage());
				}
			}
		}
		logger.info("end to assemble task from db.");
	}

	private Map<String, String> getAllJobNames() {
		Map<String, String> jobNameMap = new HashMap<String, String>();
		try {
			String[] groups = scheduler.getJobGroupNames();
			for (String group : groups) {
				String[] jobs = scheduler.getJobNames(group);
				if (jobs != null) {
					for (String job : jobs) {
						jobNameMap.put(job, group);
					}
				}
			}
		} catch (SchedulerException e1) {
			logger.error("Failed in geting all job names with exception:{}", e1.getMessage());
		}
		return jobNameMap;
	}

}