大家好,我是冰河~~
最近有不少小伙伴私信我说:看了我在【精通高并发系列】文章中写的深度解析线程池源码部分的文章,但是还是有些不明白线程池的实现原理。问我能不能手写一个简单的线程池,帮助读者深刻理解线程池的原理。
这不,我熬夜肝了这篇文章。
在【精通高并发系列】的文章中,我们曾经深度解析过线程池的源码,从源码层面深度解析了线程池的实现原理。
其实,源码是原理落地的最直接体现,看懂源码对于深刻理解原理有着很大的帮助。但是不少小伙伴看源码时,总觉得源码太枯燥了,看不懂。
那今天,我们就一起花10分钟手撸一个极简版的Java线程池,让小伙伴们更好的理解线程池的核心原理。
本文的整体结构如下所示。
Java线程池核心原理看过Java线程池源码的小伙伴都知道,在Java线程池中最核心的类就是ThreadPoolExecutor,而在ThreadPoolExecutor类中最核心的构造方法就是带有7个参数的构造方法,如下所示。
public ThreadPoolExecutor(int corePoolSize,
int maximumPoolSize,
long keepAliveTime,
TimeUnit unit,
BlockingQueue workQueue,
ThreadFactory threadFactory,
RejectedExecutionHandler handler)
各参数的含义如下所示。
corePoolSize:线程池中的常驻核心线程数。maximumPoolSize:线程池能够容纳同时执行的最大线程数,此值大于等于1。keepAliveTime:多余的空闲线程存活时间,当空间时间达到keepAliveTime值时,多余的线程会被销毁直到只剩下corePoolSize个线程为止。unit:keepAliveTime的单位。workQueue:任务队列,被提交但尚未被执行的任务。threadFactory:表示生成线程池中工作线程的线程工厂,用户创建新线程,一般用默认即可。handler:拒绝策略,表示当线程队列满了并且工作线程大于等于线程池的最大显示数(maxnumPoolSize)时,如何来拒绝请求执行的runnable的策略。
并且Java的线程池是通过 生产者-消费者模式 实现的,线程池的使用方是生产者,而线程池本身就是消费者。
Java线程池的核心工作流程如下图所示。
手撸Java线程池我们自己手动实现的线程池要比Java自身的线程池简单的多,我们去掉了各种复杂的处理方式,只保留了最核心的原理:线程池的使用者向任务队列中添加任务,而线程池本身从任务队列中消费任务并执行任务。
只要理解了这个核心原理,接下来的代码就简单多了。在实现这个简单的线程池时,我们可以将整个实现过程进行拆解。拆解后的实现流程为:定义核心字段、创建内部类WorkThread、创建ThreadPool类的构造方法和创建执行任务的方法。
定义核心字段首先,我们创建一个名称为ThreadPool的Java类,并在这个类中定义如下核心字段。
DEFAULT_WORKQUEUE_SIZE:静态常量,表示默认的阻塞队列大小。workQueue:模拟实际的线程池使用阻塞队列来实现生产者-消费者模式。workThreads:模拟实际的线程池使用List集合保存线程池内部的工作线程。
核心代码如下所示。
//默认阻塞队列大小 private static final int DEFAULT_WORKQUEUE_SIZE = 5; //模拟实际的线程池使用阻塞队列来实现生产者-消费者模式 private BlockingQueue创建内部类WordThreadworkQueue; //模拟实际的线程池使用List集合保存线程池内部的工作线程 private List workThreads = new ArrayList ();
在ThreadPool类中创建一个内部类WorkThread,模拟线程池中的工作线程。主要的作用就是消费workQueue中的任务,并执行任务。由于工作线程需要不断从workQueue中获取任务,所以,这里使用了while(true)循环不断尝试消费队列中的任务。
核心代码如下所示。
//内部类WorkThread,模拟线程池中的工作线程
//主要的作用就是消费workQueue中的任务,并执行
//由于工作线程需要不断从workQueue中获取任务,使用了while(true)循环不断尝试消费队列中的任务
class WorkThread extends Thread{
@Override
public void run() {
//不断循环获取队列中的任务
while (true){
//当没有任务时,会阻塞
try {
Runnable workTask = workQueue.take();
workTask.run();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
}
创建ThreadPool类的构造方法
这里,我们为ThreadPool类创建两个构造方法,一个构造方法中传入线程池的容量大小和阻塞队列,另一个构造方法中只传入线程池的容量大小。
核心代码如下所示。
//在ThreadPool的构造方法中传入线程池的大小和阻塞队列 public ThreadPool(int poolSize, BlockingQueue创建执行任务的方法workQueue){ this.workQueue = workQueue; //创建poolSize个工作线程并将其加入到workThreads集合中 IntStream.range(0, poolSize).forEach((i) -> { WorkThread workThread = new WorkThread(); workThread.start(); workThreads.add(workThread); }); } //在ThreadPool的构造方法中传入线程池的大小 public ThreadPool(int poolSize){ this(poolSize, new linkedBlockingQueue<>(DEFAULT_WORKQUEUE_SIZE)); }
在ThreadPool类中创建执行任务的方法execute(),execute()方法的实现比较简单,就是将方法接收到的Runnable任务加入到workQueue队列中。
核心代码如下所示。
//通过线程池执行任务
public void execute(Runnable task){
try {
workQueue.put(task);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
完整源码
这里,我们给出手动实现的ThreadPool线程池的完整源代码,如下所示。
package io.binghe.thread.pool;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.linkedBlockingQueue;
import java.util.stream.IntStream;
public class ThreadPool {
//默认阻塞队列大小
private static final int DEFAULT_WORKQUEUE_SIZE = 5;
//模拟实际的线程池使用阻塞队列来实现生产者-消费者模式
private BlockingQueue workQueue;
//模拟实际的线程池使用List集合保存线程池内部的工作线程
private List workThreads = new ArrayList();
//在ThreadPool的构造方法中传入线程池的大小和阻塞队列
public ThreadPool(int poolSize, BlockingQueue workQueue){
this.workQueue = workQueue;
//创建poolSize个工作线程并将其加入到workThreads集合中
IntStream.range(0, poolSize).forEach((i) -> {
WorkThread workThread = new WorkThread();
workThread.start();
workThreads.add(workThread);
});
}
//在ThreadPool的构造方法中传入线程池的大小
public ThreadPool(int poolSize){
this(poolSize, new linkedBlockingQueue<>(DEFAULT_WORKQUEUE_SIZE));
}
//通过线程池执行任务
public void execute(Runnable task){
try {
workQueue.put(task);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
//内部类WorkThread,模拟线程池中的工作线程
//主要的作用就是消费workQueue中的任务,并执行
//由于工作线程需要不断从workQueue中获取任务,使用了while(true)循环不断尝试消费队列中的任务
class WorkThread extends Thread{
@Override
public void run() {
//不断循环获取队列中的任务
while (true){
//当没有任务时,会阻塞
try {
Runnable workTask = workQueue.take();
workTask.run();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
}
}
没错,我们仅仅用了几十行Java代码就实现了一个极简版的Java线程池,没错,这个极简版的Java线程池的代码却体现了Java线程池的核心原理。
接下来,我们测试下这个极简版的Java线程池。
编写测试程序测试程序也比较简单,就是通过在main()方法中调用ThreadPool类的构造方法,传入线程池的大小,创建一个ThreadPool类的实例,然后循环10次调用ThreadPool类的execute()方法,向线程池中提交的任务为:打印当前线程的名称--->> Hello ThreadPool。
整体测试代码如下所示。
package io.binghe.thread.pool.test;
import io.binghe.thread.pool.ThreadPool;
import java.util.stream.IntStream;
public class ThreadPoolTest {
public static void main(String[] args){
ThreadPool threadPool = new ThreadPool(10);
IntStream.range(0, 10).forEach((i) -> {
threadPool.execute(() -> {
System.out.println(Thread.currentThread().getName() + "--->> Hello ThreadPool");
});
});
}
}
接下来,运行ThreadPoolTest类的main()方法,会输出如下信息。
Thread-0--->> Hello ThreadPool Thread-9--->> Hello ThreadPool Thread-5--->> Hello ThreadPool Thread-8--->> Hello ThreadPool Thread-4--->> Hello ThreadPool Thread-1--->> Hello ThreadPool Thread-2--->> Hello ThreadPool Thread-5--->> Hello ThreadPool Thread-9--->> Hello ThreadPool Thread-0--->> Hello ThreadPool
至此,我们自定义的Java线程池就开发完成了。
总结线程池的核心原理其实并不复杂,只要我们耐心的分析,深入其源码理解线程池的核心本质,你就会发现线程池的设计原来是如此的优雅。希望通过这个手写线程池的小例子,能够让你更好的理解线程池的核心原理。
好了,今天就到这儿吧,如果小伙伴们有啥问题可以在文末留言讨论,我是冰河,我们下期见~~
写在最后如果你想进大厂,想升职加薪,或者对自己现有的工作比较迷茫,都可以私信我交流,希望我的一些经历能够帮助到大家~~
推荐阅读:
《实践出真知:全网最强秒杀系统架构解密,不是所有的秒杀都是秒杀!!》《从零到上亿用户,我是如何一步步优化MySQL数据库的?(建议收藏)》《我用多线程进一步优化了亿级流量电商业务下的海量数据校对系统,性能再次提升了200%!!(全程干货,建议收藏)》《我用多线程优化了亿级流量电商业务下的海量数据校对系统,性能直接提升了200%!!(全程干货,建议收藏)》《我用10张图总结出了这份并发编程最佳学习路线!!(建议收藏)》《高并发场景下一种比读写锁更快的锁,看完我彻底折服了!!(建议收藏)》《全网最全性能优化总结!!(冰河吐血整理,建议收藏)》《三天撸完了MyBatis,各位随便问!!(冰河吐血整理,建议收藏)》《奉劝那些刚参加工作的学弟学妹们:要想进大厂,这些并发编程知识是你必须要掌握的!完整学习路线!!(建议收藏)》《奉劝那些刚参加工作的学弟学妹们:要想进大厂,这些核心技能是你必须要掌握的!完整学习路线!!(建议收藏)》《奉劝那些刚参加工作的学弟学妹们:这些计算机与操作系统基础知识越早知道越好!万字长文太顶了!!(建议收藏)》《我用三天时间开发了一款老少皆宜的国民级游戏,支持播放音乐,现开放完整源代码和注释(建议收藏)!!》《我是全网最硬核的高并发编程作者,CSDN最值得关注的博主,大家同意吗?(建议收藏)》《毕业五年,从月薪3000到年薪百万,我掌握了哪些核心技能?(建议收藏)》《我入侵了隔壁妹子的Wifi,发现。。。(全程实战干货,建议收藏)》《千万不要轻易尝试“熊猫烧香”,这不,我后悔了!》《清明节偷偷训练“熊猫烧香”,结果我的电脑为熊猫“献身了”!》《7.3万字肝爆Java8新特性,我不信你能看完!(建议收藏)》《在业务高峰期拔掉服务器电源是一种怎样的体验?》《全网最全Linux命令总结!!(史上最全,建议收藏)》《用Python写了个工具,完美破解了MySQL!!(建议收藏)》《SimpleDateFormat类到底为啥不是线程安全的?(附六种解决方案,建议收藏)》《MySQL 8中新增的这三大索引,直接让MySQL起飞了,你竟然还不知道!!(建议收藏)》《撸完Spring源码,我开源了这个分布式缓存框架!!(建议收藏)》《亿级流量高并发秒杀系统商品“超卖”了,只因使用的JDK同步容器中存在这两个巨大的坑!!(踩坑实录,建议收藏)》《奉劝那些刚参加工作的学弟学妹们:要想学好并发编程,这些并发容器的坑是你必须要注意的!!(建议收藏)》《公司的报表工具太难用,我三天撸了个Excel工具,运营小姐姐直呼太好用了,现已开源!!(建议收藏)》《奉劝那些刚参加工作的学弟学妹们:要想进大厂,这些并发编程核心技能是你必须要掌握的!!(建议收藏)》《阿里面试官:高并发大流量秒杀系统如何正确的解决库存超卖问题?(建议收藏)》《Redis五大数据类型与使用场景汇总!!(含完整实战案例,建议收藏)》
好了,今天就到这儿吧,小伙伴们点赞、收藏、评论,一键三连走起呀,我是冰河,我们下期见~~



