线程池的使用与源码

最新推荐文章于 2023-07-05 19:50:33 发布

小松学java

最新推荐文章于 2023-07-05 19:50:33 发布

阅读量97

点赞数

分类专栏：线程池文章标签： java 开发语言

本文链接：https://blog.csdn.net/weixin_50352768/article/details/124988446

版权

线程池专栏收录该内容

1 篇文章 0 订阅

订阅专栏

Java实现了三个线程池

ExecutorService executorService1 = Executors.newCachedThreadPool();
ExecutorService executorService2 = Executors.newFixedThreadPool(10);
ExecutorService executorService3 = Executors.newSingleThreadExecutor();

三个线程池有问题,阿里巴巴检测软件不通过

Executors.newCachedThreadPool()

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

最大线程数为int的最大值,2的31次方,可能会造成CPU百分百占用

Executors.newFixedThreadPool(10)

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

使用无界队列,会造成OOM

Executors.newSingleThreadExecutor()

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

核心现场数和最大线程数1,大部分业务不满足

Java提供的核心线程都是new的ThreadPoolExecutor,对这个类的参数进行说明.

简单使用案例

  public static void main(String[] args) throws InterruptedException {
        Long start = System.currentTimeMillis();
        final Random random = new Random();
        final List<Integer> list = new ArrayList<Integer>();

        ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(
                //核心线程数
                10,
                //当线程数>=corePoolSize，且任务队列已满时。线程池会创建新线程来处理任务
                //当线程数=maxPoolSize，且任务队列已满时，线程池会拒绝处理任务而抛出异常
                20,
                //keepAliveTime：线程空闲时间
                //当线程空闲时间达到keepAliveTime时，线程会退出，直到线程数量=corePoolSize
                //如果allowCoreThreadTimeout=true，则会直到线程数量=0
                0L,
                //允许核心线程超时
                TimeUnit.MILLISECONDS,
                //当核心线程数达到最大时，新任务会放在队列中排队等待执行
                new LinkedBlockingQueue<Runnable>(10),


                new ThreadFactory() {
            @Override
            public Thread newThread(Runnable r) {
                return null;
            }
        },


                //两种情况会拒绝处理任务：
                //当线程数已经达到maxPoolSize，切队列已满，会拒绝新任务
                //当线程池被调用shutdown()后，会等待线程池里的任务执行完毕，再shutdown。如果在调用shutdown()和线程池真正shutdown之间提交任务，会拒绝新任务
                //线程池会调用rejectedExecutionHandler来处理这个任务。如果没有设置默认是AbortPolicy，会抛出异常
                //ThreadPoolExecutor类有几个内部实现类来处理这类情况：
                //AbortPolicy 丢弃任务，抛运行时异常
                //CallerRunsPolicy 执行任务
                //DiscardPolicy 忽视，什么都不会发生
                //DiscardOldestPolicy 从队列中踢出最先进入队列（最后一个执行）的任务
                //实现RejectedExecutionHandler接口，可自定义处理器
                new RejectedExecutionHandler() {
                    @Override
                    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
                        System.out.println("");
                    }
                });
        threadPoolExecutor.execute(new Runnable() {
            //业务代码使用的run方法.
            @Override
            public void run() {
                list.add(random.nextInt());
            }
        });
        
        System.out.println("时间：" + (System.currentTimeMillis() - start));
        System.out.println("大小：" + list.size());

    }

ThreadPoolExecutor的重要参数

1.corePoolSize：核心线程数

(1)核心线程会一直存活，及时没有任务需要执行

(2)当线程数小于核心线程数时，即使有线程空闲，线程池也会优先创建新线程处理

(3)设置allowCoreThreadTimeout=true（默认false）时，核心线程会超时关闭

2.queueCapacity：任务队列容量（阻塞队列）

(1)当核心线程数达到最大时，新任务会放在队列中排队等待执行

3.maxPoolSize：最大线程数

(1)当线程数>=corePoolSize，且任务队列已满时。线程池会创建新线程来处理任务

(2)当线程数=maxPoolSize，且任务队列已满时，线程池会拒绝处理任务而抛出异常

4.keepAliveTime：线程空闲时间

(1)当线程空闲时间达到keepAliveTime时，线程会退出，直到线程数量=corePoolSize

(2)如果allowCoreThreadTimeout=true，则会直到线程数量=0

5.allowCoreThreadTimeout：允许核心线程超时

6.rejectedExecutionHandler：任务拒绝处理器

(1)两种情况会拒绝处理任务：

- - 当线程数已经达到maxPoolSize，切队列已满，会拒绝新任务.
  - 当线程池被调用shutdown()后，会等待线程池里的任务执行完毕，再shutdown。如果在调用shutdown()和线程池真正shutdown之间提交任务，会拒绝新任务

(2)线程池会调用rejectedExecutionHandler来处理这个任务。如果没有设置默认是AbortPolicy，会抛出异常

(3)ThreadPoolExecutor类有几个内部实现类来处理这类情况：

- - AbortPolicy 丢弃任务，抛运行时异常
  - CallerRunsPolicy 执行任务
  - DiscardPolicy 忽视，什么都不会发生
  - DiscardOldestPolicy 从队列中踢出最先进入队列（最后一个执行）的任务

(4)实现RejectedExecutionHandler接口，可自定义处理器

源码分析:

主要关注,提交优先级和执行优先级如下图

// runState is stored in the high‐order bits
private static final int RUNNING = ‐1 << COUNT_BITS;
private static final int SHUTDOWN = 0 << COUNT_BITS;
private static final int STOP = 1 << COUNT_BITS;
private static final int TIDYING = 2 << COUNT_BITS;
private static final int TERMINATED = 3 << COUNT_BITS;

其中COUNT_BITS是 int 位数 private static final int COUNT_BITS = Integer.SIZE - 3; //Integer.SIZE=32 所以实际 COUNT_BITS = 29，用上面的5个常量表示线程池的状态，实际上是使用32位中的高3位表示；

execute方法：

1 int c = ctl.get();
2 1、判断当前的线程数是否小于corePoolSize如果是，
3 使用入参任务通过addWord方法创建一个新的线程，
4 如果能完成新线程创建exexute方法结束，成功提交任务；
5 if (workerCountOf(c) < corePoolSize) {
6 if (addWorker(command, true))
7 return;
8 c = ctl.get();
9 }
10 2、在第一步没有完成任务提交；状态为运行并且能成功加入任务到工作队列后，
11 再进行一次check，如果状态在任务加入队列后变为了非运行（有可能是在执行到这里线程池shu
tdown了）
12 非运行状态下当然是需要reject；
13 然后再判断当前线程数是否为0（有可能这个时候线程数变为了0），如是，新增一个线程；
14 if (isRunning(c) && workQueue.offer(command)) {
15 int recheck = ctl.get();
16 if (! isRunning(recheck) && remove(command))
17 reject(command);
18 else if (workerCountOf(recheck) == 0)
19 addWorker(null, false); 判断当前工作线程池数是否为0
20 如果是创建一个null任务，任务在堵塞队列存在了就会从队列中取出 这样做的意义是
21 保证线程池在running状态必须有一个任务在执行
22
23
24
25 }
26 3、如果不能加入任务到工作队列，将尝试使用任务新增一个线程，如果失败，
27 则是线程池已经shutdown或者线程池已经达到饱和状态，所以reject;
28 从上面新增任务的execute方法也可以看出，拒绝策略不仅仅是在饱和状态下使用，
29 在线程池进入到关闭阶段同样需要使用到；
30 上面的几行代码还不能完全清楚这个新增任务的过程，
31 肯定还需要清楚addWorker方法才行：
32 else if (!addWorker(command, false))
33 reject(command);

1、判断当前的线程数是否小于corePoolSize如果是，使用入参任务通过addWord方法创建一个新的线程，如果能完成新线程创建exexute方法结束，成功提交任务；

2、在第一步没有完成任务提交；状态为运行并且能成功加入任务到工作队列后，再进行一次 check，如果状态在任务加入队列后变为了非运行（有可能是在执行到这里线程池shutdown 了），非运行状态下当然是需要reject；然后再判断当前线程数是否为0（有可能这个时候线程数变为了0），如是，新增一个线程；

3、如果不能加入任务到工作队列，将尝试使用任务新增一个线程，如果失败，则是线程池已经 shutdown或者线程池已经达到饱和状态，所以reject; 从上面新增任务的execute方法也可以看出，拒绝策略不仅仅是在饱和状态下使用，在线程池进入到关闭阶段同样需要使用到；

addWorker方法:

1
2 private boolean addWorker(Runnable firstTask, boolean core) {
3 retry: goto写法 用于重试
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 &&
10 ! (rs == SHUTDOWN &&
11 firstTask == null &&
12 ! workQueue.isEmpty()))
13 线程状态非运行并且非shutdown状态任务为空，队列非空就不能新增线程了
14
15 return false;
16
17 for (;;) {
18 int wc = workerCountOf(c);
19 if (wc >= CAPACITY ||
20 wc >= (core ? corePoolSize : maximumPoolSize))
21 当前现场大于等于最大值
22 等于核心线程数 非核心大于等于线程池数 说明达到了阈值
23 最大线程数 就不新增线程
24 return false;
25 if (compareAndIncrementWorkerCount(c)) ctl+1 工作线程池数量+1 如果成功
26 就跳出死循环。
27 cas操作 如果为true 新增成功 退出
28 break retry;
29 c = ctl.get(); // Re‐read ctl
30 if (runStateOf(c) != rs)
31 continue retry; 进来的状态和此时的状态发生改变 重头开始 重试
32 // else CAS failed due to workerCount change; retry inner loop
33 }
34 }
35 上面主要是对ctl工作现场+1
36
37 boolean workerStarted = false;
38 boolean workerAdded = false;
39 Worker w = null;
40 try {
41 w = new Worker(firstTask); 内部类 封装了线程和任务 通过threadfactory创建线程
42
43 final Thread t = w.thread; 毎一个worker就是一个线程数
44 if (t != null) {
45 final ReentrantLock mainLock = this.mainLock;
46 mainLock.lock();
47 try {
48 // Recheck while holding lock.
49 // Back out on ThreadFactory failure or if
50 // shut down before lock acquired.
51 重新获取当前线程状态
52 int rs = runStateOf(ctl.get());
53 小于shutdown就是running状态
54 if (rs < SHUTDOWN ||
55 (rs == SHUTDOWN && firstTask == null)) {
56 SHUTDOWN 和firstTask 为空是从队列中处理任务 那就可以放到集合中
57 线程还没start 就是alive就直接异常
58 if (t.isAlive()) // precheck that t is startable
59 throw new IllegalThreadStateException();
60 workers.add(w);
61 int s = workers.size();
62 if (s > largestPoolSize)
63 largestPoolSize = s; 记录最大线程数
64 workerAdded = true;
65 }
66 } finally {
67 mainLock.unlock();
68 }
69 if (workerAdded) {
70 t.start(); 启动线程
71 workerStarted = true;
72 }
73 }
74 } finally {
75 if (! workerStarted)
76 addWorkerFailed(w);//失败回退 从wokers移除w 线程数减1 尝试结束线程池
77 }
78 return workerStarted;
79 }
1
2 private final class Worker
3 extends AbstractQueuedSynchronizer
4 implements Runnable
5 {
6 /**
7 * This class will never be serialized, but we provide a
8 * serialVersionUID to suppress a javac warning.
9 */
10 private static final long serialVersionUID = 6138294804551838833L;
11
12 /** Thread this worker is running in. Null if factory fails. */
13 正在运行woker线程
14 final Thread thread;
15 /** Initial task to run. Possibly null. */
16 传入的任务
17 Runnable firstTask;
18 /** Per‐thread task counter */
19 完成的任务数 监控用
20 volatile long completedTasks;
21
22 /**
23 * Creates with given first task and thread from ThreadFactory.
24 * @param firstTask the first task (null if none)
25 */
26 Worker(Runnable firstTask) {
27 禁止线程中断
28 setState(‐1); // inhibit interrupts until runWorker
29 this.firstTask = firstTask;
30 this.thread = getThreadFactory().newThread(this);
31 }
32
33 /** Delegates main run loop to outer runWorker */
34 public void run() {
35 runWorker(this);
36 }

runwoker方法：

1 final void runWorker(Worker w) {
2 Thread wt = Thread.currentThread();//获取当前线程
3 Runnable task = w.firstTask;
4 w.firstTask = null;
5 w.unlock(); // allow interrupts 把state从‐1改为0 意思是可以允许中断
6 boolean completedAbruptly = true;
7 try { task不为空 或者阻塞队列中拿到了任务
8 while (task != null || (task = getTask()) != null) {
9 w.lock();
10 // If pool is stopping, ensure thread is interrupted;
11 // if not, ensure thread is not interrupted. This
12 // requires a recheck in second case to deal with
13 // shutdownNow race while clearing interrupt
14 如果当前线程池状态等于stop 就中断
15 //Thread.interrupted() 中断标志
16 if ((runStateAtLeast(ctl.get(), STOP) ||
17 (Thread.interrupted() &&
18 runStateAtLeast(ctl.get(), STOP))) &&
19 !wt.isInterrupted())
20 wt.interrupt();
21 try {
22 beforeExecute(wt, task);
23 Throwable thrown = null;
24 try {
25 task.run();
26 } catch (RuntimeException x) {
27 thrown = x; throw x;
28 } catch (Error x) {
29 thrown = x; throw x;
30 } catch (Throwable x) {
31 thrown = x; throw new Error(x);
32 } finally {
33 afterExecute(task, thrown);
34 }
35 } finally {
36 task = null; 这设置为空 等下次循环就会从队列里面获取
37 w.completedTasks++; 完成任务数+1
38 w.unlock();
39 }
40 }
41 completedAbruptly = false;
42 } finally {
43 processWorkerExit(w, completedAbruptly);
44 }
45 }

getTask方法：

1
2 private Runnable getTask() {
3 boolean timedOut = false; // Did the last poll() time out?
4
5 for (;;) {
6 int c = ctl.get();
7 int rs = runStateOf(c);//获取线程池运行状态
8
9 shuitdown或者weikong 那就工作现场‐1 同事返回为null
10 // Check if queue empty only if necessary.
11 if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
12 decrementWorkerCount();
13 return null;
14 }
15 重新获取工作线程数
16 int wc = workerCountOf(c);
17 timed是标志超时销毁
18 allowCoreThreadTimeOut true 核心线程池也是可以销毁的
19 // Are workers subject to culling?
20 boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;
21
22 if ((wc > maximumPoolSize || (timed && timedOut))
23 && (wc > 1 || workQueue.isEmpty())) {
24 if (compareAndDecrementWorkerCount(c))
25 return null;
26 continue;
27 }
28
29 try {
30 Runnable r = timed ?
31 workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
32 workQueue.take();
33 if (r != null)
34 return r;
35 timedOut = true;
36 } catch (InterruptedException retry) {
37 timedOut = false;
38 }
39 }
40 }

processWorkerExit方法：

1 private void processWorkerExit(Worker w, boolean completedAbruptly) {
2 if (completedAbruptly) // If abrupt, then workerCount wasn't adjusted
3 decrementWorkerCount();
4
5 final ReentrantLock mainLock = this.mainLock;
6 mainLock.lock();
7 try {
8 completedTaskCount += w.completedTasks;
9 workers.remove(w);
10 } finally {
11 mainLock.unlock();
12 }
13
14 tryTerminate();
15
16 int c = ctl.get();
17 if (runStateLessThan(c, STOP)) {
18 if (!completedAbruptly) {
19 int min = allowCoreThreadTimeOut ? 0 : corePoolSize;
20 if (min == 0 && ! workQueue.isEmpty())
21 min = 1;
22 if (workerCountOf(c) >= min)
23 return; // replacement not needed
24 }
25 addWorker(null, false);
26 }
27 }

ThreadPoolExecutor内部有实现4个拒绝策略：

(1)、CallerRunsPolicy，由调用execute方法提交任务的线程来执行这个任务；

(2)、AbortPolicy，抛出异常RejectedExecutionException拒绝提交任务；

(3)、DiscardPolicy，直接抛弃任务，不做任何处理；

(4)、DiscardOldestPolicy，去除任务队列中的第一个任务（最旧的），重新提交；

一般使用自定义的拒绝策略

ScheduledThreadPoolExecutor： schedule：延迟多长时间之后只执行一次； scheduledAtFixedRate固定：延迟指定时间后执行一次，之后按照固定的时长周期执行； scheduledWithFixedDelay非固定：延迟指定时间后执行一次，之后按照：上一次任务执行时长 + 周期的时长的时间去周期执行；

1 private void delayedExecute(RunnableScheduledFuture<?> task) {
2 //如果线程池不是RUNNING状态，则使用拒绝策略把提交任务拒绝掉
3 if (isShutdown())
4 reject(task);
5 else {
6 //与ThreadPoolExecutor不同，这里直接把任务加入延迟队列
7 super.getQueue().add(task);
8 //如果当前状态无法执行任务，则取消
9 if (isShutdown() &&
10 !canRunInCurrentRunState(task.isPeriodic()) &&
11 remove(task))
12 task.cancel(false);
13 else
14 //和ThreadPoolExecutor不一样，corePoolSize没有达到会增加Worker;
15 //增加Worker，确保提交的任务能够被执行
16 ensurePrestart();
17 }
18 }

offer方法：

1 public boolean offer(Runnable x) {
2 if (x == null)
3 throw new NullPointerException();
4 RunnableScheduledFuture<?> e = (RunnableScheduledFuture<?>)x;
5 final ReentrantLock lock = this.lock;
6 lock.lock();
7 try {
8 int i = size;
9 if (i >= queue.length)
10 // 容量扩增50%。
11 grow();
12 size = i + 1;
13 // 第一个元素,其实这里也可以统一进行sift‐up操作,没必要特判。
14 if (i == 0) {
15 queue[0] = e;
16 setIndex(e, 0);
17 } else {
18 // 插入堆尾。
19 siftUp(i, e);
20 }
21 // 如果新加入的元素成为了堆顶,则原先的leader就无效了。
22 if (queue[0] == e) {
23 leader = null;
24 // 由于原先leader已经无效被设置为null了,这里随便唤醒一个线程(未必是原先的leader)来
取走堆顶任务。
25 available.signal();
26 }
27 } finally {
28 lock.unlock();
29 }
30 return true;
31 }

siftup方法：

1 private void siftUp(int k, RunnableScheduledFuture<?> key) {
2 // 找到父节点的索引
3 while (k > 0) {
4 // 获取父节点
5 int parent = (k ‐ 1) >>> 1;
6 RunnableScheduledFuture<?> e = queue[parent];
7 // 如果key节点的执行时间大于父节点的执行时间，不需要再排序了
8 if (key.compareTo(e) >= 0)
9 break;
10 // 如果key.compareTo(e) < 0，
11 说明key节点的执行时间小于父节点的执行时间，需要把父节点移到后面
12 queue[k] = e;
13 setIndex(e, k);
14 // 设置索引为k
15 k = parent;
16 }
17 // key设置为排序后的位置中
18 queue[k] = key;
19 setIndex(key, k);
20 }

任务执行：

1 public void run() {
2 // 是否周期性，就是判断period是否为0。
3 boolean periodic = isPeriodic();
4 // 检查任务是否可以被执行。
5 if (!canRunInCurrentRunState(periodic))
6 cancel(false);
7 // 如果非周期性任务直接调用run运行即可。
8 else if (!periodic)
9 ScheduledFutureTask.super.run();
10 // 如果成功runAndRest，则设置下次运行时间并调用reExecutePeriodic。
11 else if (ScheduledFutureTask.super.runAndReset()) {
12 setNextRunTime();
13 // 需要重新将任务(outerTask)放到工作队列中。此方法源码会在后文介绍ScheduledThread
PoolExecutor本身API时提及。
14 reExecutePeriodic(outerTask);
15 }
16 }

fied-rate模式和fixed-delay模式区别:

1 private void setNextRunTime() {
2 long p = period;
3 /*
4 * fixed‐rate模式，时间设置为上一次时间+p。
5 * 提一句，这里的时间其实只是可以被执行的最小时间，不代表到点就要执行。
6 * 如果这次任务还没执行完是肯定不会执行下一次的。
7 */
8 if (p > 0)
9 time += p;
10 /**
11 * fixed‐delay模式，计算下一次任务可以被执行的时间。
12 * 简单来说差不多就是当前时间+delay值。因为代码走到这里任务就已经结束了,now()可以认
为就是任务结束时间。
13 */
14 else
15 time = triggerTime(‐p);
16 }
17
18 long triggerTime(long delay) {
19 /*
20 * 如果delay < Long.Max_VALUE/2,则下次执行时间为当前时间+delay。
21 *
22 * 否则为了避免队列中出现由于溢出导致的排序紊乱,需要调用overflowFree来修正一下
delay(如果有必要的话)。
23 */
24 return now() + ((delay < (Long.MAX_VALUE >> 1)) ? delay :
overflowFree(delay));
25 }
26
27 /**
28 * 主要就是有这么一种情况：
29 * 某个任务的delay为负数，说明当前可以执行(其实早该执行了)。
30 * 工作队列中维护任务顺序是基于compareTo的，在compareTo中比较两个任务的顺序会用time
相减，负数则说明优先级高。
31 *
32 * 那么就有可能出现一个delay为正数,减去另一个为负数的delay，结果上溢为负数，则会导致
compareTo产生错误的结果。
33 *
34 * 为了特殊处理这种情况，首先判断一下队首的delay是不是负数，如果是正数不用管了,怎么
减都不会溢出。
35 * 否则可以拿当前delay减去队首的delay来比较看，如果不出现上溢，则整个队列都ok，排序
不会乱。
36 * 不然就把当前delay值给调整为Long.MAX_VALUE + 队首delay。
37 */
38 private long overflowFree(long delay) {
39 Delayed head = (Delayed) super.getQueue().peek();
40 if (head != null) {
41 long headDelay = head.getDelay(NANOSECONDS);
42 if (headDelay < 0 && (delay ‐ headDelay < 0))
43 delay = Long.MAX_VALUE + headDelay;
44 }
45 return delay;
46 }