ScheduledThreadPoolExecutor源码分析

在项目中经常需要开启一些定时任务，只知道如何使用却不知道内部原理，今天就来分析下ScheduledThreadPoolExecutor线程池内部的运行原理。

ScheduledThreadPoolExecutor继承了ThreadPoolExecutor，所以它有线程池的功能（之前写过一篇分析ThreadPoolExecutor源码的文章）。并且它实现了ScheduledExecutorService接口，所以它还有定时或者周期执行任务的功能。

 

构造函数：

从代码中可以看出，其实就是线程池的初始化，只不过出了corePoolSize，threadFactory和rejectHandler，其他的都是写死的，看来该线程池实现的核心应该就是那个DelayWorkQueue。

和一般线程池不用，它有如下三个特有的方法：

1. schedule                 指定延迟后调度任务执行，且只执行一次
2. scheduleAtFixedRate 指定延迟后开始执行任务，然后以固定周期不停执行
3. scheduleWithFixedDelay 指定延迟后开始执行任务，任务完成后等待固定周期再执行

接下来根据自己写的demo分别对它们的内部源码进行分析，demo如下：

public class ScheduledThreadPoolExecutorTest {

    public static void main(String[] args) {
        ScheduledThreadPoolExecutor pool = new ScheduledThreadPoolExecutor(5);
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

        //test1 指定延时后调度任务执行
        pool.schedule(() -> {
            System.out.println("Thread1 start Sleep...");
            try {
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("Thread1 finish Sleep...");
        },0L, TimeUnit.SECONDS);

        //test2 指定时延后开始执行任务，以后每隔period的时长再次执行该任务
        pool.scheduleAtFixedRate(new Runnable() {
            @Override
            public void run() {
                System.out.println("Thread1 start Sleep...");
                try {
//                    TimeUnit.SECONDS.sleep(2);
                    TimeUnit.MICROSECONDS.sleep(500);
                    System.out.println(sdf.format(System.currentTimeMillis()));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("Thread1 finish Sleep...");
            }
        },0L,1L, TimeUnit.SECONDS);

        //test3 指定时延后开始执行任务，以后任务执行完成后等待delay时长，再次执行任务
        pool.scheduleWithFixedDelay(new Runnable() {
            @Override
            public void run() {
                System.out.println("Thread1 start Sleep...");
                try {
                    TimeUnit.SECONDS.sleep(1);
                    System.out.println(sdf.format(System.currentTimeMillis()));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("Thread1 finish Sleep...");
            }
        },0L,1L, TimeUnit.SECONDS);


    }
    
}

 

Scheduler():

decorateTask()里面什么也没做，直接就是返回第二个参数

triggerTime()是计算什么时刻开始执行任务，时间精确到纳秒：

可以看出如果输入负数，将直接变为0，即没有延时。

 

再来看下这个SchedulerdFutureTask对象，这个对象继承自FutureTask，所以它拥有FutureTask的功能并扩展了一些方法，稍后会说到。

最后调用delayedExecute()将任务放入队列后，线程池的机制保证会有线程不停的从队列中获取任务执行。(可以看我之前分析线程池的那篇文章)

注意：DelayedWorkQueue是一个优先阻塞队列，它会根据任务的延迟时间不同进行排序，延时时间越短的在越前面，先被获取执行。如果延时相同的话，在按照前面提到的sequenceNumber的顺序从小到大执行，最后文章再分析这个队列的实现。

 

顺便再提一下，task.cancle()调用的是FutureTask类中的方法：

True表示如果当前任务正在执行，立即interrupt任务，任务的state最终会变成INTERRUPTED

False表示等任务完成之后再取消任务

 

但是，到目前为止只是看到任务入队了，但是任务是如何做到延迟执行的？看下这个Task的run()方法：

这里的setNextRunTime()是重新设置了当前Task中的parviate long time变量，这个变量表示的是任务下次执行的时间。后下那个方法是重新入队操作，并进行线程池状态的检查。

好了，现在发现run()方法内部也只是执行任务而已，并没有延时相关的代码，那么再来看下获取任务的代码，看下是不是那里进行延时的。之前分析线程池的时候看到过，从线程池中拿去数据是take()或者offer(指定了超时时间)，那就再来看下DelayedWorkQueue中的take()方法：

可以看到，是在获取任务的时候阻塞进行延时的，所以基本是怎么执行的大概也确定了。DelaydWorkQueue队列后续出一篇文章单独分析下，里面其实是堆结构，顺便可以复习下数据结构。