Java线程池详解

一、线程池初探

所谓线程池,就是将多个线程放在一个池子里面(所谓池化技术),然后需要线程的时候不是创建一个线程,而是从线程池里面获取一个可用的线程,然后执行我们的任务。线程池的关键在于它为我们管理了多个线程,我们不需要关心如何创建线程,我们只需要关系我们的核心业务,然后需要线程来执行任务的时候从线程池中获取线程。任务执行完之后线程不会被销毁,而是会被重新放到池子里面,等待机会去执行任务。

我们为什么需要线程池呢?首先一点是线程池为我们提高了一种简易的多线程编程方案,我们不需要投入太多的精力去管理多个线程,线程池会自动帮我们管理好,它知道什么时候该做什么事情,我们只要在需要的时候去获取就可以了。其次,我们使用线程池很大程度上归咎于创建和销毁线程的代价是非常昂贵的,甚至我们创建和销毁线程的资源要比我们实际执行的任务所花费的时间还要长,这显然是不科学也是不合理的,而且如果没有一个合理的管理者,可能会出现创建了过多的线程的情况,也就是在JVM中存活的线程过多,而存活着的线程也是需要销毁资源的,另外一点,过多的线程可能会造成线程过度切换的尴尬境地。

对线程池有了一个初步的认识之后,我们来看看如何使用线程池。

  1. 创建一个线程池
    1 ExecutorService executorService = Executors.newFixedThreadPool(1);
  1. 提交任务
    1 executorService.submit(() -> System.out.println("run"));
    2 Future<String> stringFuture = executorService.submit(() -> "run");
  1. 创建一个调度线程池
    ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(1);
  1. 提交一个周期性执行的任务
    scheduledExecutorService.scheduleAtFixedRate(() -> System.out.println("schedule"), 0, 1, TimeUnit.SECONDS);
  1. shutdown
    1 executorService.shutdownNow();
    2 scheduledExecutorService.shutdownNow();

可以发现使用线程池非常简单,只需要极少的代码就可以创建出我们需要的线程池,然后将我们的任务提交到线程池中去。我们只需要在结束之时记得关闭线程池就可以了。本文的重点并非在于如何使用线程池,而是试图剖析线程池的实现,比如一个调度线程池是怎么实现的?是靠什么实现的?为什么能这样实现等等问题。

二、Java线程池实现架构

Java中与线程池相关的类有下面一些:

  • Executor
  • ExecutorService
  • ScheduledExecutorService
  • ThreadPoolExecutor
  • ScheduledThreadPoolExecutor
  • Executors

通过上面一节中的使用示例,可以发现Executors类是一个创建线程池的有用的类,事实上,Executors类的角色也就是创建线程池,它是一个工厂类,可以产生不同类型的线程池,而Executor是线程池的鼻祖类,它有两个子类是ExecutorService和ScheduledExecutorService,而ThreadPoolExecutor和ScheduledThreadPoolExecutor则是真正的线程池,我们的任务将被这两个类交由其所管理者的线程池运行,可以发现,ScheduledThreadPoolExecutor是一个集大成者类,下面我们可以看看它的类关系图:

ScheduledThreadPoolExecutor的类关系图
ScheduledThreadPoolExecutor的类关系图

ScheduledThreadPoolExecutor继承了ThreadPoolExecutor,ThreadPoolExecutor实现了一般的线程池,没有调度功能,而ScheduledThreadPoolExecutor继承了ThreadPoolExecutor的实现,然后增加了调度功能。

最为原始的Executor只有一个方法execute,它接受一个Runnable类型的参数,意思是使用线程池来执行这个Runnable,可以发现Executor不提供有返回值的任务。ExecutorService继承了Executor,并且极大的增强了Executor的功能,不仅支持有返回值的任务执行,而且还有很多十分有用的方法来为你提供服务,下面展示了ExecutorService提供的方法:

ExecutorService提供的方法
ExecutorService提供的方法

ScheduledExecutorService继承了ExecutorService,并且增加了特有的调度(schedule)功能。关于Executor、ExecutorService和ScheduledExecutorService的关系,可以见下图:

Executor、ExecutorService和ScheduledExecutorService的关系
Executor、ExecutorService和ScheduledExecutorService的关系

总结一下,经过我们的调研,可以发现其实对于我们编写多线程代码来说,最为核心的是Executors类,根据我们是需要ExecutorService类型的线程池还是ScheduledExecutorService类型的线程池调用相应的工厂方法就可以了,而ExecutorService的实现表现在ThreadPoolExecutor上,ScheduledExecutorService的实现则表现在ScheduledThreadPoolExecutor上,下文将分别剖析这两者,尝试弄清楚线程池的原理。

三、ThreadPoolExecutor解析

上文中描述了Java中线程池相关的架构,了解了这些内容其实我们就可以使用java的线程池为我们工作了,使用其提供的线程池我们可以很方便的写出高质量的多线程代码,本节将分析ThreadPoolExecutor的实现,来探索线程池的运行原理。下面的图片展示了ThreadPoolExecutor的类图:

ThreadPoolExecutor的类图
ThreadPoolExecutor的类图

下面是几个比较关键的类成员:

 1 private final BlockingQueue<Runnable> workQueue;  // 任务队列,我们的任务会添加到该队列里面,线程将从该队列获取任务来执行
 2  
 3  private final HashSet<Worker> workers = new HashSet<Worker>();//任务的执行值集合,来消费workQueue里面的任务
 4   
 5  private volatile ThreadFactory threadFactory;//线程工厂
 6       
 7  private volatile RejectedExecutionHandler handler;//拒绝策略,默认会抛出异异常,还要其他几种拒绝策略如下:
 8   
 9  1、CallerRunsPolicy:在调用者线程里面运行该任务
10  2、DiscardPolicy:丢弃任务
11  3、DiscardOldestPolicy:丢弃workQueue的头部任务
12   
13 private volatile int corePoolSize;//最下保活work数量
14  
15 private volatile int maximumPoolSize;//work上限

我们尝试执行submit方法,下面是执行的关键路径,总结起来就是:如果Worker数量还没达到上限则继续创建,否则提交任务到workQueue,然后让worker来调度运行任务。

 1 step 1: <ExecutorService>
 2 Future<?> submit(Runnable task); 
 3  
 4 step 2:<AbstractExecutorService>
 5     public Future<?> submit(Runnable task) {
 6     if (task == null) throw new NullPointerException();
 7     RunnableFuture<Void> ftask = newTaskFor(task, null);
 8     execute(ftask);
 9     return ftask;
10 }
11  
12 step 3:<Executor>
13 void execute(Runnable command);
14  
15 step 4:<ThreadPoolExecutor>
16  public void execute(Runnable command) {
17     if (command == null)
18         throw new NullPointerException();
19     /*
20      * Proceed in 3 steps:
21      *
22      * 1. If fewer than corePoolSize threads are running, try to
23      * start a new thread with the given command as its first
24      * task.  The call to addWorker atomically checks runState and
25      * workerCount, and so prevents false alarms that would add
26      * threads when it shouldn't, by returning false.
27      *
28      * 2. If a task can be successfully queued, then we still need
29      * to double-check whether we should have added a thread
30      * (because existing ones died since last checking) or that
31      * the pool shut down since entry into this method. So we
32      * recheck state and if necessary roll back the enqueuing if
33      * stopped, or start a new thread if there are none.
34      *
35      * 3. If we cannot queue task, then we try to add a new
36      * thread.  If it fails, we know we are shut down or saturated
37      * and so reject the task.
38      */
39     int c = ctl.get();
40     if (workerCountOf(c) < corePoolSize) {
41         if (addWorker(command, true))
42             return;
43         c = ctl.get();
44     }
45     if (isRunning(c) && workQueue.offer(command)) { //提交我们的额任务到workQueue
46         int recheck = ctl.get();
47         if (! isRunning(recheck) && remove(command))
48             reject(command);
49         else if (workerCountOf(recheck) == 0)
50             addWorker(null, false);
51     }
52     else if (!addWorker(command, false)) //使用maximumPoolSize作为边界
53         reject(command); //还不行?拒绝提交的任务
54 }
55  
56 step 5:<ThreadPoolExecutor>
57 private boolean addWorker(Runnable firstTask, boolean core)
58  
59  
60 step 6:<ThreadPoolExecutor>
61 w = new Worker(firstTask); //包装任务
62 final Thread t = w.thread; //获取线程(包含任务)
63 workers.add(w);   // 任务被放到works中
64 t.start(); //执行任务

上面的流程是高度概括的,实际情况远比这复杂得多,但是我们关心的是怎么打通整个流程,所以这样分析问题是没有太大的问题的。观察上面的流程,我们发现其实关键的地方在于Worker,如果弄明白它是如何工作的,那么我们也就大概明白了线程池是怎么工作的了。下面分析一下Worker类。

worker类图
worker类图

上面的图片展示了Worker的类关系图,关键在于他实现了Runnable接口,所以问题的关键就在于run方法上。在这之前,我们来看一下Worker类里面的关键成员:

1 final Thread thread;
2 Runnable firstTask; //我们提交的任务,可能被立刻执行,也可能被放到队列里面

thread是Worker的工作线程,上面的分析我们也发现了在addWorker中会获取worker里面的thread然后start,也就是这个线程的执行,而Worker实现了Runnable接口,所以在构造thread的时候Worker将自己传递给了构造函数,thread.start执行的其实就是Worker的run方法。下面是run方法的内容:

 1 public void run() {
 2         runWorker(this);
 3     }
 4      
 5     final void runWorker(Worker w) {
 6     Thread wt = Thread.currentThread();
 7     Runnable task = w.firstTask;
 8     w.firstTask = null;
 9     w.unlock(); // allow interrupts
10     boolean completedAbruptly = true;
11     try {
12         while (task != null || (task = getTask()) != null) {
13             w.lock();
14             // If pool is stopping, ensure thread is interrupted;
15             // if not, ensure thread is not interrupted.  This
16             // requires a recheck in second case to deal with
17             // shutdownNow race while clearing interrupt
18             if ((runStateAtLeast(ctl.get(), STOP) ||
19                  (Thread.interrupted() &&
20                   runStateAtLeast(ctl.get(), STOP))) &&
21                 !wt.isInterrupted())
22                 wt.interrupt();
23             try {
24                 beforeExecute(wt, task);
25                 Throwable thrown = null;
26                 try {
27                     task.run();
28                 } catch (RuntimeException x) {
29                     thrown = x; throw x;
30                 } catch (Error x) {
31                     thrown = x; throw x;
32                 } catch (Throwable x) {
33                     thrown = x; throw new Error(x);
34                 } finally {
35                     afterExecute(task, thrown);
36                 }
37             } finally {
38                 task = null;
39                 w.completedTasks++;
40                 w.unlock();
41             }
42         }
43         completedAbruptly = false;
44     } finally {
45         processWorkerExit(w, completedAbruptly);
46     }
47 }

我们来分析一下runWorker这个方法,这就是整个线程池的核心。首先获取到了我们刚提交的任务firstTask,然后会循环从workQueue里面获取任务来执行,获取任务的方法如下:

 1 private Runnable getTask() {
 2        boolean timedOut = false; // Did the last poll() time out?
 3  
 4        for (;;) {
 5            int c = ctl.get();
 6            int rs = runStateOf(c);
 7  
 8            // Check if queue empty only if necessary.
 9            if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
10                decrementWorkerCount();
11                return null;
12            }
13  
14            int wc = workerCountOf(c);
15  
16            // Are workers subject to culling?
17            boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;
18  
19            if ((wc > maximumPoolSize || (timed && timedOut))
20                && (wc > 1 || workQueue.isEmpty())) {
21                if (compareAndDecrementWorkerCount(c))
22                    return null;
23                continue;
24            }
25  
26            try {
27                Runnable r = timed ?
28                    workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
29                    workQueue.take();
30                if (r != null)
31                    return r;
32                timedOut = true;
33            } catch (InterruptedException retry) {
34                timedOut = false;
35            }
36        }
37    }

其实核心也就一句:

1 Runnable r = timed ? workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) : workQueue.take();

我们再回头看一下execute,其实我们上面只走了一条逻辑,在execute的时候,我们的worker的数量还没有到达我们设定的corePoolSize的时候,会走上面我们分析的逻辑,而如果达到了我们设定的阈值之后,execute中会尝试去提交任务,如果提交成功了就结束,否则会拒绝任务的提交。我们上面还提到一个成员:maximumPoolSize,其实线程池的最大的Worker数量应该是maximumPoolSize,但是我们上面的分析是corePoolSize,这是因为我们的private boolean addWorker(Runnable firstTask, boolean core)的参数core的值来控制的,core为true则使用corePoolSize来设定边界,否则使用maximumPoolSize来设定边界。直观的解释一下,当线程池里面的Worker数量还没有到corePoolSize,那么新添加的任务会伴随着产生一个新的worker,如果Worker的数量达到了corePoolSize,那么就将任务存放在阻塞队列中等待Worker来获取执行,如果没有办法再向阻塞队列放任务了,那么这个时候maximumPoolSize就变得有用了,新的任务将会伴随着产生一个新的Worker,如果线程池里面的Worker已经达到了maximumPoolSize,那么接下来提交的任务只能被拒绝策略拒绝了。可以参考下面的描述来理解:

 1 * When a new task is submitted in method {@link #execute(Runnable)},
 2 * and fewer than corePoolSize threads are running, a new thread is
 3 * created to handle the request, even if other worker threads are
 4 * idle.  If there are more than corePoolSize but less than
 5 * maximumPoolSize threads running, a new thread will be created only
 6 * if the queue is full.  By setting corePoolSize and maximumPoolSize
 7 * the same, you create a fixed-size thread pool. By setting
 8 * maximumPoolSize to an essentially unbounded value such as {@code
 9 * Integer.MAX_VALUE}, you allow the pool to accommodate an arbitrary
10 * number of concurrent tasks. Most typically, core and maximum pool
11 * sizes are set only upon construction, but they may also be changed
12 * dynamically using {@link #setCorePoolSize} and {@link
13 * #setMaximumPoolSize}.

在此需要说明一点,有一个重要的成员:keepAliveTime,当线程池里面的线程数量超过corePoolSize了,那么超出的线程将会在空闲keepAliveTime之后被terminated。可以参考下面的文档:

1 * If the pool currently has more than corePoolSize threads,
2 * excess threads will be terminated if they have been idle for more
3 * than the keepAliveTime (see {@link #getKeepAliveTime(TimeUnit)}).

四、ScheduledThreadPoolExecutor解析

ScheduledThreadPoolExecutor适用于延时执行,或者周期性执行的任务调度,ScheduledThreadPoolExecutor在实现上继承了ThreadPoolExecutor,所以你依然可以将ScheduledThreadPoolExecutor当成ThreadPoolExecutor来使用,但是ScheduledThreadPoolExecutor的功能要强大得多,因为ScheduledThreadPoolExecutor可以根据设定的参数来周期性调度运行,下面的图片展示了四个和周期性相关的方法:

四个Scheduled方法
四个Scheduled方法
  • 如果你想延时一段时间之后运行一个Runnable,那么使用第一个方法
  • 如果你想延时一段时间然后运行一个Callable,那么使用的第二个方法
  • 如果你想要延时一段时间,然后根据设定的参数周期执行Runnable,那么可以选择第三个和第四个方法,第三个方法和第四个方法的区别在于:第三个方法严格按照规划的时间路径来执行,比如周期为2,延时为0,那么执行的序列为0,2,4,6,8....,而第四个方法将基于上次执行时间来规划下次的执行,也就是在上次执行完成之后再次执行。比如上面的执行序列0,2,4,6,8...,如果第2秒没有被调度执行,而在第三秒的时候才被调度,那么下次执行的时间不是4,而是5,以此类推。

下面来看一下这四个方法的一些细节:

 1 public <V> ScheduledFuture<V> schedule(Callable<V> callable,
 2                                            long delay,
 3                                            TimeUnit unit) {
 4         if (callable == null || unit == null)
 5             throw new NullPointerException();
 6         RunnableScheduledFuture<V> t = decorateTask(callable,
 7             new ScheduledFutureTask<V>(callable,
 8                                        triggerTime(delay, unit)));
 9         delayedExecute(t);
10         return t;
11     }
12      
13      public ScheduledFuture<?> scheduleAtFixedRate(Runnable command,
14                                                   long initialDelay,
15                                                   long period,
16                                                   TimeUnit unit) {
17         if (command == null || unit == null)
18             throw new NullPointerException();
19         if (period <= 0)
20             throw new IllegalArgumentException();
21         ScheduledFutureTask<Void> sft =
22             new ScheduledFutureTask<Void>(command,
23                                           null,
24                                           triggerTime(initialDelay, unit),
25                                           unit.toNanos(period));
26         RunnableScheduledFuture<Void> t = decorateTask(command, sft);
27         sft.outerTask = t;
28         delayedExecute(t);
29         return t;
30     }
31      
32      
33 public ScheduledFuture<?> scheduleAtFixedRate(Runnable command,
34                                                   long initialDelay,
35                                                   long period,
36                                                   TimeUnit unit) {
37         if (command == null || unit == null)
38             throw new NullPointerException();
39         if (period <= 0)
40             throw new IllegalArgumentException();
41         ScheduledFutureTask<Void> sft =
42             new ScheduledFutureTask<Void>(command,
43                                           null,
44                                           triggerTime(initialDelay, unit),
45                                           unit.toNanos(period));
46         RunnableScheduledFuture<Void> t = decorateTask(command, sft);
47         sft.outerTask = t;
48         delayedExecute(t);
49         return t;
50     }
51      
52         public ScheduledFuture<?> scheduleWithFixedDelay(Runnable command,
53                                                      long initialDelay,
54                                                      long delay,
55                                                      TimeUnit unit) {
56         if (command == null || unit == null)
57             throw new NullPointerException();
58         if (delay <= 0)
59             throw new IllegalArgumentException();
60         ScheduledFutureTask<Void> sft =
61             new ScheduledFutureTask<Void>(command,
62                                           null,
63                                           triggerTime(initialDelay, unit),
64                                           unit.toNanos(-delay));
65         RunnableScheduledFuture<Void> t = decorateTask(command, sft);
66         sft.outerTask = t;
67         delayedExecute(t);
68         return t;
69     }

通过上面的代码我们可以发现,前两个方法是类似的,后两个方法也是类似的。前两个方法属于一次性调度,所以period都为0,区别在于参数不同,一个是Runnable,而一个是Callable,可笑的是,最后都变为了Callable了,见下面的构造函数:

1 public FutureTask(Runnable runnable, V result) {
2     this.callable = Executors.callable(runnable, result);
3     this.state = NEW;       // ensure visibility of callable
4 }

 

对于后两个方法,区别仅仅在于period的,scheduleWithFixedDelay对参数进行了操作,将原来的时间变为负数了,而后面在计算下次被调度的时间的时候会根据这个参数的正负值来分别处理,正数代表scheduleAtFixedRate,而负数代表了scheduleWithFixedDelay。

一个需要被我们注意的细节是,以上四个方法最后都会调用一个方法: delayedExecute(t),下面看一下这个方法:

 1 private void delayedExecute(RunnableScheduledFuture<?> task) {
 2     if (isShutdown())
 3         reject(task);
 4     else {
 5         super.getQueue().add(task);
 6         if (isShutdown() &&
 7             !canRunInCurrentRunState(task.isPeriodic()) &&
 8             remove(task))
 9             task.cancel(false);
10         else
11             ensurePrestart();
12     }
13 }

大概的意思就是先判断线程池是否被关闭了,如果被关闭了,则拒绝任务的提交,否则将任务加入到任务队列中去等待被调度执行。最后的ensurePrestart的意思是需要确保线程池已经被启动起来了。下面是这个方法:

1 void ensurePrestart() {
2     int wc = workerCountOf(ctl.get());
3     if (wc < corePoolSize)
4         addWorker(null, true);
5     else if (wc == 0)
6         addWorker(null, false);
7 }

主要是增加了一个没有任务的worker,有什么用呢?我们还记得Worker的逻辑吗?addWorker方法的执行,会触发Worker的run方法的执行,然后runWorker方法就会被执行,而runWorker方法是循环从workQueue中取任务执行的,所以确保线程池被启动起来是重要的,而只需要简单的执行addWorker便会触发线程池的启动流程。对于调度线程池来说,只要执行了addWorker方法,那么线程池就会一直在后台周期性的调度执行任务。

到此,似乎我们还是没有闹明白ScheduledThreadPoolExecutor是如何实现周期性的,上面讲到四个scheduled方法时,我们没有提一个重要的类:ScheduledFutureTask,对,所有神奇的事情将会发生在这个类中,下面来分析一下这个类。

ScheduledFutureTask类图
ScheduledFutureTask类图

看上面的类图,貌似这个类非常复杂,还好,我们发现他实现了Runnable接口,那么必然会有一个run方法,而这个run方法必然是整个类的核心,下面来看一下这个run方法的内容:

 1 public void run() {
 2         boolean periodic = isPeriodic();
 3         if (!canRunInCurrentRunState(periodic))
 4             cancel(false);
 5         else if (!periodic)
 6             ScheduledFutureTask.super.run();
 7         else if (ScheduledFutureTask.super.runAndReset()) {
 8             setNextRunTime();
 9             reExecutePeriodic(outerTask);
10         }
11     }
12 }

首先,判断是否是周期性的任务,如果不是,则直接执行(一次性),否则执行,然后设置下次执行的时间,然后重新调度,等待下次执行。这里有一个方法需要注意,也就是setNextRunTime,上面我们提到scheduleAtFixedRate和scheduleWithFixedDelay在传递参数时不一样,后者将delay值变为了负数,所以下面的处理正好印证了前文所述。

1 private void setNextRunTime() {
2     long p = period;
3     if (p > 0)
4         time += p;
5     else
6         time = triggerTime(-p);
7 }

下面来看一下reExecutePeriodic方法是如何做的,他的目标是将任务再次被调度执行,下面的代码展示了这个功能的实现:

1 void reExecutePeriodic(RunnableScheduledFuture<?> task) {
2     if (canRunInCurrentRunState(true)) {
3         super.getQueue().add(task);
4         if (!canRunInCurrentRunState(true) && remove(task))
5             task.cancel(false);
6         else
7             ensurePrestart();
8     }
9 }

可以看到,这个方法就是将我们的任务再次放到了workQueue里面,那这个参数是什么?在上面的run方法中我们调用了reExecutePeriodic方法,参数为outerTask,而这个变量是什么?看下面的代码:

1 /** The actual task to be re-enqueued by reExecutePeriodic */
2 RunnableScheduledFuture<V> outerTask = this;

这个变量指向了自己,而this的类型是什么?是ScheduledFutureTask,也就是可以被调度的task,这样就实现了循环执行任务了。

上面的分析已经到了循环执行,但是ScheduledThreadPoolExecutor的功能是周期性执行,所以我们接着分析ScheduledThreadPoolExecutor是如何根据我们的参数走走停停的。这个时候,是应该看一下ScheduledThreadPoolExecutor的构造函数了,我们来看一个最简单的构造函数:

1 public ScheduledThreadPoolExecutor(int corePoolSize) {
2     super(corePoolSize, Integer.MAX_VALUE, 0, NANOSECONDS,
3           new DelayedWorkQueue());
4 }

我们知道ScheduledThreadPoolExecutor的父类是ThreadPoolExecutor,所以这里的super其实是ThreadPoolExecutor的构造函数,我们发现其中有一个参数DelayedWorkQueue,看名字貌似是一个延迟队列的样子,进一步跟踪代码,发现了下面的一行代码(构造函数中):

1 this.workQueue = workQueue;

所以在ScheduledThreadPoolExecutor中,workQueue是一个DelayedWorkQueue类型的队列,我们暂且认为DelayedWorkQueue是一种具备延迟功能的队列吧,那么,到此我们便可以想明白了,上面的分析我们明白了ScheduledThreadPoolExecutor是如何循环执行任务的,而这里我们明白了ScheduledThreadPoolExecutor使用DelayedWorkQueue来达到延迟的目标,所以组合起来,就可以实现ScheduledThreadPoolExecutor周期性执行的目标。下面我们来看一下DelayedWorkQueue是如何做到延迟的吧,上文中提到一个方法:getTask,这个方法的作用是从workQueue中取出任务来执行,而在ScheduledThreadPoolExecutor里面,getTask方法是从DelayedWorkQueue中取任务的,而取任务无非两个方法:poll或者take,下面我们对DelayedWorkQueue的take方法来分析一下:

 1 public RunnableScheduledFuture<?> take() throws InterruptedException {
 2            final ReentrantLock lock = this.lock;
 3            lock.lockInterruptibly();
 4            try {
 5                for (;;) {
 6                    RunnableScheduledFuture<?> first = queue[0];
 7                    if (first == null)
 8                        available.await();
 9                    else {
10                        long delay = first.getDelay(NANOSECONDS);
11                        if (delay <= 0)
12                            return finishPoll(first);
13                        first = null; // don't retain ref while waiting
14                        if (leader != null)
15                            available.await();
16                        else {
17                            Thread thisThread = Thread.currentThread();
18                            leader = thisThread;
19                            try {
20                                available.awaitNanos(delay);
21                            } finally {
22                                if (leader == thisThread)
23                                    leader = null;
24                            }
25                        }
26                    }
27                }
28            } finally {
29                if (leader == null && queue[0] != null)
30                    available.signal();
31                lock.unlock();
32            }
33        }

在for循环里面,首先从queue中获取第一个任务,然后从任务中取出延迟时间,而后使用available变量来实现延迟效果。这里面需要几个点需要探索一下:

  • 这个queue是什么东西?
  • 延迟时间的来龙去脉?
  • available变量的来龙去脉?

对于第一个问题,看下面的代码:

1 private RunnableScheduledFuture<?>[] queue = new RunnableScheduledFuture<?>[INITIAL_CAPACITY];

它是一个RunnableScheduledFuture类型的数组,下面是RunnableScheduledFuture类的类关系图:

RunnableScheduledFuture类关系
RunnableScheduledFuture类关系

数组里面保存了我们的RunnableScheduledFuture,对queue的操作,主要来看一下增加元素和消费元素的操作。首先,假设使用add方法来增加RunnableScheduledFuture到queue,调用的链路如下:

 1 public boolean add(Runnable e) {
 2     return offer(e);
 3 }
 4  
 5  
 6  public boolean offer(Runnable x) {
 7     if (x == null)
 8         throw new NullPointerException();
 9     RunnableScheduledFuture<?> e = (RunnableScheduledFuture<?>)x;
10     final ReentrantLock lock = this.lock;
11     lock.lock();
12     try {
13         int i = size;
14         if (i >= queue.length)
15             grow();
16         size = i + 1;
17         if (i == 0) {
18             queue[0] = e;
19             setIndex(e, 0);
20         } else {
21             siftUp(i, e);
22         }
23         if (queue[0] == e) {
24             leader = null;
25             available.signal();
26         }
27     } finally {
28         lock.unlock();
29     }
30     return true;
31 }

解释一下,add方法直接转到了offer方法,该方法中,首先判断数组的容量是否足够,如果不够则grow,增长的策略如下:

1 int newCapacity = oldCapacity + (oldCapacity >> 1); // grow 50%

每次增长50%,入戏下去。增长完成后,如果这是第一个元素,则放在坐标为0的位置,否则,使用siftUp操作,下面是该方法的内容:

 1 private void siftUp(int k, RunnableScheduledFuture<?> key) {
 2     while (k > 0) {
 3         int parent = (k - 1) >>> 1;
 4         RunnableScheduledFuture<?> e = queue[parent];
 5         if (key.compareTo(e) >= 0)
 6             break;
 7         queue[k] = e;
 8         setIndex(e, k);
 9         k = parent;
10     }
11     queue[k] = key;
12     setIndex(key, k);
13 }

这个数组实现了堆这种数据结构,使用对象比较将最需要被调度执行的RunnableScheduledFuture放到数组的前面,而这得力于compareTo方法,下面是RunnableScheduledFuture类的compareTo方法的实现,主要是通过延迟时间来做比较。

 1 public int compareTo(Delayed other) {
 2     if (other == this) // compare zero if same object
 3         return 0;
 4     if (other instanceof ScheduledFutureTask) {
 5         ScheduledFutureTask<?> x = (ScheduledFutureTask<?>)other;
 6         long diff = time - x.time;
 7         if (diff < 0)
 8             return -1;
 9         else if (diff > 0)
10             return 1;
11         else if (sequenceNumber < x.sequenceNumber)
12             return -1;
13         else
14             return 1;
15     }
16     long diff = getDelay(NANOSECONDS) - other.getDelay(NANOSECONDS);
17     return (diff < 0) ? -1 : (diff > 0) ? 1 : 0;
18 }

上面是生产元素,下面来看一下消费数据。在上面我们提到的take方法中,使用了一个方法如下:

1 private RunnableScheduledFuture<?> finishPoll(RunnableScheduledFuture<?> f) {
2     int s = --size;
3     RunnableScheduledFuture<?> x = queue[s];
4     queue[s] = null;
5     if (s != 0)
6         siftDown(0, x);
7     setIndex(f, -1);
8     return f;
9 }

这个方法中调用了一个方法siftDown,这个方法如下:

 1 private void siftDown(int k, RunnableScheduledFuture<?> key) {
 2     int half = size >>> 1;
 3     while (k < half) {
 4         int child = (k << 1) + 1;
 5         RunnableScheduledFuture<?> c = queue[child];
 6         int right = child + 1;
 7         if (right < size && c.compareTo(queue[right]) > 0)
 8             c = queue[child = right];
 9         if (key.compareTo(c) <= 0)
10             break;
11         queue[k] = c;
12         setIndex(c, k);
13         k = child;
14     }
15     queue[k] = key;
16     setIndex(key, k);
17 }

对其的解释就是:

1 Replaces first element with last and sifts it down.  Call only when holding lock.

总结一下,当我们向queue插入任务的时候,会发生siftUp方法的执行,这个时候会把任务尽量往根部移动,而当我们完成任务调度之后,会发生siftDown方法的执行,与siftUp相反,siftDown方法会将任务尽量移动到queue的末尾。总之,大概的意思就是queue通过compareTo实现了类似于优先级队列的功能。

下面我们来看一下第二个问题:延迟时间的来龙去脉。在上面的take方法里面,首先获取了delay,然后再使用available来做延迟效果,那这个delay从哪里来的呢?通过上面的类图RunnableScheduledFuture的类图我们知道,RunnableScheduledFuture类实现了Delayed接口,而Delayed接口里面的唯一方法是getDelay,我们到RunnableScheduledFuture里面看一下这个方法的具体实现:

1 public long getDelay(TimeUnit unit) {
2      return unit.convert(time - now(), NANOSECONDS);
3  }

time是我们设定的下次执行的时间,所以延迟就是(time - now()),没毛病!

第三个问题:available变量的来龙去脉,至于这个问题,我们看下面的代码:

1 /**
2  * Condition signalled when a newer task becomes available at the
3  * head of the queue or a new thread may need to become leader.
4  */
5 private final Condition available = lock.newCondition();

这是一个条件变量,take方法里面使用这个变量来做延迟效果。Condition可以在多个线程间做同步协调工作,更为具体细致的关于Condition的内容,可以参考更多的资料来学习,本文对此知识点点到为止。

到此为止,我们梳理了ScheduledThreadPoolExecutor是如何实现周期性调度的,首先分析了它的循环性,然后分析了它的延迟效果,本文到此也就结束了,对于线程池的学习现在才刚刚起步,需要更多更专业的知识类帮我理解更为底层的内容,当然,为了更进一步理解线程池的实现细节,首先需要对线程间通信有足够的把握,其次是要对各种数据结构有清晰的认识,比如队列、优先级队列、堆等高级的数据结构,以及java语言对于这些数据结构的实现,更为重要的是要结合实际情况分析问题,在工作和平时的学习中不断总结,不断迭代对于线程、线程池的认知。

posted @ 2019-02-23 13:01  smile_lg  阅读(299)  评论(0编辑  收藏  举报