[Java]线程池
在没有看任何代码之前首先想一下线程池应该有哪几部分:
- 任务队列
- 线程
任务队列很好办,直接用阻塞队列就可以了: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 ---------- ---------- ---------- ----------
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。