java实现多线程之定时器任务

package com.zkn.newlearn.thread; 
 
import java.util.Timer; 
import java.util.TimerTask; 
 
/** 
 * 传统的定时器 
 * Created by zkn on 2016/11/1. 
 */ 
public class TraditionalTimerTest01 { 
 
  public static void main(String[] args){ 
    //TimerTask是Runnable接口的一个实现类是，它是一个抽像类 
    //schedule是一个重载方法：第一个参数TimerTask的实现类。 
    // 第二个参数是第一次执行的时间。 
    // 第三个参数是间隔时间 
    new Timer().schedule(new TimerTask() { 
      @Override 
      public void run() { 
 
        System.out.println("这是一个定时器任务!"); 
      } 
    },1000,2000); 
  } 
} 

package com.zkn.newlearn.thread; 
 
import java.util.Timer; 
import java.util.TimerTask; 
 
/** 
 * Created by zkn on 2016/11/1. 
 */ 
public class TraditionalTimerTest02 { 
 
  public static void main(String[] args){ 
 
    new Timer().schedule(new MyTimerTask01(),4000); 
  } 
 
  private static class MyTimerTask01 extends TimerTask{ 
 
    @Override 
    public void run() { 
      System.out.println("我是TimerTask1，我被执行了!"); 
      new Timer().schedule(new MyTimerTask02(),2000); 
    } 
  } 
 
  private static class MyTimerTask02 extends TimerTask { 
 
    @Override 
    public void run() { 
      System.out.println("我是TimerTask2，我被执行了!"); 
      new Timer().schedule(new MyTimerTask01(),4000); 
    } 
  } 
} 

private final TaskQueue queue = new TaskQueue(); 
 
/** 
 * The timer thread. 
 */ 
private final TimerThread thread = new TimerThread(queue); 

public Timer(String name) { 
  thread.setName(name); 
  thread.start(); 
} 

public Timer() { 
  this("Timer-" + serialNumber()); 
} 

public void run() { 
  try { 
    mainLoop(); 
  } finally { 
    // Someone killed this Thread, behave as if Timer cancelled 
    synchronized(queue) { 
      newTasksMayBeScheduled = false; 
      queue.clear(); // Eliminate obsolete references 
    } 
  } 
} 

private void mainLoop() { 
  while (true) { // 开始死循环 
    try { 
      TimerTask task; 
      boolean taskFired; 
      synchronized(queue) { 
        // 如果任务队列中为空并且定时任务没有被取消话，线程被挂起 等待执行任务的到来 
        while (queue.isEmpty() && newTasksMayBeScheduled) 
          queue.wait(); 
        if (queue.isEmpty()) 
          break; // 如果任务队列中没有任务了，则结束循环结束任务 
 
        // 如果队列中有执行任务的话，接着往下走 
        long currentTime, executionTime; 
        task = queue.getMin(); 
        synchronized(task.lock) { 
          if (task.state == TimerTask.CANCELLED) { 
            queue.removeMin(); 
            continue; // 如果执行任务被取消的话 则移除当前任务。这里会重新排队列里的任务执行顺序 
          } 
          currentTime = System.currentTimeMillis(); 
          executionTime = task.nextExecutionTime; 
          if (taskFired = (executionTime<=currentTime)) { 
            if (task.period == 0) { // 如果只执行一次的话，则在执行完之后，结束执行任务 
              queue.removeMin(); 
              task.state = TimerTask.EXECUTED; 
            } else { // 如果是固定频率执行任务的话，则计算下次执行的时间 
              queue.rescheduleMin( 
               task.period<0 ? currentTime  - task.period 
                      : executionTime + task.period); 
            } 
          } 
        } 
        if (!taskFired) // 不到任务执行的时候 等待线程调用 
          queue.wait(executionTime - currentTime); 
      } 
      if (taskFired) // 任务执行时间到，调用任务的run方法，执行任务 
        task.run(); 
    } catch(InterruptedException e) { 
    } 
  } 
} 

public void schedule(TimerTask task, long delay, long period) { 
  if (delay < 0) // 如果第一次执行的时间小于0 抛出异常 
    throw new IllegalArgumentException("Negative delay."); 
  if (period <= 0) //间隔时间小于等于 0 抛出异常 
    throw new IllegalArgumentException("Non-positive period."); 
  sched(task, System.currentTimeMillis()+delay, -period); 
} 

private void sched(TimerTask task, long time, long period) { 
  if (time < 0) 
    throw new IllegalArgumentException("Illegal execution time."); 
 
  // Constrain value of period sufficiently to prevent numeric 
  // overflow while still being effectively infinitely large.这个间隔时间到死基本上也执行不到 
  if (Math.abs(period) > (Long.MAX_VALUE >> 1)) 
    period >>= 1; 
 
  synchronized(queue) { 
    if (!thread.newTasksMayBeScheduled) //在任务的执行方法中 如果定时任务已经被取消的话 则抛出异常 
      throw new IllegalStateException("Timer already cancelled."); 
 
    synchronized(task.lock) { //object对象锁 
      if (task.state != TimerTask.VIRGIN) // 刚开是执行任务的时候 任务的状态应该是0的 
        throw new IllegalStateException( 
          "Task already scheduled or cancelled"); 
      task.nextExecutionTime = time; //下次执行时间 在上面的mainLoop方法中有用到 
      task.period = period; //设置任务的间隔时间，在上面的mainLoop方法中有用到 
      task.state = TimerTask.SCHEDULED; // 调度方法被调用 设置定时任务的状态为 已调度未执行 
    } 
 
    queue.add(task); //把执行任务加入到任务队列中 
    if (queue.getMin() == task) 
      queue.notify(); // 如果任务队列中的第一个任务为当前任务的话，则把当前任务放入到等锁池中 等待执行 
  } 
} 

  class TaskQueue { 
  //定义一个TimerTask的堆数组 <span style="white-space:pre">  </span> 
  private TimerTask[] queue = new TimerTask[128]; 
 
  //任务队列中的任务数<span style="white-space:pre"> </span> 
  private int size = 0; 
 
 
  int size() { 
    return size; 
  } 
 
  //添加任务到优先队列中 如果数组的长度不够的话会扩展数组 
  void add(TimerTask task) { 
    // Grow backing store if necessary 
    if (size + 1 == queue.length) 
      queue = Arrays.copyOf(queue, 2*queue.length); 
 
    queue[++size] = task; 
    fixUp(size); 
  } 
 
  //获取优先执行的任务 
  TimerTask getMin() { 
    return queue[1]; 
  } 
 
   
  TimerTask get(int i) { 
    return queue[i]; 
  } 
  //移除掉排在第一位的不能执行的任务 
  void removeMin() { 
    queue[1] = queue[size]; 
    queue[size--] = null; // Drop extra reference to prevent memory leak 把对象置空 等待gc回收 
    fixDown(1); 
  } 
 
  //删除任务队列队列中的任务 这里用来一个断言 来判断 i 不能大于 size 
  void quickRemove(int i) { 
    assert i <= size; 
 
    queue[i] = queue[size]; 
    queue[size--] = null; // Drop extra ref to prevent memory leak 
  } 
 
  //重新设置优先执行任务的执行时间 并对任务队列进行重新排序 以确保最优先的任务 优先被执行 
  void rescheduleMin(long newTime) { 
    queue[1].nextExecutionTime = newTime; 
    fixDown(1); 
  } 
 
 
  boolean isEmpty() { 
    return size==0; 
  } 
 
  //清空任务队列 定时任务结束  
  void clear() { 
    // Null out task references to prevent memory leak 
    for (int i=1; i<=size; i++) 
      queue[i] = null; 
 
    size = 0; 
  } 
 
  //两个堆排序 选出最优先的执行任务 
  private void fixUp(int k) { 
    while (k > 1) { 
      int j = k >> 1; 
      if (queue[j].nextExecutionTime <= queue[k].nextExecutionTime) 
        break; 
      TimerTask tmp = queue[j]; queue[j] = queue[k]; queue[k] = tmp; 
      k = j; 
    } 
  } 
 
  private void fixDown(int k) { 
    int j; 
    while ((j = k << 1) <= size && j > 0) { 
      if (j < size && 
        queue[j].nextExecutionTime > queue[j+1].nextExecutionTime) 
        j++; // j indexes smallest kid 
      if (queue[k].nextExecutionTime <= queue[j].nextExecutionTime) 
        break; 
      TimerTask tmp = queue[j]; queue[j] = queue[k]; queue[k] = tmp; 
      k = j; 
    } 
  } 
 
  void heapify() { 
    for (int i = size/2; i >= 1; i--) 
      fixDown(i); 
  } 
}