在没有看任何代码之前首先想一下线程池应该有哪几部分:
- 任务队列
- 线程
任务队列很好办,直接用阻塞队列就可以了:BlockingQueue<Runnable> workQueue。而线程是用来执行任务的,那么理所当然应该是不断地从任务队列中取出任务来执行,我们来看ThreadPoolExecutor中的Worker的实现:
private final class Worker implements Runnable {
private final ReentrantLock runLock = new ReentrantLock();
private Runnable firstTask;
volatile long completedTasks;
Thread thread;
// 执行task
private void runTask(Runnable task) {
final ReentrantLock runLock = this.runLock;
runLock.lock();
try {
// 检查线程池的状态,判断是否中断
if (runState < STOP && Thread.interrupted() && runState >= STOP)
thread.interrupt();
boolean ran = false;
beforeExecute(thread, task);// 执行任务前的操作
try {
task.run(); // 任务开始执行
ran = true;
afterExecute(task, null);// 执行任务后的操作
++completedTasks;
} catch (RuntimeException ex) {
if (!ran)
afterExecute(task, ex);// 执行任务后的操作
throw ex;
}
} finally {
runLock.unlock();
}
}
// 线程启动之后开始执行
public void run() {
try {
Runnable task = firstTask;
firstTask = null;
// 从线程池中取出Runnable然后执行
while (task != null || (task = getTask()) != null) {
runTask(task);
task = null;
}
} finally {
workerDone(this);
}
}
}
通过判断runLock是不是锁上的状态就可以判断Worker是否在执行任务了。当然getTask失败的时候,Worker的任务也就结束了,我们有时候会希望线程去等一段时间,如果这段时间里面没有任务到达线程才退出。这个在阻塞队列中有线程的方法,在getTask可以看到,如下:
r = workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS);
在拿到任务的时候,如果现在线程池还很小,那么现在不需要将任务放到阻塞队列里面去,直接创建一个线程运行就可以了,在这条路走不通的时候才会“绕弯路”来做。因为阻塞队列也是有容量限制的,那么在尝试将任务放进去的时候可能会失败。如果让我写的话就不会尝试了,而是直接调用阻塞方法让主线程阻塞在这里,这样其实并不好:
- 不够灵活
- 调用的线程被阻塞了,浪费资源,其实它是可以用来执行任务的
如果自己要实现reject策略的话实现下面接口就可以了:
public interface RejectedExecutionHandler {
void rejectedExecution(Runnable r, ThreadPoolExecutor executor);
}
有可能出现一种情况:插入任务之后线程池的状态改变了,那么也要保证该任务能够被处理(并不一定是任务被完成,也可能是拒绝),那么下面来看提交任务的完整逻辑:
public void execute(Runnable command) {
if (command == null)
throw new NullPointerException();
// 尝试创建线程并执行任务
if (poolSize >= corePoolSize || !addIfUnderCorePoolSize(command)) {
// 尝试将线程放入阻塞队列
if (runState == RUNNING && workQueue.offer(command)) {
if (runState != RUNNING || poolSize == 0)
// 确保任务会被处理
ensureQueuedTaskHandled(command);
} else if (!addIfUnderMaximumPoolSize(command))// 如果现在能创建一个线程执行该任务的话,就不要拒绝它了。。。
reject(command);
}
}
可以发现在线程池运行期间修改各种阀值都是可以起到作用的。如果在spring中想用简单的线程池的话没有必要自己写一个,直接用现成的配置一个就可以了,比如:
<bean id="taskExecutor" class="org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor">
<property name="corePoolSize" value="5" />
<property name="maxPoolSize" value="10" />
<property name="queueCapacity" value="25" />
</bean>
---------- ---------- ---------- ---------- END ---------- ---------- ---------- ----------