开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情
前言
每一个线程的启动和结束都是比较消耗时间和占用资源的。
如果在系统中使用到了很多的线程,大量的启动和结束动作会导致系统的性能变卡,响应变慢。
为了解决这个问题,引入线程池这种设计思想。
线程池的模式很像生产者消费者模式,消费的对象是一个一个能够运行的任务。
一、线程池设计思路
线程池的思路和生产者消费者模型是很接近的:
1、准备一个任务容器
2、一次性启动 10 个消费者线程
3、刚开始任务容器是空的,所以线程都 wait 在上面
4、直到一个外部线程往这个任务容器中扔了一个任务,就会有一个消费者线程被唤醒 notify
5、这个消费者线程取出任务,并且执行这个任务,执行完毕后,继续等待下一次任务的到来。
6、如果短时间内,有较多的任务加入,那么就会有多个线程被唤醒,去执行这些任务。
在整个过程中,都不需要创建新的线程,而是循环使用这些已经存在的线程
如下图:
二、开发一个自定义线程池
如下代码实现,虽然不够完善和健壮,但是已经足以说明线程池的工作原理:
//1、自定义线程池
public class ThreadPool {
/**
* 线程池大小
*/
private int threadPoolSize;
/**
* 任务容器
*/
private LinkedList<Runnable> tasks = new LinkedList<>();
/**
* 试图消费任务的线程
*/
public ThreadPool(){
threadPoolSize = 10;
//启动 10 个任务消费者线程
synchronized (tasks){
for (int i = 0; i < threadPoolSize; i++) {
new TaskConsumeThread("任务消费者线程 " + i).start();
}
}
}
public void add(Runnable runnable){
synchronized (tasks){
tasks.add(runnable);
//唤醒等待的任务消费者线程
tasks.notifyAll();
}
}
class TaskConsumeThread extends Thread{
private Runnable task;
public TaskConsumeThread(@NonNull String name) {
super(name);
}
@Override
public void run() {
System.out.println("启动:" + getName());
while (true){
synchronized (tasks){
while (tasks.isEmpty()){
try {
tasks.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
task = tasks.removeLast();
//允许添加任务的线程可以继续添加任务
tasks.notifyAll();
}
System.out.println(getName() + " 获取到任务,并执行");
task.run();
}
}
}
}
//2、测试
public class TestThread {
public static void main(String[] args) {
ThreadPool pool = new ThreadPool();
for (int i = 0; i < 5; i++) {
int finalI = i;
Runnable task = new Runnable() {
@Override
public void run() {
System.out.println("执行任务:" + finalI);
}
};
pool.add(task);
}
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
//打印结果
启动:任务消费者线程 0
启动:任务消费者线程 2
启动:任务消费者线程 1
启动:任务消费者线程 3
启动:任务消费者线程 5
启动:任务消费者线程 6
启动:任务消费者线程 7
启动:任务消费者线程 8
启动:任务消费者线程 4
启动:任务消费者线程 9
任务消费者线程 7 获取到任务,并执行
任务消费者线程 2 获取到任务,并执行
任务消费者线程 9 获取到任务,并执行
任务消费者线程 0 获取到任务,并执行
执行任务:3
执行任务:0
执行任务:2
任务消费者线程 1 获取到任务,并执行
执行任务:1
执行任务:4
三、测试线程池
创造一个情景,每个任务执行的时间都是 1 秒,刚开始是间隔 1 秒钟向线程池中添加任务,然后间隔时间越来越短,执行任务的线程还没来得及结束,新的任务又来了。此时就会观察到线程池里的其他线程被唤醒来执行这些任务
public class TestThread {
public static void main(String[] args) {
ThreadPool pool = new ThreadPool();
int sleep = 1000;
while (true) {
pool.add(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
});
try {
Thread.sleep(sleep);
sleep = sleep > 100 ? sleep - 100 : sleep;
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
}
//打印结果
启动:任务消费者线程 0
启动:任务消费者线程 3
启动:任务消费者线程 4
启动:任务消费者线程 2
启动:任务消费者线程 1
启动:任务消费者线程 5
启动:任务消费者线程 6
启动:任务消费者线程 7
启动:任务消费者线程 8
启动:任务消费者线程 9
任务消费者线程 8 获取到任务,并执行
任务消费者线程 4 获取到任务,并执行
任务消费者线程 7 获取到任务,并执行
任务消费者线程 8 获取到任务,并执行
任务消费者线程 4 获取到任务,并执行
任务消费者线程 7 获取到任务,并执行
任务消费者线程 8 获取到任务,并执行
任务消费者线程 4 获取到任务,并执行
任务消费者线程 6 获取到任务,并执行
任务消费者线程 7 获取到任务,并执行
任务消费者线程 5 获取到任务,并执行
任务消费者线程 8 获取到任务,并执行
任务消费者线程 9 获取到任务,并执行
任务消费者线程 0 获取到任务,并执行
任务消费者线程 1 获取到任务,并执行
任务消费者线程 4 获取到任务,并执行
任务消费者线程 2 获取到任务,并执行
任务消费者线程 3 获取到任务,并执行
任务消费者线程 6 获取到任务,并执行
任务消费者线程 7 获取到任务,并执行
任务消费者线程 5 获取到任务,并执行
任务消费者线程 8 获取到任务,并执行
任务消费者线程 9 获取到任务,并执行
任务消费者线程 0 获取到任务,并执行
...
四、使用 Java 自带线程池
Java 给我们提供了自带的线程池 ThreadPoolExecutor 在 java.util.concurrent 包下。
如下代码:
ThreadPoolExecutor pool = new ThreadPoolExecutor(10, 15, 60, TimeUnit.SECONDS, new LinkedBlockingQueue<>());
pool.execute(new Runnable() {
@Override
public void run() {
System.out.println("执行任务");
}
});
第一个参数 10 :表示线程池初始化了 10 线程在里面工作
第二个参数 15 :表示如果 10 个线程不够用了,就会增加到最多 15 个线程
第三个参数 60 :结合第四个参数 TimeUnit.SECONDS 表示 经过 60 秒,多出来的线程没有接到活,就会回收,最后保持池子里面就 10 个
第四个参数 TimeUnit.SECONDS 如上
第五个参数 new LinkedBlockingQueue<>() :表示存放任务的集合
execute 方法用于添加并执行新的任务
五、总结
本篇文章我们介绍了:
1、设计一个线程池的思路并进行了具体实现
2、自定义线程池测试
3、Java 自带线程池介绍
好了,本篇文章到这里就结束了,感谢你的阅读🤝