En " [Alta simultaneidad] ¡Deje de crear problemas, por lo que es correcto comprender el proceso central de ejecutar tareas en el grupo de subprocesos! ! "En este artículo, analizamos en profundidad el proceso central de ejecución de tareas en el grupo de subprocesos. En el método addWorker(Runnable, boolean) de la clase ThreadPoolExecutor, después de usar CAS para actualizar de forma segura la cantidad de subprocesos, el siguiente paso es crear un nuevo subproceso Worker para ejecutar la tarea, entonces, primero analicemos el código fuente de la clase Worker.
Análisis de la clase obrera
La clase Worker, en términos de estructura de clase, hereda AQS (clase AbstractQueuedSynchronizer) e implementa la interfaz Runnable. Esencialmente, la clase Worker es tanto un componente de sincronización como un subproceso que ejecuta tareas. A continuación, veamos el código fuente de la clase Worker, como se muestra a continuación.
private final class Worker extends AbstractQueuedSynchronizer implements Runnable {
private static final long serialVersionUID = 6138294804551838833L;
//执行任务的线程类
final Thread thread;
//初始化执行的任务，第一次执行的任务
Runnable firstTask;
//完成任务的计数
volatile long completedTasks;
//Worker类的构造方法，初始化任务并调用线程工厂创建执行任务的线程
Worker(Runnable firstTask) {
setState(-1);
this.firstTask = firstTask;
this.thread = getThreadFactory().newThread(this);
}
//重写Runnable接口的run()方法
public void run() {
//调用ThreadPoolExecutor类的runWorker(Worker)方法
runWorker(this);
}
//检测是否是否获取到锁
//state=0表示未获取到锁
//state=1表示已获取到锁
protected boolean isHeldExclusively() {
return getState() != 0;
}
//使用AQS设置线程状态
protected boolean tryAcquire(int unused) {
if (compareAndSetState(0, 1)) {
setExclusiveOwnerThread(Thread.currentThread());
return true;
}
return false;
}
//尝试释放锁
protected boolean tryRelease(int unused) {
setExclusiveOwnerThread(null);
setState(0);
return true;
}
public void lock() { acquire(1); }
public boolean tryLock() { return tryAcquire(1); }
public void unlock() { release(1); }
public boolean isLocked() { return isHeldExclusively(); }
void interruptIfStarted() {
Thread t;
if (getState() >= 0 && (t = thread) != null && !t.isInterrupted()) {
try {
t.interrupt();
} catch (SecurityException ignore) {
}
}
}
}
En el constructor de la clase Worker, se puede ver que el estado de sincronización se establece primero en -1, que se establece en -1 para evitar que el método runWorker se interrumpa antes de ejecutarse. Esto se debe a que si otros subprocesos llaman al método shutdownNow() del grupo de subprocesos, si el valor del estado del estado en la clase Worker es mayor que 0, el subproceso se interrumpirá y si el valor del estado del estado es -1 , el hilo no se interrumpirá.
La clase Worker implementa la interfaz Runnable y necesita reescribir el método de ejecución. El método de ejecución de Worker esencialmente llama al método runWorker de la clase ThreadPoolExecutor. En el método runWorker, el método de desbloqueo se llamará primero, lo que establecerá el estado en 0, por lo que en este momento, llamar al método shutdownDownNow interrumpirá el subproceso actual, y el método runWork se ingresó en este momento, por lo que el subproceso no se interrumpirá antes de que se ejecute el método runWorker.
Nota: debe concentrarse en comprender la implementación de la clase Worker.
El método runWorker(Worker) de la clase ThreadPoolExecutor se llama en la clase Worker. A continuación, echemos un vistazo a la implementación del método runWorker(Worker) de la clase ThreadPoolExecutor.
runWorker(Worker)方法
首先，我们看下RunWorker(Worker)方法的源码，如下所示。
final void runWorker(Worker w) {
Thread wt = Thread.currentThread();
Runnable task = w.firstTask;
w.firstTask = null;
//释放锁，将state设置为0,允许中断任务的执行
w.unlock();
boolean completedAbruptly = true;
try {
//如果任务不为空，或者从任务队列中获取的任务不为空，则执行while循环
while (task != null || (task = getTask()) != null) {
//如果任务不为空，则获取Worker工作线程的独占锁
w.lock();
//如果线程已经停止，或者中断线程后线程终止并且没有成功中断线程
//大家好好理解下这个逻辑
if ((runStateAtLeast(ctl.get(), STOP) ||
(Thread.interrupted() &&
runStateAtLeast(ctl.get(), STOP))) &&
!wt.isInterrupted())
//中断线程
wt.interrupt();
try {
//执行任务前执行的逻辑
beforeExecute(wt, task);
Throwable thrown = null;
try {
//调用Runable接口的run方法执行任务
task.run();
} catch (RuntimeException x) {
thrown = x; throw x;
} catch (Error x) {
thrown = x; throw x;
} catch (Throwable x) {
thrown = x; throw new Error(x);
} finally {
//执行任务后执行的逻辑
afterExecute(task, thrown);
}
} finally {
//任务执行完成后，将其设置为空
task = null;
//完成的任务数量加1
w.completedTasks++;
//释放工作线程获得的锁
w.unlock();
}
}
completedAbruptly = false;
} finally {
//执行退出Worker线程的逻辑
processWorkerExit(w, completedAbruptly);
}
}
这里，我们拆解runWorker(Worker)方法。
（1）获取当前线程的句柄和工作线程中的任务，并将工作线程中的任务设置为空，执行unlock方法释放锁，将state状态设置为0，此时可以中断工作线程，代码如下所示。
Thread wt = Thread.currentThread();
Runnable task = w.firstTask;
w.firstTask = null;
//释放锁，将state设置为0,允许中断任务的执行
w.unlock();
（2）在while循环中进行判断，如果任务不为空，或者从任务队列中获取的任务不为空，则执行while循环，否则，调用processWorkerExit(Worker, boolean)方法退出Worker工作线程。
while (task != null || (task = getTask()) != null)
（3）如果满足while的循环条件，首先获取工作线程内部的独占锁，并执行一系列的逻辑判断来检测是否需要中断当前线程的执行，代码如下所示。
//如果任务不为空，则获取Worker工作线程的独占锁
w.lock();
//如果线程已经停止，或者中断线程后线程终止并且没有成功中断线程
//大家好好理解下这个逻辑
if ((runStateAtLeast(ctl.get(), STOP) ||
(Thread.interrupted() &&
runStateAtLeast(ctl.get(), STOP))) &&
!wt.isInterrupted())
//中断线程
wt.interrupt();
（4）调用执行任务前执行的逻辑，如下所示
//执行任务前执行的逻辑
beforeExecute(wt, task);
（5）调用Runable接口的run方法执行任务
//调用Runable接口的run方法执行任务
task.run();
（6）调用执行任务后执行的逻辑
//执行任务后执行的逻辑
afterExecute(task, thrown);
（7）将完成的任务设置为空，完成的任务数量加1并释放工作线程的锁。
//任务执行完成后，将其设置为空
task = null;
//完成的任务数量加1
w.completedTasks++;
//释放工作线程获得的锁
w.unlock();
（8）退出Worker线程的执行，如下所示
//执行退出Worker线程的逻辑
processWorkerExit(w, completedAbruptly);
从代码分析上可以看到，当从Worker线程中获取的任务为空时，会调用getTask()方法从任务队列中获取任务，接下来，我们看下getTask()方法的实现。
getTask()方法
我们先来看下getTask()方法的源代码，如下所示。
private Runnable getTask() {
//轮询是否超时的标识
boolean timedOut = false;
//自旋for循环
for (;;) {
//获取ctl
int c = ctl.get();
//获取线程池的状态
int rs = runStateOf(c);
//检测任务队列是否在线程池停止或关闭的时候为空
//也就是说任务队列是否在线程池未正常运行时为空
if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
//减少Worker线程的数量
decrementWorkerCount();
return null;
}
//获取线程池中线程的数量
int wc = workerCountOf(c);
//检测当前线程池中的线程数量是否大于corePoolSize的值或者是否正在等待执行任务
boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;
//如果线程池中的线程数量大于corePoolSize
//获取大于corePoolSize或者是否正在等待执行任务并且轮询超时
//并且当前线程池中的线程数量大于1或者任务队列为空
if ((wc > maximumPoolSize || (timed && timedOut))
&& (wc > 1 || workQueue.isEmpty())) {
//成功减少线程池中的工作线程数量
if (compareAndDecrementWorkerCount(c))
return null;
continue;
}
try {
//从任务队列中获取任务
Runnable r = timed ?
workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
workQueue.take();
//任务不为空直接返回任务
if (r != null)
return r;
timedOut = true;
} catch (InterruptedException retry) {
timedOut = false;
}
}
}
getTask()方法的逻辑比较简单，大家看源码就可以了，我这里就不重复描述了。
接下来，我们看下在正式调用Runnable的run()方法前后，执行的beforeExecute方法和afterExecute方法。
beforeExecute(Thread, Runnable)方法
beforeExecute(Thread, Runnable)方法的源代码如下所示。
protected void beforeExecute(Thread t, Runnable r) { }
可以看到，beforeExecute(Thread, Runnable)方法的方法体为空，我们可以创建ThreadPoolExecutor的子类来重写beforeExecute(Thread, Runnable)方法，使得线程池正式执行任务之前，执行我们自己定义的业务逻辑。
afterExecute(Runnable, Throwable)方法
afterExecute(Runnable, Throwable)方法的源代码如下所示。
protected void afterExecute(Runnable r, Throwable t) { }
可以看到，afterExecute(Runnable, Throwable)方法的方法体同样为空，我们可以创建ThreadPoolExecutor的子类来重写afterExecute(Runnable, Throwable)方法，使得线程池在执行任务之后执行我们自己定义的业务逻辑。
接下来，就是退出工作线程的processWorkerExit(Worker, boolean)方法。
processWorkerExit(Worker, boolean)方法
processWorkerExit(Worker, boolean)方法的逻辑主要是执行退出Worker线程，并且对一些资源进行清理，源代码如下所示。
private void processWorkerExit(Worker w, boolean completedAbruptly) {
//执行过程中出现了异常，突然中断
if (completedAbruptly)
//将工作线程的数量减1
decrementWorkerCount();
//获取全局锁
final ReentrantLock mainLock = this.mainLock;
mainLock.lock();
try {
//累加完成的任务数量
completedTaskCount += w.completedTasks;
//将完成的任务从workers集合中移除
workers.remove(w);
} finally {
//释放锁
mainLock.unlock();
}
//尝试终止工作线程的执行
tryTerminate();
//获取ctl
int c = ctl.get();
//判断当前线程池的状态是否小于STOP（RUNNING或者SHUTDOWN）
if (runStateLessThan(c, STOP)) {
//如果没有突然中断完成
if (!completedAbruptly) {
//如果allowCoreThreadTimeOut为true，为min赋值为0，否则赋值为corePoolSize
int min = allowCoreThreadTimeOut ? 0 : corePoolSize;
//如果min为0并且工作队列不为空
if (min == 0 && ! workQueue.isEmpty())
//min的值设置为1
min = 1;
//如果线程池中的线程数量大于min的值
if (workerCountOf(c) >= min)
//返回，不再执行程序
return;
}
//调用addWorker方法
addWorker(null, false);
}
}
接下来，我们拆解processWorkerExit(Worker, boolean)方法。
（1）执行过程中出现了异常，突然中断执行，则将工作线程数量减1，如下所示。
//执行过程中出现了异常，突然中断
if (completedAbruptly)
//将工作线程的数量减1
decrementWorkerCount();
（2）获取锁累加完成的任务数量，并将完成的任务从workers集合中移除，并释放，如下所示。
//获取全局锁
final ReentrantLock mainLock = this.mainLock;
mainLock.lock();
try {
//累加完成的任务数量
completedTaskCount += w.completedTasks;
//将完成的任务从workers集合中移除
workers.remove(w);
} finally {
//释放锁
mainLock.unlock();
}
（3）尝试终止工作线程的执行
//尝试终止工作线程的执行
tryTerminate();
（4）处判断当前线程池中的线程个数是否小于核心线程数，如果是，需要新增一个线程保证有足够的线程可以执行任务队列中的任务或者提交的任务。
//获取ctl
int c = ctl.get();
//判断当前线程池的状态是否小于STOP（RUNNING或者SHUTDOWN）
if (runStateLessThan(c, STOP)) {
//如果没有突然中断完成
if (!completedAbruptly) {
//如果allowCoreThreadTimeOut为true，为min赋值为0，否则赋值为corePoolSize
int min = allowCoreThreadTimeOut ? 0 : corePoolSize;
//如果min为0并且工作队列不为空
if (min == 0 && ! workQueue.isEmpty())
//min的值设置为1
min = 1;
//如果线程池中的线程数量大于min的值
if (workerCountOf(c) >= min)
//返回，不再执行程序
return;
}
//调用addWorker方法
addWorker(null, false);
}
接下来，我们看下tryTerminate()方法。
tryTerminate()方法
tryTerminate()方法的源代码如下所示。
final void tryTerminate() {
//自旋for循环
for (;;) {
//获取ctl
int c = ctl.get();
//如果线程池的状态为RUNNING
//或者状态大于TIDYING
//或者状态为SHUTDOWN并且任务队列为空
//直接返回程序，不再执行后续逻辑
if (isRunning(c) ||
runStateAtLeast(c, TIDYING) ||
(runStateOf(c) == SHUTDOWN && ! workQueue.isEmpty()))
return;
//如果当前线程池中的线程数量不等于0
if (workerCountOf(c) != 0) {
//中断线程的执行
interruptIdleWorkers(ONLY_ONE);
return;
}
//获取线程池的全局锁
final ReentrantLock mainLock = this.mainLock;
mainLock.lock();
try {
//通过CAS将线程池的状态设置为TIDYING
if (ctl.compareAndSet(c, ctlOf(TIDYING, 0))) {
try {
//调用terminated()方法
terminated();
} finally {
//将线程池状态设置为TERMINATED
ctl.set(ctlOf(TERMINATED, 0));
//唤醒所有因为调用线程池的awaitTermination方法而被阻塞的线程
termination.signalAll();
}
return;
}
} finally {
//释放锁
mainLock.unlock();
}
}
}
（1）获取ctl，根据情况设置线程池状态或者中断线程的执行，并返回。
//获取ctl
int c = ctl.get();
//如果线程池的状态为RUNNING
//或者状态大于TIDYING
//或者状态为SHUTDOWN并且任务队列为空
//直接返回程序，不再执行后续逻辑
if (isRunning(c) ||
runStateAtLeast(c, TIDYING) ||
(runStateOf(c) == SHUTDOWN && ! workQueue.isEmpty()))
return;
//如果当前线程池中的线程数量不等于0
if (workerCountOf(c) != 0) {
//中断线程的执行
interruptIdleWorkers(ONLY_ONE);
return;
}
(2) Adquiera el bloqueo global, establezca el estado del grupo de subprocesos a través de CAS, llame al método terminado () para ejecutar la lógica, finalmente establezca el estado del grupo de subprocesos en TERMINADO, despierte todos los subprocesos bloqueados llamando a awaitTermination método del grupo de subprocesos y, finalmente, suelte el bloqueo, como se muestra a continuación.
//获取线程池的全局
final ReentrantLock mainLock = this.mainLock;
mainLock.lock();
try {
//通过CAS将线程池的状态设置为TIDYING
if (ctl.compareAndSet(c, ctlOf(TIDYING, 0))) {
try {
//调用terminated()方法
terminated();
} finally {
//将线程池状态设置为TERMINATED
ctl.set(ctlOf(TERMINATED, 0));
//唤醒所有因为调用线程池的awaitTermination方法而被阻塞的线程
termination.signalAll();
}
return;
}
} finally {
//释放锁
mainLock.unlock();
}
A continuación, mire el método terminado().
método terminado ()
El código fuente del método terminado() se muestra a continuación.
protected void terminated() { }
Se puede ver que el cuerpo del método del método terminado () está vacío. Podemos crear una subclase de ThreadPoolExecutor para reescribir el método terminado (). Vale la pena que el subproceso de trabajo ejecute la lógica comercial del método terminado (). definido por nosotros cuando se llama al método tryTerminate().
Bien, detengámonos aquí hoy, soy Glacier, hasta la próxima~~
Estoy participando en el reclutamiento del programa de firma de creadores de la Comunidad Tecnológica de Nuggets, haga clic en el enlace para registrarse y enviar .