主题 : 简易而不简单的定时任务框架
级别: 举人
UID: 977
积分:194 加为好友
威望: 1 精华: 0
主题:33 回复:62
注册时间:2009-10-14
在线时长:0.01
1#   发表于:2011-11-03 21:53:43  IP:222.158.*.*
今天在网上发现了这个定时任务框架,绝对的好东西啊,废话不多直接上代码了。
整个框架就3个类:
ScheduleIterator.java

import java.util.Date;

public interface ScheduleIterator {
Date next();
}


Scheduler.java

import java.util.Date;
import java.util.Timer;
import java.util.TimerTask;

public class Scheduler {
class SchedulerTimerTask extends TimerTask {
private SchedulerTask schedulerTask;
private ScheduleIterator iterator;

public SchedulerTimerTask(SchedulerTask schedulerTask,
ScheduleIterator iterator) {
this.schedulerTask = schedulerTask;
this.iterator = iterator;
}

public void run() {
schedulerTask.run();
reschedule(schedulerTask, iterator);
}
}

private final Timer timer = new Timer();

public Scheduler() {
}

public void cancel() {
timer.cancel();
}

public void schedule(SchedulerTask schedulerTask, ScheduleIterator iterator) {

Date time = iterator.next();
if (time == null) {
schedulerTask.cancel();
} else {
synchronized (schedulerTask.lock) {
if (schedulerTask.state != SchedulerTask.VIRGIN) {
throw new IllegalStateException("Task already scheduled or cancelled");
}
schedulerTask.state = SchedulerTask.SCHEDULED;
schedulerTask.timerTask = new SchedulerTimerTask(schedulerTask,
iterator);
timer.schedule(schedulerTask.timerTask, time);
}
}
}

private void reschedule(SchedulerTask schedulerTask,
ScheduleIterator iterator) {

Date time = iterator.next();
if (time == null) {
schedulerTask.cancel();
} else {
synchronized (schedulerTask.lock) {
if (schedulerTask.state != SchedulerTask.CANCELLED) {
schedulerTask.timerTask = new SchedulerTimerTask(
schedulerTask, iterator);
timer.schedule(schedulerTask.timerTask, time);
}
}
}
}

}

SchedulerTask.java

import java.util.TimerTask;

public abstract class SchedulerTask implements Runnable {
final Object lock = new Object();

int state = VIRGIN;
static final int VIRGIN = 0;
static final int SCHEDULED = 1;
static final int CANCELLED = 2;

TimerTask timerTask;

protected SchedulerTask() {
}

public abstract void run();

public boolean cancel() {
synchronized (lock) {
if (timerTask != null) {
timerTask.cancel();
}
boolean result = (state == SCHEDULED);
state = CANCELLED;
return result;
}
}

public long scheduledExecutionTime() {
synchronized (lock) {
return timerTask == null ? 0 : timerTask.scheduledExecutionTime();
}
}

}
========================================================================================================
以下是我自己写的一个 ScheduleIterator  实现类 以及一个测试类
SimpleScheduleIterator.java

import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;

import com.jeecms.common.task.scheduling.ScheduleIterator;
/**
 * 时间进度迭代器
 * <li>返回 月/周/天/小时/分钟/秒 计划的下一次执行时间</li>
 * <li>约定:参数以逗号分隔,*号表示无值</li>
 * <li>参数解释:
 * <br>第一位:每个月的第几周</br>
 * <br>第二位:每周的第几天</br>
 * <br>第三位:天(几号)</br>
 * <br>第四位:小时(24小时制)</br>
 * <br>第五位:分钟</br>
 * <br>第六位:秒</br>
 * </li>
 * <li>参数样例:
 *  <br> 1,6,4,15,20,30       表示 从今天的15:20:30开始,每隔一个月执行一次,即下次执行时间是  下个月的第一周的第6天的15:20:30</br>
 *  <br> *,6,4,15,20,30       表示 从今天的15:20:30开始,每隔一周执行一次,即下次执行时间是  下一周的第6天的15:20:30</br>
 *  <br> *,*,4,15,20,30       表示 从今天的15:20:30开始,每隔一天执行一次,即下次执行时间是  下一天的15:20:30</br> 
 *  <br> *,*,*,15,20,30       表示 从今天的15:20:30开始,每隔一小时执行一次,即下次执行时间是  16:20:30</br>
 *  <br> *,*,*,*,20,30        表示 从这个小时的20:30开始,每隔一分钟执行一次,即下次执行时间是  *:21:30</br>
 *  <br> *,*,*,*,*,30         表示 从当前时间的30秒开始,每隔一秒执行一次,即下次执行时间是  *:*:31</br>
 * </li>
 * @author javacoo
 * @since 2011-11-03
 */
public class SimpleScheduleIterator implements ScheduleIterator {
private final ScheduleParamBean scheduleParamBean;
private final Calendar calendar = Calendar.getInstance();
private final Calendar orginCalendar = Calendar.getInstance();
public SimpleScheduleIterator(final ScheduleParamBean scheduleParamBean) {
this(scheduleParamBean, new Date());
}

public SimpleScheduleIterator(final ScheduleParamBean scheduleParamBean, Date date) {
this.scheduleParamBean = scheduleParamBean;

orginCalendar.setTime(date);
calendar.setTime(date);
if(null != scheduleParamBean.getWeekOfMonth()){
calendar.set(Calendar.WEEK_OF_MONTH, scheduleParamBean.getWeekOfMonth());
}
//如果设置了每周的第几天和一个月的第几天,则忽略一个月的第几天
if(null != scheduleParamBean.getDayOfWeek()){
calendar.set(Calendar.DAY_OF_WEEK, scheduleParamBean.getDayOfWeek());
}else if(null != scheduleParamBean.getDayOfMonth()){
calendar.set(Calendar.DAY_OF_MONTH, scheduleParamBean.getDayOfMonth());
}
if(null != scheduleParamBean.getHourOfDay()){
calendar.set(Calendar.HOUR_OF_DAY, scheduleParamBean.getHourOfDay());
}
if(null != scheduleParamBean.getMinute()){
calendar.set(Calendar.MINUTE, scheduleParamBean.getMinute());
}
if(null != scheduleParamBean.getSecond()){
calendar.set(Calendar.SECOND, scheduleParamBean.getSecond());
}
calendar.set(Calendar.MILLISECOND, 0);

if (!calendar.getTime().before(date)) {
System.out.println(calendar.getTime() +"大于当前时间:"+date);
if(null != scheduleParamBean.getWeekOfMonth()){
calendar.add(Calendar.MONTH, -1);
}else if(null != scheduleParamBean.getDayOfWeek()){
calendar.add(Calendar.DAY_OF_WEEK, -6);
}else if(null != scheduleParamBean.getDayOfMonth()){
calendar.add(Calendar.DAY_OF_MONTH, -1);
}else if(null != scheduleParamBean.getHourOfDay()){
calendar.add(Calendar.HOUR_OF_DAY, -1);
}else if(null != scheduleParamBean.getMinute()){
calendar.add(Calendar.MINUTE, -1);
}else if(null != scheduleParamBean.getSecond()){
calendar.add(Calendar.SECOND, -1);
}
}else{
System.out.println(calendar.getTime() +"小于当前时间:"+date);
if(null != scheduleParamBean.getDayOfMonth()){
calendar.add(Calendar.DAY_OF_MONTH, orginCalendar.get(Calendar.DAY_OF_MONTH) - scheduleParamBean.getDayOfMonth());
}else if(null != scheduleParamBean.getHourOfDay()){
calendar.add(Calendar.HOUR_OF_DAY, orginCalendar.get(Calendar.HOUR_OF_DAY) - scheduleParamBean.getHourOfDay());
}else if(null != scheduleParamBean.getMinute()){
calendar.add(Calendar.MINUTE, orginCalendar.get(Calendar.MINUTE) - scheduleParamBean.getMinute());
}else if(null != scheduleParamBean.getSecond()){
calendar.add(Calendar.SECOND, orginCalendar.get(Calendar.SECOND) - scheduleParamBean.getSecond());
}
}
}

public Date next() {
if(null != scheduleParamBean.getWeekOfMonth()){
calendar.add(Calendar.MONTH, 1);
}else if(null != scheduleParamBean.getDayOfWeek()){
calendar.add(Calendar.DAY_OF_WEEK, 6);
}else if(null != scheduleParamBean.getDayOfMonth()){
calendar.add(Calendar.DAY_OF_MONTH, 1);
}else if(null != scheduleParamBean.getHourOfDay()){
calendar.add(Calendar.HOUR_OF_DAY, 1);
}else if(null != scheduleParamBean.getMinute()){
calendar.add(Calendar.MINUTE, 1);
}else if(null != scheduleParamBean.getSecond()){
calendar.add(Calendar.SECOND, 1);
}
System.out.println("下次执行时间:"+calendar.getTime());
return calendar.getTime();
}


}


ScheduleParamBean.java

/**
 * 时间计划参数bean
 * @author javacoo
 * @since 2011-11-04
 */
public class ScheduleParamBean {
/**每个月的第几周,每周的第几天,每个月的第几天,小时(24小时制),分钟,秒*/
private Integer weekOfMonth,dayOfWeek,dayOfMonth,hourOfDay, minute, second;
        public ScheduleParamBean(){

}
        public ScheduleParamBean(Integer weekOfMonth, Integer dayOfWeek,
Integer dayOfMonth, Integer hourOfDay, Integer minute,
Integer second) {
super();
this.weekOfMonth = weekOfMonth;
this.dayOfWeek = dayOfWeek;
this.dayOfMonth = dayOfMonth;
this.hourOfDay = hourOfDay;
this.minute = minute;
this.second = second;
}
public Integer getWeekOfMonth() {
return weekOfMonth;
}

public void setWeekOfMonth(Integer weekOfMonth) {
this.weekOfMonth = weekOfMonth;
}

public Integer getDayOfWeek() {
return dayOfWeek;
}

public void setDayOfWeek(Integer dayOfWeek) {
this.dayOfWeek = dayOfWeek;
}

public Integer getDayOfMonth() {
return dayOfMonth;
}

public void setDayOfMonth(Integer dayOfMonth) {
this.dayOfMonth = dayOfMonth;
}

public Integer getHourOfDay() {
return hourOfDay;
}

public void setHourOfDay(Integer hourOfDay) {
this.hourOfDay = hourOfDay;
}

public Integer getMinute() {
return minute;
}

public void setMinute(Integer minute) {
this.minute = minute;
}

public Integer getSecond() {
return second;
}

public void setSecond(Integer second) {
this.second = second;
}

@Override
public String toString() {
return "ScheduleParamBean [dayOfMonth=" + dayOfMonth + ", dayOfWeek="
+ dayOfWeek + ", hourOfDay=" + hourOfDay + ", minute=" + minute
+ ", second=" + second + ", weekOfMonth=" + weekOfMonth + "]";
}




}



TestSchedule.java

/**
 * 测试
 * @author javacoo
 * @since 2011-11-03
 */
public class TestSchedule {
private final Scheduler scheduler = new Scheduler();

private final ScheduleParamBean scheduleParamBean;

public TestSchedule(final ScheduleParamBean scheduleParamBean) {
this.scheduleParamBean = scheduleParamBean;
}

public void start() {
scheduler.schedule(new SchedulerTask() {
public void run() {
execute();
}
private void execute() {
System.out.println("任务执行");
}
}, new SimpleScheduleIterator(scheduleParamBean));
}

public static void main(String[] args) {

ScheduleParamBean scheduleParamBean = new ScheduleParamBean();
scheduleParamBean.setWeekOfMonth(1);
scheduleParamBean.setDayOfWeek(6);
scheduleParamBean.setDayOfMonth(4);
scheduleParamBean.setHourOfDay(16);
scheduleParamBean.setMinute(22);
scheduleParamBean.setSecond(0);
TestSchedule test = new TestSchedule(scheduleParamBean);
test.start();

}

}








[/hide]
=====================================================================================================

以下是整合到JEECMS采集器多线程版的代码片段


[hide]/**
 * 开始执行采集任务
 */
public boolean start(Integer id) {
CmsAcquisition acqu = cmsAcquisitionMng.findById(id);
if (acqu == null || acqu.getStatus() == CmsAcquisition.START) {
return false;
}
TaskSchedulerManage taskManage = new TaskSchedulerManage(this,acqu);
taskManage.start();
return true;
}


private class TaskSchedulerManage {
private final Scheduler scheduler = new Scheduler();
private final MultiThreadAcquisitionSvcImpl multiThreadAcquisitionSvc;
private final CmsAcquisition acqu;
public TaskSchedulerManage(MultiThreadAcquisitionSvcImpl multiThreadAcquisitionSvc,CmsAcquisition acqu) {
this.multiThreadAcquisitionSvc = multiThreadAcquisitionSvc;
this.acqu = acqu;
}
public void start() {
scheduler.schedule(new SchedulerTask() {
public void run() {
processer();
}
private void processer() {
System.out.println("============开始执行计划任务=================");
new Thread(new MainThreadProcesser(multiThreadAcquisitionSvc,acqu)).start();
}
}, new SimpleScheduleIterator(new ScheduleParamBean(1,6,4,17,24,30)));
}
}
很简单吧!!关键是 ScheduleIterator实现类了,框架有多强大 完全取决于你的ScheduleIterator实现类了。



路漫漫其修远兮,唔将上下而求索,www.javacoo.com
级别: 举人
UID: 564
积分:113 加为好友
威望: 0 精华: 0
主题:6 回复:73
注册时间:2009-08-21
在线时长:0
2#   发表于:2011-11-05 08:28:56  IP:119.117.*.*
看一下
级别: 举人
UID: 1829
积分:106 加为好友
威望: 0 精华: 0
主题:1 回复:89
注册时间:2010-01-19
在线时长:0
3#   发表于:2011-11-06 14:28:01  IP:113.72.*.*
看一下
级别: 举人
UID: 25164
积分:105 加为好友
威望: 0 精华: 0
主题:8 回复:56
注册时间:2011-10-29
在线时长:0
4#   发表于:2011-11-07 00:44:32  IP:61.198.*.*
dfdsdfsd
级别: 白丁
UID: 25313
积分:6 加为好友
威望: 0 精华: 0
主题:0 回复:5
注册时间:2011-11-02
在线时长:0
5#   发表于:2011-11-07 08:35:17  IP:58.129.*.*
看看
级别: 秀才
UID: 24001
积分:74 加为好友
威望: 0 精华: 0
主题:4 回复:39
注册时间:2011-09-27
在线时长:0
6#   发表于:2011-11-07 10:19:10  IP:219.26.*.*
看看,谢谢分享
级别: 童生
UID: 25465
积分:29 加为好友
威望: 0 精华: 0
主题:0 回复:27
注册时间:2011-11-07
在线时长:0
7#   发表于:2011-11-08 11:35:21  IP:211.32.*.*
好东西 学习学习~
级别: 童生
UID: 24854
积分:46 加为好友
威望: 0 精华: 0
主题:2 回复:28
注册时间:2011-10-26
在线时长:0
8#   发表于:2011-11-08 15:13:38  IP:183.183.*.*
快快快快快
级别: 白丁
UID: 25526
积分:4 加为好友
威望: 0 精华: 0
主题:0 回复:4
注册时间:2011-11-08
在线时长:0
9#   发表于:2011-11-08 23:47:20  IP:115.227.*.*
什么框架
级别: 白丁
UID: 25535
积分:6 加为好友
威望: 0 精华: 0
主题:0 回复:5
注册时间:2011-11-09
在线时长:0
10#   发表于:2011-11-09 09:45:33  IP:163.87.*.*
学习
1 2 3 4 5 6 7 8 9 10 > >| 共39页