调度线程池ScheduledThreadPoolExecutor源码解析

前言

ScheduledThreadPoolExecutor可以用来很方便实现我们的调度任务，那大家知道它是怎么实现的吗，本文就带大家来揭晓谜底。

实现机制分析

我们先思考下，如果让大家去实现ScheduledThreadPoolExecutor可以周期性执行任务的功能，需要考虑哪些方面呢？

1. ScheduledThreadPoolExecutor的整体实现思路是什么呢？

答： 我们是不是可以继承线程池类，按照线程池的思路，将任务先丢到阻塞队列中，等到时间到了，工作线程就从阻塞队列获取任务执行。

1. 如何实现等到了未来的时间点就开始执行呢？

答： 我们可以根据参数获取这个任务还要多少时间执行，那么我们是不是可以从阻塞队列中获取任务的时候，通过条件队列的的awaitNanos(delay)方法，阻塞一定时间。

1. 如何实现 任务的重复性执行呢？

答：这就更加简单了，任务执行完成后，把她再次加入到队列不就行了吗。

源码解析

类结构图

ScheduledThreadPoolExecutor的类结构图如上图所示，很明显它是在我们的线程池ThreadPoolExecutor框架基础上扩展的。

• ScheduledExecutorService：实现了该接口，封装了调度相关的API
• ThreadPoolExecutor：继承了该类，保留了线程池的能力和整个实现的框架
• DelayedWorkQueue：内部类，延迟阻塞队列。
• ScheduledFutureTask：延迟任务对象，包含了任务、任务状态、剩余的时间、结果等信息。

重要属性

通过ScheduledThreadPoolExecutor类的成员属性，我们可以了解它的数据结构。

1. shutdown 后是否继续执行周期任务(重复执行)

private volatile boolean continueExistingPeriodicTasksAfterShutdown;
复制代码

1. shutdown 后是否继续执行延迟任务（只执行一次）

private volatile boolean executeExistingDelayedTasksAfterShutdown = true;
复制代码

1. 调用cancel()方法后，是否将该任务从队列中移除，默认false

private volatile boolean removeOnCancel = false;
复制代码

1. 任务的序列号，保证FIFO队列的顺序，用来比较优先级

private static final AtomicLong sequencer = new AtomicLong()
复制代码

1. ScheduledFutureTask延迟任务类

• ScheduledFutureTask 继承 FutureTask，实现 RunnableScheduledFuture 接口，无论是 runnable 还是 callable，无论是否需要延迟和定时，所有的任务都会被封装成 ScheduledFutureTask。
• 该类具有延迟执行的特点, 覆盖 FutureTask 的 run 方法来实现对延时执行、周期执行的支持。
• 对于延时任务调用 FutureTask#run，而对于周期性任务则调用 FutureTask#runAndReset 并且在成功之后根据 fixed-delay/fixed-rate 模式来设置下次执行时间并重新将任务塞到工作队列。
• 成员属性如下：

// 任务序列号
private final long sequenceNumber;
// 任务可以被执行的时间，交付时间，以纳秒表示
private long time;	
// 0 表示非周期任务
// 正数表示 fixed-rate（两次开始启动的间隔）模式的周期，
// 负数表示 fixed-delay（一次执行结束到下一次开始启动） 模式
private final long period;	
// 执行的任务对象
RunnableScheduledFuture<V> outerTask = this;
// 任务在队列数组中的索引下标， -1表示删除
int heapIndex;
复制代码

1. DelayedWorkQueue延迟队列

• DelayedWorkQueue 是支持延时获取元素的阻塞队列， 内部采用优先队列 PriorityQueue（小根堆、满二叉树）存储元素。
• 内部数据结构是数组，所以延迟队列出队头元素后需要让其他元素（尾）替换到头节点，防止空指针异常。
• 成员属性如下：

// 初始容量
private static final int INIT