Bootstrap

什么是线程池?

什么是线程池

线程池的概念大家应该都很清楚,帮我们重复管理线程,避免创建大量的线程增加开销。

除了降低开销以外,线程池也可以提高响应速度,了解点 JVM 的同学可能知道,一个对象的创建大概需要经过以下几步:

  1. 检查对应的类是否已经被加载、解析和初始化
  2. 类加载后,为新生对象分配内存
  3. 将分配到的内存空间初始为 0
  4. 对对象进行关键信息的设置,比如对象的哈希码等
  5. 然后执行 init 方法初始化对象

创建一个对象的开销需要经过这么多步,也是需要时间的嘛,那可以复用已经创建好的线程的线程池,自然也在提高响应速度上做了贡献。

线程池的处理流程

创建线程池需要使用 ThreadPoolExecutor 类,它的构造函数参数如下:

public ThreadPoolExecutor(int corePoolSize,    //核心线程的数量
                          int maximumPoolSize,    //最大线程数量
                          long keepAliveTime,    //超出核心线程数量以外的线程空余存活时间
                          TimeUnit unit,    //存活时间的单位
                          BlockingQueue<Runnable> workQueue,    //保存待执行任务的队列
                          ThreadFactory threadFactory,    //创建新线程使用的工厂
                          RejectedExecutionHandler handler // 当任务无法执行时的处理器
                          ) {...}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

参数介绍如注释所示,要了解这些参数左右着什么,就需要了解线程池具体的执行方法ThreadPoolExecutor.execute:

public void execute(Runnable command) {
    if (command == null)
        throw new NullPointerException();

    int c = ctl.get();
    //1.当前池中线程比核心数少,新建一个线程执行任务
    if (workerCountOf(c) < corePoolSize) {   
        if (addWorker(command, true))
            return;
        c = ctl.get();
    }
    //2.核心池已满,但任务队列未满,添加到队列中
    if (isRunning(c) && workQueue.offer(command)) {   
        int recheck = ctl.get();
        if (! isRunning(recheck) && remove(command))    //如果这时被关闭了,拒绝任务
            reject(command);
        else if (workerCountOf(recheck) == 0)    //如果之前的线程已被销毁完,新建一个线程
            addWorker(null, false);
    }
    //3.核心池已满,队列已满,试着创建一个新线程
    else if (!addWorker(command, false))
        reject(command);    //如果创建新线程失败了,说明线程池被关闭或者线程池完全满了,拒绝任务
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23

可以看到,线程池处理一个任务主要分三步处理,代码注释里已经介绍了,我再用通俗易懂的例子解释一下:

(线程比作员工,线程池比作一个团队,核心池比作团队中核心团队员工数,核心池外的比作外包员工)

  1. 有了新需求,先看核心员工数量超没超出最大核心员工数,还有名额的话就新招一个核心员工来做 
    • 需要获取全局锁
  2. 核心员工已经最多了,HR 不给批 HC 了,那这个需求只好攒着,放到待完成任务列表吧
  3. 如果列表已经堆满了,核心员工基本没机会搞完这么多任务了,那就找个外包吧 
    • 需要获取全局锁
  4. 如果核心员工 + 外包员工的数量已经是团队最多能承受人数了,没办法,这个需求接不了了

结合这张图,这回流程你明白了吗?

这里写图片描述

由于 1 和 3 新建线程时需要获取全局锁,这将严重影响性能。因此 ThreadPoolExecutor 这样的处理流程是为了在执行 execute() 方法时尽量少地执行 1 和 3,多执行 2。

在 ThreadPoolExecutor 完成预热后(当前线程数不少于核心线程数),几乎所有的 execute() 都是在执行步骤 2。

前面提到的 ThreadPoolExecutor 构造函数的参数,分别影响以下内容:

  • corePoolSize:核心线程池数量 
    • 在线程数少于核心数量时,有新任务进来就新建一个线程,即使有的线程没事干
    • 等超出核心数量后,就不会新建线程了,空闲的线程就得去任务队列里取任务执行了
  • maximumPoolSize:最大线程数量 
    • 包括核心线程池数量 + 核心以外的数量
    • 如果任务队列满了,并且池中线程数小于最大线程数,会再创建新的线程执行任务
  • keepAliveTime:核心池以外的线程存活时间,即没有任务的外包的存活时间 
    • 如果给线程池设置 allowCoreThreadTimeOut(true),则核心线程在空闲时头上也会响起死亡的倒计时
    • 如果任务是多而容易执行的,可以调大这个参数,那样线程就可以在存活的时间里有更大可能接受新任务
  • workQueue:保存待执行任务的阻塞队列 
    • 不同的任务类型有不同的选择,下一小节介绍
  • threadFactory:每个线程创建的地方 
    • 可以给线程起个好听的名字,设置个优先级啥的
  • handler:饱和策略,大家都很忙,咋办呢,有四种策略 
    • CallerRunsPolicy:只要线程池没关闭,就直接用调用者所在线程来运行任务
    • AbortPolicy:直接抛出 RejectedExecutionException 异常
    • DiscardPolicy:悄悄把任务放生,不做了
    • DiscardOldestPolicy:把队列里待最久的那个任务扔了,然后再调用 execute() 试试看能行不
    • 我们也可以实现自己的 RejectedExecutionHandler 接口自定义策略,比如如记录日志什么的

保存待执行任务的阻塞队列

当线程池中的核心线程数已满时,任务就要保存到队列中了。

线程池中使用的队列是 BlockingQueue 接口,常用的实现有如下几种:

  • ArrayBlockingQueue:基于数组、有界,按 FIFO(先进先出)原则对元素进行排序
  • LinkedBlockingQueue:基于链表,按FIFO (先进先出) 排序元素 
    • 吞吐量通常要高于 ArrayBlockingQueue
    • Executors.newFixedThreadPool() 使用了这个队列
  • SynchronousQueue:不存储元素的阻塞队列 
    • 每个插入操作必须等到另一个线程调用移除操作,否则插入操作一直处于阻塞状态
    • 吞吐量通常要高于 LinkedBlockingQueue
    • Executors.newCachedThreadPool使用了这个队列
  • PriorityBlockingQueue:具有优先级的、无限阻塞队列

关于阻塞队列的详细介绍请看这篇:

创建自己的线程池

了解上面的内容后,我们就可以创建自己的线程池了。

①先定义线程池的几个关键属性的值:

private static final int CORE_POOL_SIZE = Runtime.getRuntime().availableProcessors() * 2; // 核心线程数为 CPU 数*2
private static final int MAXIMUM_POOL_SIZE = 64;    // 线程池最大线程数
private static final int KEEP_ALIVE_TIME = 1;    // 保持存活时间 1秒
  • 1
  • 2
  • 3
  • 1
  • 2
  • 3
  • 设置核心池的数量为 CPU 数的两倍,一般是 4、8,好点的 16 个线程
  • 最大线程数设置为 64
  • 空闲线程的存活时间设置为 1 秒

②然后根据处理的任务类型选择不同的阻塞队列

如果是要求高吞吐量的,可以使用 SynchronousQueue 队列;如果对执行顺序有要求,可以使用 PriorityBlockingQueue;如果最大积攒的待做任务有上限,可以使用 LinkedBlockingQueue

private final BlockingQueue<Runnable> mWorkQueue = new LinkedBlockingQueue<>(128);
  • 1
  • 1

③然后创建自己的 ThreadFactory

在其中为每个线程设置个名称:

private final ThreadFactory DEFAULT_THREAD_FACTORY = new ThreadFactory() {
    private final AtomicInteger mCount = new AtomicInteger(1);

    public Thread newThread(Runnable r) {
        Thread thread = new Thread(r, TAG + " #" + mCount.getAndIncrement());
        thread.setPriority(Thread.NORM_PRIORITY);
        return thread;
    }
};
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9

④然后就可以创建线程池了

private ThreadPoolExecutor mExecutor = new ThreadPoolExecutor(CORE_POOL_SIZE, MAXIMUM_POOL_SIZE, KEEP_ALIVE_TIME,
        TimeUnit.SECONDS, mWorkQueue, DEFAULT_THREAD_FACTORY,
        new ThreadPoolExecutor.DiscardOldestPolicy());
  • 1
  • 2
  • 3
  • 1
  • 2
  • 3

这里我们选择的饱和策略为 DiscardOldestPolicy,你可以可以创建自己的。

⑤完整代码:

public class ThreadPoolManager {
    private final String TAG = this.getClass().getSimpleName();
    private static final int CORE_POOL_SIZE = Runtime.getRuntime().availableProcessors() * 2; // 核心线程数为 CPU数*2
    private static final int MAXIMUM_POOL_SIZE = 64;    // 线程队列最大线程数
    private static final int KEEP_ALIVE_TIME = 1;    // 保持存活时间 1秒

    private final BlockingQueue<Runnable> mWorkQueue = new LinkedBlockingQueue<>(128);

    private final ThreadFactory DEFAULT_THREAD_FACTORY = new ThreadFactory() {
        private final AtomicInteger mCount = new AtomicInteger(1);

        public Thread newThread(Runnable r) {
            Thread thread = new Thread(r, TAG + " #" + mCount.getAndIncrement());
            thread.setPriority(Thread.NORM_PRIORITY);
            return thread;
        }
    };

    private ThreadPoolExecutor mExecutor = new ThreadPoolExecutor(CORE_POOL_SIZE, MAXIMUM_POOL_SIZE, KEEP_ALIVE_TIME,
            TimeUnit.SECONDS, mWorkQueue, DEFAULT_THREAD_FACTORY,
            new ThreadPoolExecutor.DiscardOldestPolicy());

    private static volatile ThreadPoolManager mInstance = new ThreadPoolManager();

    public static ThreadPoolManager getInstance() {
        return mInstance;
    }

    public void addTask(Runnable runnable) {
        mExecutor.execute(runnable);
    }

    @Deprecated
    public void shutdownNow() {
        mExecutor.shutdownNow();
    }
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37

这样我们就有了自己的线程池。

JDK 提供的线程池及使用场景

JDK 为我们内置了五种常见线程池的实现,均可以使用 Executors 工厂类创建。

1.newFixedThreadPool

public static ExecutorService newFixedThreadPool(int nThreads) {
    return new ThreadPoolExecutor(nThreads, nThreads,
                                  0L, TimeUnit.MILLISECONDS,
                                  new LinkedBlockingQueue<Runnable>());
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 1
  • 2
  • 3
  • 4
  • 5

不招外包,有固定数量核心成员的正常互联网团队。

可以看到,FixedThreadPool 的核心线程数和最大线程数都是指定值,也就是说当线程池中的线程数超过核心线程数后,任务都会被放到阻塞队列中。

此外 keepAliveTime 为 0,也就是多余的空余线程会被立即终止(由于这里没有多余线程,这个参数也没什么意义了)。

而这里选用的阻塞队列是 LinkedBlockingQueue,使用的是默认容量 Integer.MAX_VALUE,相当于没有上限。

因此这个线程池执行任务的流程如下:

  1. 线程数少于核心线程数,也就是设置的线程数时,新建线程执行任务
  2. 线程数等于核心线程数后,将任务加入阻塞队列 
    • 由于队列容量非常大,可以一直加加加
  3. 执行完任务的线程反复去队列中取任务执行

FixedThreadPool 用于负载比较重的服务器,为了资源的合理利用,需要限制当前线程数量。

2.newSingleThreadExecutor

public static ExecutorService newSingleThreadExecutor() {
    return new FinalizableDelegatedExecutorService
        (new ThreadPoolExecutor(1, 1,
                                0L, TimeUnit.MILLISECONDS,
                                new LinkedBlockingQueue<Runnable>()));
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

不招外包,只有一个核心成员的创业团队。

从参数可以看出来,SingleThreadExecutor 相当于特殊的 FixedThreadPool,它的执行流程如下:

  1. 线程池中没有线程时,新建一个线程执行任务
  2. 有一个线程以后,将任务加入阻塞队列,不停加加加
  3. 唯一的这一个线程不停地去队列里取任务执行

听起来很可怜的样子 - -。

SingleThreadExecutor 用于串行执行任务的场景,每个任务必须按顺序执行,不需要并发执行。

3.newCachedThreadPool

public static ExecutorService newCachedThreadPool() {
    return new ThreadPoolExecutor(0, Integer.MAX_VALUE,
                                  60L, TimeUnit.SECONDS,
                                  new SynchronousQueue<Runnable>());
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 1
  • 2
  • 3
  • 4
  • 5

全部外包,没活最多待 60 秒的外包团队。

可以看到,CachedThreadPool 没有核心线程,非核心线程数无上限,也就是全部使用外包,但是每个外包空闲的时间只有 60 秒,超过后就会被回收。

CachedThreadPool 使用的队列是 SynchronousQueue,这个队列的作用就是传递任务,并不会保存。

因此当提交任务的速度大于处理任务的速度时,每次提交一个任务,就会创建一个线程。极端情况下会创建过多的线程,耗尽 CPU 和内存资源。

它的执行流程如下:

  1. 没有核心线程,直接向 SynchronousQueue 中提交任务
  2. 如果有空闲线程,就去取出任务执行;如果没有空闲线程,就新建一个
  3. 执行完任务的线程有 60 秒生存时间,如果在这个时间内可以接到新任务,就可以继续活下去,否则就拜拜

由于空闲 60 秒的线程会被终止,长时间保持空闲的 CachedThreadPool 不会占用任何资源。

CachedThreadPool 用于并发执行大量短期的小任务,或者是负载较轻的服务器。

4.newScheduledThreadPool

public static ScheduledExecutorService newScheduledThreadPool(int corePoolSize) {
    return new ScheduledThreadPoolExecutor(corePoolSize);
}
public ScheduledThreadPoolExecutor(int corePoolSize) {
    super(corePoolSize, Integer.MAX_VALUE,
          DEFAULT_KEEPALIVE_MILLIS, MILLISECONDS,
          new DelayedWorkQueue());
}
private static final long DEFAULT_KEEPALIVE_MILLIS = 10L;
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9

定期维护的 2B 业务团队,核心与外包成员都有。

ScheduledThreadPoolExecutor 继承自 ThreadPoolExecutor, 最多线程数为 Integer.MAX_VALUE ,使用 DelayedWorkQueue 作为任务队列。

ScheduledThreadPoolExecutor 添加任务和执行任务的机制与ThreadPoolExecutor 有所不同。

ScheduledThreadPoolExecutor 添加任务提供了另外两个方法:

  • scheduleAtFixedRate() :按某种速率周期执行
  • scheduleWithFixedDelay():在某个延迟后执行

它俩的代码如下:

public ScheduledFuture<?> scheduleAtFixedRate(Runnable command,
                                              long initialDelay,
                                              long period,
                                              TimeUnit unit) {
    if (command == null || unit == null)
        throw new NullPointerException();
    if (period <= 0L)
        throw new IllegalArgumentException();
    ScheduledFutureTask<Void> sft =
        new ScheduledFutureTask<Void>(command,
                                      null,
                                      triggerTime(initialDelay, unit),
                                      unit.toNanos(period),
                                      sequencer.getAndIncrement());
    RunnableScheduledFuture<Void> t = decorateTask(command, sft);
    sft.outerTask = t;
    delayedExecute(t);
    return t;
}
public ScheduledFuture<?> scheduleWithFixedDelay(Runnable command,
                                                 long initialDelay,
                                                 long delay,
                                                 TimeUnit unit) {
    if (command == null || unit == null)
        throw new NullPointerException();
    if (delay <= 0L)
        throw new IllegalArgumentException();
    ScheduledFutureTask<Void> sft =
        new ScheduledFutureTask<Void>(command,
                                      null,
                                      triggerTime(initialDelay, unit),
                                      -unit.toNanos(delay),
                                      sequencer.getAndIncrement());
    RunnableScheduledFuture<Void> t = decorateTask(command, sft);
    sft.outerTask = t;
    delayedExecute(t);
    return t;
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38

可以看到,这两种方法都是创建了一个 ScheduledFutureTask 对象,调用 decorateTask() 方法转成 RunnableScheduledFuture 对象,然后添加到队列中。

看下 ScheduledFutureTask 的主要属性:

private class ScheduledFutureTask<V>
        extends FutureTask<V> implements RunnableScheduledFuture<V> {

    //添加到队列中的顺序
    private final long sequenceNumber;
    //何时执行这个任务
    private volatile long time;
    //执行的间隔周期
    private final long period;
    //实际被添加到队列中的 task
    RunnableScheduledFuture<V> outerTask = this;
    //在 delay queue 中的索引,便于取消时快速查找
    int heapIndex;
    //...
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15

DelayQueue 中封装了一个优先级队列,这个队列会对队列中的 ScheduledFutureTask 进行排序,两个任务的执行 time 不同时,time 小的先执行;否则比较添加到队列中的顺序 sequenceNumber ,先提交的先执行。

ScheduledThreadPoolExecutor 的执行流程如下:

  1. 调用上面两个方法添加一个任务
  2. 线程池中的线程从 DelayQueue 中取任务
  3. 然后执行任务

具体执行任务的步骤也比较复杂:

  1. 线程从 DelayQueue 中获取 time 大于等于当前时间的 ScheduledFutureTask 
    • DelayQueue.take()
  2. 执行完后修改这个 task 的 time 为下次被执行的时间
  3. 然后再把这个 task 放回队列中 
    • DelayQueue.add()

ScheduledThreadPoolExecutor 用于需要多个后台线程执行周期任务,同时需要限制线程数量的场景。

两种提交任务的方法

ExecutorService 提供了两种提交任务的方法:

  1. execute():提交不需要返回值的任务
  2. submit():提交需要返回值的任务

execute

void execute(Runnable command);
  • 1
  • 1

execute() 的参数是一个 Runnable,也没有返回值。因此提交后无法判断该任务是否被线程池执行成功。

ExecutorService executor = Executors.newCachedThreadPool();
executor.execute(new Runnable() {
    @Override
    public void run() {
        //do something
    }
});
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

submit

<T> Future<T> submit(Callable<T> task);
<T> Future<T> submit(Runnable task, T result);
Future<?> submit(Runnable task);
  • 1
  • 2
  • 3
  • 1
  • 2
  • 3

submit() 有三种重载,参数可以是 Callable 也可以是 Runnable

同时它会返回一个 Funture 对象,通过它我们可以判断任务是否执行成功。

获得执行结果调用 Future.get() 方法,这个方法会阻塞当前线程直到任务完成。

提交一个 Callable 任务时,需要使用 FutureTask 包一层:

FutureTask futureTask = new FutureTask(new Callable<String>() {    //创建 Callable 任务
    @Override
    public String call() throws Exception {
        String result = "";
        //do something
        return result;
    }
});
Future<?> submit = executor.submit(futureTask);    //提交到线程池
try {
    Object result = submit.get();    //获取结果
} catch (InterruptedException e) {
    e.printStackTrace();
} catch (ExecutionException e) {
    e.printStackTrace();
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16

关闭线程池

线程池即使不执行任务也会占用一些资源,所以在我们要退出任务时最好关闭线程池。

有两个方法关闭线程池:

  1. `shutdown() 
    • 将线程池的状态设置为 SHUTDOWN,然后中断所有没有正在执行的线程
  2. shutdownNow() 
    • 将线程池设置为 STOP,然后尝试停止所有线程,并返回等待执行任务的列表

它们的共同点是:都是通过遍历线程池中的工作线程,逐个调用 Thread.interrup() 来中断线程,所以一些无法响应中断的任务可能永远无法停止(比如 Runnable)。

如何合理地选择或者配置

了解 JDK 提供的几种线程池实现,在实际开发中如何选择呢?

根据任务类型决定。 
前面已经介绍了,这里再小节一下:

  • CachedThreadPool 用于并发执行大量短期的小任务,或者是负载较轻的服务器。
  • FixedThreadPool 用于负载比较重的服务器,为了资源的合理利用,需要限制当前线程数量。
  • SingleThreadExecutor 用于串行执行任务的场景,每个任务必须按顺序执行,不需要并发执行。
  • ScheduledThreadPoolExecutor 用于需要多个后台线程执行周期任务,同时需要限制线程数量的场景。

自定义线程池时,如果任务是 CPU 密集型(需要进行大量计算、处理),则应该配置尽量少的线程,比如 CPU 个数 + 1,这样可以避免出现每个线程都需要使用很长时间但是有太多线程争抢资源的情况; 
如果任务是 IO密集型(主要时间都在 I/O,CPU 空闲时间比较多),则应该配置多一些线程,比如 CPU 数的两倍,这样可以更高地压榨 CPU。

为了错误避免创建过多线程导致系统奔溃,建议使用有界队列。因为它在无法添加更多任务时会拒绝任务,这样可以提前预警,避免影响整个系统。

执行时间、顺序有要求的话可以选择优先级队列,同时也要保证低优先级的任务有机会被执行。

总结

这篇文章简单介绍了 Java 中线程池的工作原理和一些常见线程池的使用,在实际开发中最好使用线程池来统一管理异步任务,而不是直接 new 一个线程执行任务。


深入理解线程池及其原理


我们使用线程的时候就去创建一个线程,这样实现起来非常简便,但是就会有一个问题:

如果并发的线程数量很多,并且每个线程都是执行一个时间很短的任务就结束了,这样频繁创建线程就会大大降低系统的效率,因为频繁创建线程和销毁线程需要时间。

那么有没有一种办法使得线程可以复用,就是执行完一个任务,并不被销毁,而是可以继续执行其他的任务?

在Java中可以通过线程池来达到这样的效果。今天我们就来详细讲解一下Java的线程池,首先我们从最核心的ThreadPoolExecutor类中的方法讲起,然后再讲述它的实现原理,接着给出了它的使用示例,最后讨论了一下如何合理配置线程池的大小。

以下是本文的目录大纲:

  • 一.Java中的ThreadPoolExecutor类
  • 二.深入剖析线程池实现原理
  • 三.使用示例
  • 四.如何合理配置线程池的大小 

若有不正之处请多多谅解,并欢迎批评指正。

一.Java中的ThreadPoolExecutor类

java.uitl.concurrent.ThreadPoolExecutor类是线程池中最核心的一个类,因此如果要透彻地了解Java中的线程池,必须先了解这个类。下面我们来看一下ThreadPoolExecutor类的具体实现源码。

在ThreadPoolExecutor类中提供了四个构造方法:

[java] view plain copy
  1. public class ThreadPoolExecutor extends AbstractExecutorService {  
  2.     .....  
  3.     public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,  
  4.             BlockingQueue<Runnable> workQueue);  
  5.   
  6.     public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,  
  7.             BlockingQueue<Runnable> workQueue,ThreadFactory threadFactory);  
  8.   
  9.     public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,  
  10.             BlockingQueue<Runnable> workQueue,RejectedExecutionHandler handler);  
  11.   
  12.     public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,  
  13.         BlockingQueue<Runnable> workQueue,ThreadFactory threadFactory,RejectedExecutionHandler handler);  
  14.     ...  
  15. }  

从上面的代码可以得知,ThreadPoolExecutor继承了AbstractExecutorService类,并提供了四个构造器,事实上,通过观察每个构造器的源码具体实现,发现前面三个构造器都是调用的第四个构造器进行的初始化工作。

下面解释下一下构造器中各个参数的含义:

  • corePoolSize:核心池的大小,这个参数跟后面讲述的线程池的实现原理有非常大的关系。在创建了线程池后,默认情况下,线程池中并没有任何线程,而是等待有任务到来才创建线程去执行任务,除非调用了prestartAllCoreThreads()或者prestartCoreThread()方法,从这2个方法的名字就可以看出,是预创建线程的意思,即在没有任务到来之前就创建corePoolSize个线程或者一个线程。默认情况下,在创建了线程池后,线程池中的线程数为0,当有任务来之后,就会创建一个线程去执行任务,当线程池中的线程数目达到corePoolSize后,就会把到达的任务放到缓存队列当中;
  • maximumPoolSize:线程池最大线程数,这个参数也是一个非常重要的参数,它表示在线程池中最多能创建多少个线程;
  • keepAliveTime:表示线程没有任务执行时最多保持多久时间会终止。默认情况下,只有当线程池中的线程数大于corePoolSize时,keepAliveTime才会起作用,直到线程池中的线程数不大于corePoolSize,即当线程池中的线程数大于corePoolSize时,如果一个线程空闲的时间达到keepAliveTime,则会终止,直到线程池中的线程数不超过corePoolSize。但是如果调用了allowCoreThreadTimeOut(boolean)方法,在线程池中的线程数不大于corePoolSize时,keepAliveTime参数也会起作用,直到线程池中的线程数为0;
  • unit:参数keepAliveTime的时间单位,有7种取值,在TimeUnit类中有7种静态属性:
  • [java] view plain copy
    1. TimeUnit.DAYS;               //天  
    2. TimeUnit.HOURS;             //小时  
    3. TimeUnit.MINUTES;           //分钟  
    4. TimeUnit.SECONDS;           //秒  
    5. TimeUnit.MILLISECONDS;      //毫秒  
    6. TimeUnit.MICROSECONDS;      //微妙  
    7. TimeUnit.NANOSECONDS;       //纳秒  
  • workQueue:一个阻塞队列,用来存储等待执行的任务,这个参数的选择也很重要,会对线程池的运行过程产生重大影响,一般来说,这里的阻塞队列有以下几种选择:
  • [java] view plain copy
    1. ArrayBlockingQueue;  
    2. LinkedBlockingQueue;  
    3. SynchronousQueue;  

ArrayBlockingQueue和PriorityBlockingQueue使用较少,一般使用LinkedBlockingQueue和Synchronous。线程池的排队策略与BlockingQueue有关。

  • threadFactory:线程工厂,主要用来创建线程;
  • handler:表示当拒绝处理任务时的策略,有以下四种取值:
  • [java] view plain copy
    1. ThreadPoolExecutor.AbortPolicy:丢弃任务并抛出RejectedExecutionException异常。   
    2. ThreadPoolExecutor.DiscardPolicy:也是丢弃任务,但是不抛出异常。   
    3. ThreadPoolExecutor.DiscardOldestPolicy:丢弃队列最前面的任务,然后重新尝试执行任务(重复此过程)  
    4. ThreadPoolExecutor.CallerRunsPolicy:由调用线程处理该任务  

具体参数的配置与线程池的关系将在下一节讲述。

从上面给出的ThreadPoolExecutor类的代码可以知道,ThreadPoolExecutor继承了AbstractExecutorService,我们来看一下AbstractExecutorService的实现:

[java] view plain copy
  1. public abstract class AbstractExecutorService implements ExecutorService {  
  2.   
  3.     protected <T> RunnableFuture<T> newTaskFor(Runnable runnable, T value) { };  
  4.     protected <T> RunnableFuture<T> newTaskFor(Callable<T> callable) { };  
  5.     public Future<?> submit(Runnable task) {};  
  6.     public <T> Future<T> submit(Runnable task, T result) { };  
  7.     public <T> Future<T> submit(Callable<T> task) { };  
  8.     private <T> T doInvokeAny(Collection<? extends Callable<T>> tasks,  
  9.                             boolean timed, long nanos)  
  10.         throws InterruptedException, ExecutionException, TimeoutException {  
  11.     };  
  12.     public <T> T invokeAny(Collection<? extends Callable<T>> tasks)  
  13.         throws InterruptedException, ExecutionException {  
  14.     };  
  15.     public <T> T invokeAny(Collection<? extends Callable<T>> tasks,  
  16.                            long timeout, TimeUnit unit)  
  17.         throws InterruptedException, ExecutionException, TimeoutException {  
  18.     };  
  19.     public <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks)  
  20.         throws InterruptedException {  
  21.     };  
  22.     public <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks,  
  23.                                          long timeout, TimeUnit unit)  
  24.         throws InterruptedException {  
  25.     };  
  26. }  
  27.   
  28. AbstractExecutorService是一个抽象类,它实现了ExecutorService接口。  
  29.   
  30. 我们接着看ExecutorService接口的实现:  
  31.   
  32. public interface ExecutorService extends Executor {  
  33.   
  34.     void shutdown();  
  35.     boolean isShutdown();  
  36.     boolean isTerminated();  
  37.     boolean awaitTermination(long timeout, TimeUnit unit)  
  38.         throws InterruptedException;  
  39.     <T> Future<T> submit(Callable<T> task);  
  40.     <T> Future<T> submit(Runnable task, T result);  
  41.     Future<?> submit(Runnable task);  
  42.     <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks)  
  43.         throws InterruptedException;  
  44.     <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks,  
  45.                                   long timeout, TimeUnit unit)  
  46.         throws InterruptedException;  
  47.   
  48.     <T> T invokeAny(Collection<? extends Callable<T>> tasks)  
  49.         throws InterruptedException, ExecutionException;  
  50.     <T> T invokeAny(Collection<? extends Callable<T>> tasks,  
  51.                     long timeout, TimeUnit unit)  
  52.         throws InterruptedException, ExecutionException, TimeoutException;  
  53. }  
  54.   
  55. 而ExecutorService又是继承了Executor接口,我们看一下Executor接口的实现:  
  56.   
  57. public interface Executor {  
  58.     void execute(Runnable command);  
  59. }  

到这里,大家应该明白了ThreadPoolExecutor、AbstractExecutorService、ExecutorService和Executor几个之间的关系了。

Executor是一个顶层接口,在它里面只声明了一个方法execute(Runnable),返回值为void,参数为Runnable类型,从字面意思可以理解,就是用来执行传进去的任务的;

然后ExecutorService接口继承了Executor接口,并声明了一些方法:submit、invokeAll、invokeAny以及shutDown等;

抽象类AbstractExecutorService实现了ExecutorService接口,基本实现了ExecutorService中声明的所有方法;

然后ThreadPoolExecutor继承了类AbstractExecutorService。

在ThreadPoolExecutor类中有几个非常重要的方法:

[java] view plain copy
  1. execute()  
  2. submit()  
  3. shutdown()  
  4. shutdownNow()  

execute()方法实际上是Executor中声明的方法,在ThreadPoolExecutor进行了具体的实现,这个方法是ThreadPoolExecutor的核心方法,通过这个方法可以向线程池提交一个任务,交由线程池去执行。

submit()方法是在ExecutorService中声明的方法,在AbstractExecutorService就已经有了具体的实现,在ThreadPoolExecutor中并没有对其进行重写,这个方法也是用来向线程池提交任务的,但是它和execute()方法不同,它能够返回任务执行的结果,去看submit()方法的实现,会发现它实际上还是调用的execute()方法,只不过它利用了Future来获取任务执行结果(Future相关内容将在下一篇讲述)。

shutdown()和shutdownNow()是用来关闭线程池的。

还有很多其他的方法:

比如:getQueue() 、getPoolSize() 、getActiveCount()、getCompletedTaskCount()等获取与线程池相关属性的方法,有兴趣的朋友可以自行查阅API。

二.深入剖析线程池实现原理

在上一节我们从宏观上介绍了ThreadPoolExecutor,下面我们来深入解析一下线程池的具体实现原理,将从下面几个方面讲解:

  • 1.线程池状态
  • 2.任务的执行
  • 3.线程池中的线程初始化
  • 4.任务缓存队列及排队策略
  • 5.任务拒绝策略
  • 6.线程池的关闭
  • 7.线程池容量的动态调整

1.线程池状态

在ThreadPoolExecutor中定义了一个Volatile变量,另外定义了几个static final变量表示线程池的各个状态:

[java] view plain copy
  1. volatile int runState;  
  2. static final int RUNNING    = 0;  
  3. static final int SHUTDOWN   = 1;  
  4. static final int STOP       = 2;  
  5. static final int TERMINATED = 3;  

runState表示当前线程池的状态,它是一个volatile变量用来保证线程之间的可见性;

下面的几个static final变量表示runState可能的几个取值。

当创建线程池后,初始时,线程池处于RUNNING状态;

如果调用了shutdown()方法,则线程池处于SHUTDOWN状态,此时线程池不能够接受新的任务,它会等待所有任务执行完毕;

如果调用了shutdownNow()方法,则线程池处于STOP状态,此时线程池不能接受新的任务,并且会去尝试终止正在执行的任务;

当线程池处于SHUTDOWN或STOP状态,并且所有工作线程已经销毁,任务缓存队列已经清空或执行结束后,线程池被设置为TERMINATED状态。

2.任务的执行

在了解将任务提交给线程池到任务执行完毕整个过程之前,我们先来看一下ThreadPoolExecutor类中其他的一些比较重要成员变量:

[java] view plain copy
  1. private final BlockingQueue<Runnable> workQueue;              //任务缓存队列,用来存放等待执行的任务  
  2. private final ReentrantLock mainLock = new ReentrantLock();   //线程池的主要状态锁,对线程池状态(比如线程池大小  
  3.                                                               //、runState等)的改变都要使用这个锁  
  4. private final HashSet<Worker> workers = new HashSet<Worker>();  //用来存放工作集  
  5.   
  6. private volatile long  keepAliveTime;    //线程存活时间     
  7. private volatile boolean allowCoreThreadTimeOut;   //是否允许为核心线程设置存活时间  
  8. private volatile int   corePoolSize;     //核心池的大小(即线程池中的线程数目大于这个参数时,提交的任务会被放进任务缓存队列)  
  9. private volatile int   maximumPoolSize;   //线程池最大能容忍的线程数  
  10.   
  11. private volatile int   poolSize;       //线程池中当前的线程数  
  12.   
  13. private volatile RejectedExecutionHandler handler; //任务拒绝策略  
  14.   
  15. private volatile ThreadFactory threadFactory;   //线程工厂,用来创建线程  
  16.   
  17. private int largestPoolSize;   //用来记录线程池中曾经出现过的最大线程数  
  18.   
  19. private long completedTaskCount;   //用来记录已经执行完毕的任务个数  

每个变量的作用都已经标明出来了,这里要重点解释一下corePoolSize、maximumPoolSize、largestPoolSize三个变量。

corePoolSize在很多地方被翻译成核心池大小,其实我的理解这个就是线程池的大小。举个简单的例子:

假如有一个工厂,工厂里面有10个工人,每个工人同时只能做一件任务。

因此只要当10个工人中有工人是空闲的,来了任务就分配给空闲的工人做;

当10个工人都有任务在做时,如果还来了任务,就把任务进行排队等待;

如果说新任务数目增长的速度远远大于工人做任务的速度,那么此时工厂主管可能会想补救措施,比如重新招4个临时工人进来;

然后就将任务也分配给这4个临时工人做;

如果说着14个工人做任务的速度还是不够,此时工厂主管可能就要考虑不再接收新的任务或者抛弃前面的一些任务了。

当这14个工人当中有人空闲时,而新任务增长的速度又比较缓慢,工厂主管可能就考虑辞掉4个临时工了,只保持原来的10个工人,毕竟请额外的工人是要花钱的。

这个例子中的corePoolSize就是10,而maximumPoolSize就是14(10+4)。

也就是说corePoolSize就是线程池大小,maximumPoolSize在我看来是线程池的一种补救措施,即任务量突然过大时的一种补救措施。

不过为了方便理解,在本文后面还是将corePoolSize翻译成核心池大小。

largestPoolSize只是一个用来起记录作用的变量,用来记录线程池中曾经有过的最大线程数目,跟线程池的容量没有任何关系。

下面我们进入正题,看一下任务从提交到最终执行完毕经历了哪些过程。

在ThreadPoolExecutor类中,最核心的任务提交方法是execute()方法,虽然通过submit也可以提交任务,但是实际上submit方法里面最终调用的还是execute()方法,所以我们只需要研究execute()方法的实现原理即可:

[java] view plain copy
  1. public void execute(Runnable command) {  
  2.     if (command == null)  
  3.         throw new NullPointerException();  
  4.     if (poolSize >= corePoolSize || !addIfUnderCorePoolSize(command)) {  
  5.         if (runState == RUNNING && workQueue.offer(command)) {  
  6.             if (runState != RUNNING || poolSize == 0)  
  7.                 ensureQueuedTaskHandled(command);  
  8.         }  
  9.         else if (!addIfUnderMaximumPoolSize(command))  
  10.             reject(command); // is shutdown or saturated  
  11.     }  
  12. }  

上面的代码可能看起来不是那么容易理解,下面我们一句一句解释:

首先,判断提交的任务command是否为null,若是null,则抛出空指针异常;

接着是这句,这句要好好理解一下:

[java] view plain copy
  1. if (poolSize >= corePoolSize || !addIfUnderCorePoolSize(command))  

由于是或条件运算符,所以先计算前半部分的值,如果线程池中当前线程数不小于核心池大小,那么就会直接进入下面的if语句块了。

如果线程池中当前线程数小于核心池大小,则接着执行后半部分,也就是执行:

[java] view plain copy
  1. addIfUnderCorePoolSize(command)  

如果执行完addIfUnderCorePoolSize这个方法返回false,则继续执行下面的if语句块,否则整个方法就直接执行完毕了。

如果执行完addIfUnderCorePoolSize这个方法返回false,然后接着判断:

[java] view plain copy
  1. if (runState == RUNNING && workQueue.offer(command))  

如果当前线程池处于RUNNING状态,则将任务放入任务缓存队列;如果当前线程池不处于RUNNING状态或者任务放入缓存队列失败,则执行:

[java] view plain copy
  1. addIfUnderMaximumPoolSize(command)  

如果执行addIfUnderMaximumPoolSize方法失败,则执行reject()方法进行任务拒绝处理。

回到前面:

[java] view plain copy
  1. if (runState == RUNNING && workQueue.offer(command))  

这句的执行,如果说当前线程池处于RUNNING状态且将任务放入任务缓存队列成功,则继续进行判断:

[java] view plain copy
  1. if (runState != RUNNING || poolSize == 0)  

这句判断是为了防止在将此任务添加进任务缓存队列的同时其他线程突然调用shutdown或者shutdownNow方法关闭了线程池的一种应急措施。如果是这样就执行:

[java] view plain copy
  1. ensureQueuedTaskHandled(command)  

进行应急处理,从名字可以看出是保证 添加到任务缓存队列中的任务得到处理。

我们接着看2个关键方法的实现:addIfUnderCorePoolSize和addIfUnderMaximumPoolSize:

[java] view plain copy
  1. private boolean addIfUnderCorePoolSize(Runnable firstTask) {  
  2.     Thread t = null;  
  3.     final ReentrantLock mainLock = this.mainLock;  
  4.     mainLock.lock();  
  5.     try {  
  6.         if (poolSize < corePoolSize && runState == RUNNING)  
  7.             t = addThread(firstTask);        //创建线程去执行firstTask任务     
  8.         } finally {  
  9.         mainLock.unlock();  
  10.     }  
  11.     if (t == null)  
  12.         return false;  
  13.     t.start();  
  14.     return true;  
  15. }  

这个是addIfUnderCorePoolSize方法的具体实现,从名字可以看出它的意图就是当低于核心吃大小时执行的方法。下面看其具体实现,首先获取到锁,因为这地方涉及到线程池状态的变化,先通过if语句判断当前线程池中的线程数目是否小于核心池大小,有朋友也许会有疑问:前面在execute()方法中不是已经判断过了吗,只有线程池当前线程数目小于核心池大小才会执行addIfUnderCorePoolSize方法的,为何这地方还要继续判断?原因很简单,前面的判断过程中并没有加锁,因此可能在execute方法判断的时候poolSize小于corePoolSize,而判断完之后,在其他线程中又向线程池提交了任务,就可能导致poolSize不小于corePoolSize了,所以需要在这个地方继续判断。然后接着判断线程池的状态是否为RUNNING,原因也很简单,因为有可能在其他线程中调用了shutdown或者shutdownNow方法。然后就是执行

[java] view plain copy
  1. t = addThread(firstTask);  

这个方法也非常关键,传进去的参数为提交的任务,返回值为Thread类型。然后接着在下面判断t是否为空,为空则表明创建线程失败(即poolSize>=corePoolSize或者runState不等于RUNNING),否则调用t.start()方法启动线程。

我们来看一下addThread方法的实现:

[java] view plain copy
  1. private Thread addThread(Runnable firstTask) {  
  2.     Worker w = new Worker(firstTask);  
  3.     Thread t = threadFactory.newThread(w);  //创建一个线程,执行任务     
  4.     if (t != null) {  
  5.         w.thread = t;            //将创建的线程的引用赋值为w的成员变量         
  6.         workers.add(w);  
  7.         int nt = ++poolSize;     //当前线程数加1         
  8.         if (nt > largestPoolSize)  
  9.             largestPoolSize = nt;  
  10.     }  
  11.     return t;  
  12. }  

在addThread方法中,首先用提交的任务创建了一个Worker对象,然后调用线程工厂threadFactory创建了一个新的线程t,然后将线程t的引用赋值给了Worker对象的成员变量thread,接着通过workers.add(w)将Worker对象添加到工作集当中。

下面我们看一下Worker类的实现:

[java] view plain copy
  1. private final class Worker implements Runnable {  
  2.     private final ReentrantLock runLock = new ReentrantLock();  
  3.     private Runnable firstTask;  
  4.     volatile long completedTasks;  
  5.     Thread thread;  
  6.     Worker(Runnable firstTask) {  
  7.         this.firstTask = firstTask;  
  8.     }  
  9.     boolean isActive() {  
  10.         return runLock.isLocked();  
  11.     }  
  12.     void interruptIfIdle() {  
  13.         final ReentrantLock runLock = this.runLock;  
  14.         if (runLock.tryLock()) {  
  15.             try {  
  16.         if (thread != Thread.currentThread())  
  17.         thread.interrupt();  
  18.             } finally {  
  19.                 runLock.unlock();  
  20.             }  
  21.         }  
  22.     }  
  23.     void interruptNow() {  
  24.         thread.interrupt();  
  25.     }  
  26.   
  27.     private void runTask(Runnable task) {  
  28.         final ReentrantLock runLock = this.runLock;  
  29.         runLock.lock();  
  30.         try {  
  31.             if (runState < STOP &&  
  32.                 Thread.interrupted() &&  
  33.                 runState >= STOP)  
  34.             boolean ran = false;  
  35.             beforeExecute(thread, task);   //beforeExecute方法是ThreadPoolExecutor类的一个方法,没有具体实现,用户可以根据  
  36.             //自己需要重载这个方法和后面的afterExecute方法来进行一些统计信息,比如某个任务的执行时间等             
  37.             try {  
  38.                 task.run();  
  39.                 ran = true;  
  40.                 afterExecute(task, null);  
  41.                 ++completedTasks;  
  42.             } catch (RuntimeException ex) {  
  43.                 if (!ran)  
  44.                     afterExecute(task, ex);  
  45.                 throw ex;  
  46.             }  
  47.         } finally {  
  48.             runLock.unlock();  
  49.         }  
  50.     }  
  51.   
  52.     public void run() {  
  53.         try {  
  54.             Runnable task = firstTask;  
  55.             firstTask = null;  
  56.             while (task != null || (task = getTask()) != null) {  
  57.                 runTask(task);  
  58.                 task = null;  
  59.             }  
  60.         } finally {  
  61.             workerDone(this);   //当任务队列中没有任务时,进行清理工作         
  62.         }  
  63.     }  
  64. }  

它实际上实现了Runnable接口,因此上面的Thread t = threadFactory.newThread(w);效果跟下面这句的效果基本一样:

[java] view plain copy
  1. Thread t = new Thread(w);  

相当于传进去了一个Runnable任务,在线程t中执行这个Runnable。

既然Worker实现了Runnable接口,那么自然最核心的方法便是run()方法了:

[java] view plain copy
  1. public void run() {  
  2.     try {  
  3.         Runnable task = firstTask;  
  4.         firstTask = null;  
  5.         while (task != null || (task = getTask()) != null) {  
  6.             runTask(task);  
  7.             task = null;  
  8.         }  
  9.     } finally {  
  10.         workerDone(this);  
  11.     }  
  12. }  

从run方法的实现可以看出,它首先执行的是通过构造器传进来的任务firstTask,在调用runTask()执行完firstTask之后,在while循环里面不断通过getTask()去取新的任务来执行,那么去哪里取呢?自然是从任务缓存队列里面去取,getTask是ThreadPoolExecutor类中的方法,并不是Worker类中的方法,下面是getTask方法的实现:

[java] view plain copy
  1. Runnable getTask() {  
  2.     for (;;) {  
  3.         try {  
  4.             int state = runState;  
  5.             if (state > SHUTDOWN)  
  6.                 return null;  
  7.             Runnable r;  
  8.             if (state == SHUTDOWN)  // Help drain queue  
  9.                 r = workQueue.poll();  
  10.             else if (poolSize > corePoolSize || allowCoreThreadTimeOut) //如果线程数大于核心池大小或者允许为核心池线程设置空闲时间,  
  11.                 //则通过poll取任务,若等待一定的时间取不到任务,则返回null  
  12.                 r = workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS);  
  13.             else  
  14.                 r = workQueue.take();  
  15.             if (r != null)  
  16.                 return r;  
  17.             if (workerCanExit()) {    //如果没取到任务,即r为null,则判断当前的worker是否可以退出  
  18.                 if (runState >= SHUTDOWN) // Wake up others  
  19.                     interruptIdleWorkers();   //中断处于空闲状态的worker  
  20.                 return null;  
  21.             }  
  22.             // Else retry  
  23.         } catch (InterruptedException ie) {  
  24.             // On interruption, re-check runState  
  25.         }  
  26.     }  
  27. }  

在getTask中,先判断当前线程池状态,如果runState大于SHUTDOWN(即为STOP或者TERMINATED),则直接返回null。

如果runState为SHUTDOWN或者RUNNING,则从任务缓存队列取任务。

如果当前线程池的线程数大于核心池大小corePoolSize或者允许为核心池中的线程设置空闲存活时间,则调用poll(time,timeUnit)来取任务,这个方法会等待一定的时间,如果取不到任务就返回null。

然后判断取到的任务r是否为null,为null则通过调用workerCanExit()方法来判断当前worker是否可以退出,我们看一下workerCanExit()的实现:

[java] view plain copy
  1. private boolean workerCanExit() {  
  2.     final ReentrantLock mainLock = this.mainLock;  
  3.     mainLock.lock();  
  4.     boolean canExit;  
  5.     //如果runState大于等于STOP,或者任务缓存队列为空了  
  6.     //或者  允许为核心池线程设置空闲存活时间并且线程池中的线程数目大于1  
  7.     try {  
  8.         canExit = runState >= STOP ||  
  9.             workQueue.isEmpty() ||  
  10.             (allowCoreThreadTimeOut &&  
  11.              poolSize > Math.max(1, corePoolSize));  
  12.     } finally {  
  13.         mainLock.unlock();  
  14.     }  
  15.     return canExit;  
  16. }  

也就是说如果线程池处于STOP状态、或者任务队列已为空或者允许为核心池线程设置空闲存活时间并且线程数大于1时,允许worker退出。如果允许worker退出,则调用interruptIdleWorkers()中断处于空闲状态的worker,我们看一下interruptIdleWorkers()的实现:

[java] view plain copy
  1. void interruptIdleWorkers() {  
  2.     final ReentrantLock mainLock = this.mainLock;  
  3.     mainLock.lock();  
  4.     try {  
  5.         for (Worker w : workers)  //实际上调用的是worker的interruptIfIdle()方法  
  6.             w.interruptIfIdle();  
  7.     } finally {  
  8.         mainLock.unlock();  
  9.     }  
  10. }  

从实现可以看出,它实际上调用的是worker的interruptIfIdle()方法,在worker的interruptIfIdle()方法中:

[java] view plain copy
  1. void interruptIfIdle() {  
  2.     final ReentrantLock runLock = this.runLock;  
  3.     if (runLock.tryLock()) {    //注意这里,是调用tryLock()来获取锁的,因为如果当前worker正在执行任务,锁已经被获取了,是无法获取到锁的  
  4.                                 //如果成功获取了锁,说明当前worker处于空闲状态  
  5.         try {  
  6.     if (thread != Thread.currentThread())    
  7.     thread.interrupt();  
  8.         } finally {  
  9.             runLock.unlock();  
  10.         }  
  11.     }  
  12. }  

这里有一个非常巧妙的设计方式,假如我们来设计线程池,可能会有一个任务分派线程,当发现有线程空闲时,就从任务缓存队列中取一个任务交给空闲线程执行。但是在这里,并没有采用这样的方式,因为这样会要额外地对任务分派线程进行管理,无形地会增加难度和复杂度,这里直接让执行完任务的线程去任务缓存队列里面取任务来执行。

我们再看addIfUnderMaximumPoolSize方法的实现,这个方法的实现思想和addIfUnderCorePoolSize方法的实现思想非常相似,唯一的区别在于addIfUnderMaximumPoolSize方法是在线程池中的线程数达到了核心池大小并且往任务队列中添加任务失败的情况下执行的:

[java] view plain copy
  1. private boolean addIfUnderMaximumPoolSize(Runnable firstTask) {  
  2.     Thread t = null;  
  3.     final ReentrantLock mainLock = this.mainLock;  
  4.     mainLock.lock();  
  5.     try {  
  6.         if (poolSize < maximumPoolSize && runState == RUNNING)  
  7.             t = addThread(firstTask);  
  8.     } finally {  
  9.         mainLock.unlock();  
  10.     }  
  11.     if (t == null)  
  12.         return false;  
  13.     t.start();  
  14.     return true;  
  15. }  

看到没有,其实它和addIfUnderCorePoolSize方法的实现基本一模一样,只是if语句判断条件中的poolSize < maximumPoolSize不同而已。

到这里,大部分朋友应该对任务提交给线程池之后到被执行的整个过程有了一个基本的了解,下面总结一下:

1)首先,要清楚corePoolSize和maximumPoolSize的含义;

2)其次,要知道Worker是用来起到什么作用的;

3)要知道任务提交给线程池之后的处理策略,这里总结一下主要有4点:

  • 如果当前线程池中的线程数目小于corePoolSize,则每来一个任务,就会创建一个线程去执行这个任务;
  • 如果当前线程池中的线程数目>=corePoolSize,则每来一个任务,会尝试将其添加到任务缓存队列当中,若添加成功,则该任务会等待空闲线程将其取出去执行;若添加失败(一般来说是任务缓存队列已满),则会尝试创建新的线程去执行这个任务;
  • 如果当前线程池中的线程数目达到maximumPoolSize,则会采取任务拒绝策略进行处理;
  • 如果线程池中的线程数量大于 corePoolSize时,如果某线程空闲时间超过keepAliveTime,线程将被终止,直至线程池中的线程数目不大于corePoolSize;如果允许为核心池中的线程设置存活时间,那么核心池中的线程空闲时间超过keepAliveTime,线程也会被终止。

3.线程池中的线程初始化

默认情况下,创建线程池之后,线程池中是没有线程的,需要提交任务之后才会创建线程。

在实际中如果需要线程池创建之后立即创建线程,可以通过以下两个方法办到:

  • prestartCoreThread():初始化一个核心线程;
  • prestartAllCoreThreads():初始化所有核心线程

下面是这2个方法的实现:

[java] view plain copy
  1. public boolean prestartCoreThread() {  
  2.     return addIfUnderCorePoolSize(null); //注意传进去的参数是null  
  3. }  
  4.   
  5. public int prestartAllCoreThreads() {  
  6.     int n = 0;  
  7.     while (addIfUnderCorePoolSize(null))//注意传进去的参数是null  
  8.         ++n;  
  9.     return n;  
  10. }  

注意上面传进去的参数是null,根据第2小节的分析可知如果传进去的参数为null,则最后执行线程会阻塞在getTask方法中的

[java] view plain copy
  1. r = workQueue.take();  

即等待任务队列中有任务。

4.任务缓存队列及排队策略

在前面我们多次提到了任务缓存队列,即workQueue,它用来存放等待执行的任务。

workQueue的类型为BlockingQueue<Runnable>,通常可以取下面三种类型:

1)ArrayBlockingQueue:基于数组的先进先出队列,此队列创建时必须指定大小;

2)LinkedBlockingQueue:基于链表的先进先出队列,如果创建时没有指定此队列大小,则默认为Integer.MAX_VALUE;

3)synchronousQueue:这个队列比较特殊,它不会保存提交的任务,而是将直接新建一个线程来执行新来的任务。

5.任务拒绝策略

当线程池的任务缓存队列已满并且线程池中的线程数目达到maximumPoolSize,如果还有任务到来就会采取任务拒绝策略,通常有以下四种策略:

[java] view plain copy
  1. ThreadPoolExecutor.AbortPolicy:丢弃任务并抛出RejectedExecutionException异常。  
  2. ThreadPoolExecutor.DiscardPolicy:也是丢弃任务,但是不抛出异常。  
  3. ThreadPoolExecutor.DiscardOldestPolicy:丢弃队列最前面的任务,然后重新尝试执行任务(重复此过程)  
  4. ThreadPoolExecutor.CallerRunsPolicy:由调用线程处理该任务  

6.线程池的关闭

ThreadPoolExecutor提供了两个方法,用于线程池的关闭,分别是shutdown()和shutdownNow(),其中:

  • shutdown():不会立即终止线程池,而是要等所有任务缓存队列中的任务都执行完后才终止,但再也不会接受新的任务
  • shutdownNow():立即终止线程池,并尝试打断正在执行的任务,并且清空任务缓存队列,返回尚未执行的任务

7.线程池容量的动态调整

ThreadPoolExecutor提供了动态调整线程池容量大小的方法:setCorePoolSize()和setMaximumPoolSize(),

  • setCorePoolSize:设置核心池大小
  • setMaximumPoolSize:设置线程池最大能创建的线程数目大小

当上述参数从小变大时,ThreadPoolExecutor进行线程赋值,还可能立即创建新的线程来执行任务。

三.使用示例

前面我们讨论了关于线程池的实现原理,这一节我们来看一下它的具体使用:

[java] view plain copy
  1. public class Test {  
  2.      public static void main(String[] args) {     
  3.          ThreadPoolExecutor executor = new ThreadPoolExecutor(510200, TimeUnit.MILLISECONDS,  
  4.                  new ArrayBlockingQueue<Runnable>(5));  
  5.   
  6.          for(int i=0;i<15;i++){  
  7.              MyTask myTask = new MyTask(i);  
  8.              executor.execute(myTask);  
  9.              System.out.println("线程池中线程数目:"+executor.getPoolSize()+",队列中等待执行的任务数目:"+  
  10.              executor.getQueue().size()+",已执行玩别的任务数目:"+executor.getCompletedTaskCount());  
  11.          }  
  12.          executor.shutdown();  
  13.      }  
  14. }  
  15.   
  16. class MyTask implements Runnable {  
  17.     private int taskNum;  
  18.   
  19.     public MyTask(int num) {  
  20.         this.taskNum = num;  
  21.     }  
  22.   
  23.     @Override  
  24.     public void run() {  
  25.         System.out.println("正在执行task "+taskNum);  
  26.         try {  
  27.             Thread.currentThread().sleep(4000);  
  28.         } catch (InterruptedException e) {  
  29.             e.printStackTrace();  
  30.         }  
  31.         System.out.println("task "+taskNum+"执行完毕");  
  32.     }  
  33. }  

执行结果:

[java] view plain copy
  1. 正在执行task 0  
  2. 线程池中线程数目:1,队列中等待执行的任务数目:0,已执行玩别的任务数目:0  
  3. 线程池中线程数目:2,队列中等待执行的任务数目:0,已执行玩别的任务数目:0  
  4. 正在执行task 1  
  5. 线程池中线程数目:3,队列中等待执行的任务数目:0,已执行玩别的任务数目:0  
  6. 正在执行task 2  
  7. 线程池中线程数目:4,队列中等待执行的任务数目:0,已执行玩别的任务数目:0  
  8. 正在执行task 3  
  9. 线程池中线程数目:5,队列中等待执行的任务数目:0,已执行玩别的任务数目:0  
  10. 正在执行task 4  
  11. 线程池中线程数目:5,队列中等待执行的任务数目:1,已执行玩别的任务数目:0  
  12. 线程池中线程数目:5,队列中等待执行的任务数目:2,已执行玩别的任务数目:0  
  13. 线程池中线程数目:5,队列中等待执行的任务数目:3,已执行玩别的任务数目:0  
  14. 线程池中线程数目:5,队列中等待执行的任务数目:4,已执行玩别的任务数目:0  
  15. 线程池中线程数目:5,队列中等待执行的任务数目:5,已执行玩别的任务数目:0  
  16. 线程池中线程数目:6,队列中等待执行的任务数目:5,已执行玩别的任务数目:0  
  17. 正在执行task 10  
  18. 线程池中线程数目:7,队列中等待执行的任务数目:5,已执行玩别的任务数目:0  
  19. 正在执行task 11  
  20. 线程池中线程数目:8,队列中等待执行的任务数目:5,已执行玩别的任务数目:0  
  21. 正在执行task 12  
  22. 线程池中线程数目:9,队列中等待执行的任务数目:5,已执行玩别的任务数目:0  
  23. 正在执行task 13  
  24. 线程池中线程数目:10,队列中等待执行的任务数目:5,已执行玩别的任务数目:0  
  25. 正在执行task 14  
  26. task 3执行完毕  
  27. task 0执行完毕  
  28. task 2执行完毕  
  29. task 1执行完毕  
  30. 正在执行task 8  
  31. 正在执行task 7  
  32. 正在执行task 6  
  33. 正在执行task 5  
  34. task 4执行完毕  
  35. task 10执行完毕  
  36. task 11执行完毕  
  37. task 13执行完毕  
  38. task 12执行完毕  
  39. 正在执行task 9  
  40. task 14执行完毕  
  41. task 8执行完毕  
  42. task 5执行完毕  
  43. task 7执行完毕  
  44. task 6执行完毕  
  45. task 9执行完毕  

从执行结果可以看出,当线程池中线程的数目大于5时,便将任务放入任务缓存队列里面,当任务缓存队列满了之后,便创建新的线程。如果上面程序中,将for循环中改成执行20个任务,就会抛出任务拒绝异常了。

不过在Javadoc中,并不提倡我们直接使用ThreadPoolExecutor,而是使用Executors类中提供的几个静态方法来创建线程池:

[java] view plain copy
  1. Executors.newCachedThreadPool();        //创建一个缓冲池,缓冲池容量大小为Integer.MAX_VALUE  
  2. Executors.newSingleThreadExecutor();   //创建容量为1的缓冲池  
  3. Executors.newFixedThreadPool(int);    //创建固定容量大小的缓冲池  

下面是这三个静态方法的具体实现:

[java] view plain copy
  1. public static ExecutorService newFixedThreadPool(int nThreads) {  
  2.     return new ThreadPoolExecutor(nThreads, nThreads,  
  3.                                   0L, TimeUnit.MILLISECONDS,  
  4.                                   new LinkedBlockingQueue<Runnable>());  
  5. }  
  6. public static ExecutorService newSingleThreadExecutor() {  
  7.     return new FinalizableDelegatedExecutorService  
  8.         (new ThreadPoolExecutor(11,  
  9.                                 0L, TimeUnit.MILLISECONDS,  
  10.                                 new LinkedBlockingQueue<Runnable>()));  
  11. }  
  12. public static ExecutorService newCachedThreadPool() {  
  13.     return new ThreadPoolExecutor(0, Integer.MAX_VALUE,  
  14.                                   60L, TimeUnit.SECONDS,  
  15.                                   new SynchronousQueue<Runnable>());  
  16. }  

从它们的具体实现来看,它们实际上也是调用了ThreadPoolExecutor,只不过参数都已配置好了。

newFixedThreadPool创建的线程池corePoolSize和maximumPoolSize值是相等的,它使用的LinkedBlockingQueue;

newSingleThreadExecutor将corePoolSize和maximumPoolSize都设置为1,也使用的LinkedBlockingQueue;

newCachedThreadPool将corePoolSize设置为0,将maximumPoolSize设置为Integer.MAX_VALUE,使用的SynchronousQueue,也就是说来了任务就创建线程运行,当线程空闲超过60秒,就销毁线程。

实际中,如果Executors提供的三个静态方法能满足要求,就尽量使用它提供的三个方法,因为自己去手动配置ThreadPoolExecutor的参数有点麻烦,要根据实际任务的类型和数量来进行配置。

另外,如果ThreadPoolExecutor达不到要求,可以自己继承ThreadPoolExecutor类进行重写。

四.如何合理配置线程池的大小

本节来讨论一个比较重要的话题:如何合理配置线程池大小,仅供参考。

一般需要根据任务的类型来配置线程池大小:

如果是CPU密集型任务,就需要尽量压榨CPU,参考值可以设为 NCPU+1

如果是IO密集型任务,参考值可以设置为2*NCPU

当然,这只是一个参考值,具体的设置还需要根据实际情况进行调整,比如可以先将线程池大小设置为参考值,再观察任务运行情况和系统负载、资源利用率来进行适当调整。

参考资料:

  • http://ifeve.com/java-threadpool/
  • http://blog.163.com/among_1985/blog/static/275005232012618849266/
  • http://developer.51cto.com/art/201203/321885.htm
  • http://blog.csdn.net/java2000_wl/article/details/22097059
  • http://blog.csdn.net/cutesource/article/details/6061229
;