【Java】实现一个简单的线程池

  📝个人主页：哈__

期待您的关注 

目录

​编辑

一、线程池的模式

二、线程池的一些参数 

三、代码实现

1.BlockingQueue

2.ThreadPool

四、拒绝策略

一、线程池的模式

线程池顾名思义就是管理线程的一个池子，我们把创建线程的过程交给线程池来处理，而这个线程池当中的线程都会从阻塞队列当中取获取任务执行。

我们不在直接把任务的创建过程写到我们初始化的线程对象中，而是通过调用线程池的execute()方法，同时把我们的具体任务交作为参数传给线程池，之后线程池就会把任务添加到阻塞队列当中，而线程池当中的线程会从阻塞队列当中获取任务并执行。

二、线程池的一些参数 

  1. corePoolSize：线程池核心线程大小，即最小线程数（初始化线程数）。线程池会维护当前数量的线程在线程池中，即使这些线程一直处于闲置状态，也不会被销毁，除非设置了allowCoreThreadTimeOut。
  2. maximumPoolSize：线程池最大线程数量。当任务提交到线程池后，如果当前线程数小于核心线程数，则会创建新线程来处理任务；如果当前线程数大于或等于核心线程数，但小于最大线程数，并且任务队列已满，则会创建新线程来处理任务。
  3. keepAliveTime：空闲线程的存活时间。当线程池中的线程数量大于核心线程数且线程处于空闲状态时，在指定时间后，这个空闲线程将会被销毁，从而逐渐恢复到稳定的核心线程数数量。
  4. unit：keepAliveTime的存活时间的计量单位，通常使用TimeUnit枚举类中的方法，如TimeUnit.SECONDS表示秒级。
  5. workQueue：任务队列。用于存放等待执行的任务，常见的实现类有LinkedBlockingQueue、ArrayBlockingQueue等。
  6. threadFactory：线程工厂。用于创建新的线程，可以自定义线程的名称、优先级等。
  7. handler：拒绝策略。当任务无法执行（如线程池已满）时，可以选择的策略有：AbortPolicy（抛出异常）、CallerRunsPolicy（调用者运行）、DiscardOldestPolicy（丢弃最老的任务）、DiscardPolicy（无声丢弃）。

三、代码实现

因为我们只是简单的实现，所以有一些情况和实际不太相似。

1.BlockingQueue

先来看看我们阻塞队列当中的一些参数，为了在多线程环境下防止并发问题，我使用了ReentrantLock，使用它的目的是为了创建多个不同的阻塞条件。

在我们调用一个对象的await()方法后，我们的当前线程就会加入到一个特定的队列当中去等待，直到有调用了这个对象的notify()方法后才会从这个队列中抽取一个线程唤醒。

举个例子，我们去医院的时候，一个医生同一时间只能看一个病人，剩下的人都只能等待，如果只有一个大厅的话，看不同病的病人都只能等待在一个候诊室中。使用ReentrentLock的意思就是为了创建多个不同的候诊室，将不同医生的病人分开在不同的候诊室当中。

    //1.阻塞队列private Deque<T> deque = new ArrayDeque<>();//2.实现阻塞的锁private ReentrantLock lock = new ReentrantLock();//3. 生产者等待条件private Condition fullWaitSet = lock.newCondition();//4.消费者等待条件private Condition emptyWaitSet = lock.newCondition();//5.阻塞队列的大小private  int CAPACITY;

在自定义的阻塞队列中，我使用了一个双向队列来存储任务，并且设置了一个队列大小的属性，在我们创建这个队列的时候我们可以进行初始化。

先来看看阻塞队列任务的添加过程。这个逻辑并不难，我们在代码的上方上锁，在finally中解锁。如果这时我们的队列是满的，就无法在继续添加任务了，这个时候我们就把当前线程挂起（注意我们的挂起条件）。如果队列不是满的话那我们就加入到队尾，同时把另一类挂起的线程唤醒（这类线程在队列为空的时候挂起，等待任务的添加）。

 // 生产者放入数据public void put(T t) {lock.lock();try {while (deque.size() == CAPACITY) {fullWaitSet.await();}deque.addLast(t);emptyWaitSet.signal();} catch (InterruptedException e) {e.printStackTrace();} finally {lock.unlock();}}

在看看我们取任务的过程。同样加锁，当我们的队列为空的时候，线程挂起，等待任务的添加之后线程唤醒，如果队列不为空的话，我们从队列头部取出一个任务，并且唤起一类线程（这类线程在任务已经满了的时候无法在添加任务了，进行挂起，等待队列不为满）。

  // 消费者从线程池当中获取任务public T take(){T t = null;lock.lock();try {while(deque.size() == 0){emptyWaitSet.await();}t = deque.removeFirst();fullWaitSet.signal();}catch (Exception e){e.printStackTrace();}finally {lock.unlock();}return t;}

我们上边的代码展示的队列的存取的过程都是死等状态，什么是死等状态？就是任务添加不进去或者取不出来的时候，线程会被一直挂起。真实并不是如此，这里只是简单的展示。

阻塞队列需要的就是这两个存取的过程。

2.ThreadPool

先看看线程池当中的属性。把刚才创建的任务队列加进去，因为线程池要时常和任务队列沟通。然后创建了一个HashSet结构用于存储我们的线程。下边的都是我们线程池需要的一些参数了，拒绝策略在这里没有写。

    // 任务队列private BlockedQueue<Runnable> taskQueue;// 线程集合private HashSet<Worker> workers = new HashSet<>();//核心线程数private int coreSize;// 超时时间private int timeout;// 超时单位private TimeUnit timeUnit;

来看看我们的线程池是如何工作的吧，可以看到我们线程池保存的是Worker对象，我们来看看这个Worker对象是干啥的。这个Worker对象实现了Runnable接口，我们可以把这个类当作线程类，这个类中有一个task属性，因为我们线程池当中的线程是要获取任务执行的，这个任务就用这个task属性代表。

这个Worker类一直在干一件事情，就是不断地从我们的任务队列当中获取任务（Worker类是ThreadPool的内部类），如果获取的任务不为空的话就执行任务，一旦没有任务可以执行那么就把当前的线程从线程池当中移除。

class Worker implements Runnable{private Runnable task;public Worker(Runnable task){this.task = task;}@Overridepublic void run() {while(task!=null || (task = taskQueue.take())!=null){System.out.println("取出的任务是"+task);try {task.run();}catch (Exception e){e.printStackTrace();}finally {task = null;}synchronized (workers){workers.remove(this);}}}}

那什么时候用到这个Worker类呢？当我们调用ThreadPool中的execute()方法时，线程池中的线程会就调用这个run()方法。

来看我们的execute()方法。当我们的线程数小于我们的核心线程数的时候，我们可以直接创建一个新的线程，并且把我们的任务直接交给这个核心线程。反之我们不能创建，而是把任务添加到我们的任务队列当中，等待核心线程去执行这个任务。

 // 任务执行public void execute(Runnable task){synchronized (workers){if(workers.size() < coreSize){// 创建核心线程Worker worker = new Worker(task);workers.add(worker);Thread thread = new Thread(worker);thread.start();}else {taskQueue.put(task);}}}

写完了上边的代码我们测试一下。

 public static void main(String[] args) {ThreadPool threadPool = new ThreadPool(2,10,TimeUnit.MILLISECONDS,10);for(int i = 0;i<12;i++){int j = i;threadPool.execute(()->{System.out.println("当前线程"+Thread.currentThread().getName()+"task "+j+" is running");try {Thread.currentThread().sleep(1000);} catch (InterruptedException e) {e.printStackTrace();}});}}

方法运行了之后，即使任务全部执行，线程也不会结束。这是因为我们的worker类中的run方法调用了任务队列的take()方法，而take方法是会一直挂起的。

我们现在换一种带超时获取，在规定时间内获取不到任务就自动结束任务。这时候就用到我们传入的时间参数了，我们不再调用await()方法了，而是调用awaitNanos()方法，方法可以接收一个时间参数，这个方法可以消耗我们的nanos时间，在这个时间内如果获取不到的话线程就不在挂起了，这时还会进入到我们的while循环当中，判断我们的nanos是不是被消耗完了，如果被消耗完了就说明在规定时间内获取不到任务，直接return结束线程。

 // 带超时获取public T poll(int timeout,TimeUnit timeUnit){T t = null;lock.lock();try {long nanos = timeUnit.toNanos(timeout);while(deque.size() == 0){if(nanos <= 0){return null;}nanos = emptyWaitSet.awaitNanos(nanos);}t = deque.removeFirst();fullWaitSet.signal();}catch (Exception e){e.printStackTrace();}finally {lock.unlock();}return t;}

 修改Worker类。

 class Worker implements Runnable{private Runnable task;public Worker(Runnable task){this.task = task;}@Overridepublic void run() {while(task!=null || (task = taskQueue.poll(timeout,timeUnit))!=null){System.out.println("取出的任务是"+task);try {task.run();}catch (Exception e){e.printStackTrace();}finally {task = null;}synchronized (workers){workers.remove(this);}}}}

现在就可以正常结束了。

四、拒绝策略

全部代码如下。要使用拒绝策略，我们定义一个函数式接口，同时写一个参数传给线程池，参数的具体内容就是拒绝策略的拒绝方法，是我们自己定义的。

同时我们的execute()方法不在使用put来添加任务了，而是使用tryPut，如果大家对这一块感兴趣的话，可以在bilibili上观看黑马程序员的课程学习一下。

/*** 自定义线程池*/
public class TestPool {public static void main(String[] args) {ThreadPool threadPool = new ThreadPool(2,10,TimeUnit.SECONDS,10,((queue, task) -> {queue.put(task);}));for(int i = 0;i<12;i++){int j = i;threadPool.execute(()->{System.out.println("当前线程"+Thread.currentThread().getName()+"task "+j+" is running");try {Thread.currentThread().sleep(1000);} catch (InterruptedException e) {e.printStackTrace();}});}}
}/*** 拒绝策略*/
@FunctionalInterface
interface RejectPolicy<T>{void reject(BlockedQueue<T> queue,T task);
}
/*** 阻塞队列*/
class BlockedQueue <T>{//1.阻塞队列private Deque<T> deque = new ArrayDeque<>();//2.实现阻塞的锁private ReentrantLock lock = new ReentrantLock();//3. 生产者等待条件private Condition fullWaitSet = lock.newCondition();//4.消费者等待条件private Condition emptyWaitSet = lock.newCondition();//5.阻塞队列的大小private  int CAPACITY;public BlockedQueue(int queueCapacity) {this.CAPACITY = queueCapacity;}// 生产者放入数据public void put(T t) {lock.lock();try {while (deque.size() == CAPACITY) {fullWaitSet.await();}deque.addLast(t);emptyWaitSet.signal();} catch (InterruptedException e) {e.printStackTrace();} finally {lock.unlock();}}// 带超时添加public boolean offer(T t,int timeout,TimeUnit timeUnit) {lock.lock();long nanos = timeUnit.toNanos(timeout);try {while (deque.size() == CAPACITY) {if(nanos <= 0){return  false;}nanos = fullWaitSet.awaitNanos(nanos);}deque.addLast(t);emptyWaitSet.signal();return  true;} catch (InterruptedException e) {e.printStackTrace();} finally {lock.unlock();}return true;}// 带超时获取public T poll(int timeout,TimeUnit timeUnit){T t = null;lock.lock();try {long nanos = timeUnit.toNanos(timeout);while(deque.size() == 0){if(nanos <= 0){return null;}nanos = emptyWaitSet.awaitNanos(nanos);}t = deque.removeFirst();fullWaitSet.signal();}catch (Exception e){e.printStackTrace();}finally {lock.unlock();}return t;}// 消费者从线程池当中获取任务public T take(){T t = null;lock.lock();try {while(deque.size() == 0){emptyWaitSet.await();}t = deque.removeFirst();fullWaitSet.signal();}catch (Exception e){e.printStackTrace();}finally {lock.unlock();}return t;}public void tryPut(RejectPolicy<T> rejectPolicy, T task) {lock.lock();try {if(deque.size()==CAPACITY){rejectPolicy.reject(this,task);}else{deque.addLast(task);emptyWaitSet.signal();}} finally {lock.unlock();}}
}
/*** 线程池*/
class ThreadPool{// 任务队列private BlockedQueue<Runnable> taskQueue;// 线程集合private HashSet<Worker> workers = new HashSet<>();//核心线程数private int coreSize;// 超时时间private int timeout;// 超时单位private TimeUnit timeUnit;//拒绝策略private RejectPolicy<Runnable> rejectPolicy;// 任务执行public void execute(Runnable task){synchronized (workers){if(workers.size() < coreSize){// 创建核心线程Worker worker = new Worker(task);workers.add(worker);Thread thread = new Thread(worker);thread.start();}else {// 任务队列//taskQueue.offer(task,timeout,timeUnit);taskQueue.tryPut(rejectPolicy,task);//taskQueue.put(task);}}}public ThreadPool(int coreSize, int timeout, TimeUnit timeUnit,int queueCapacity,RejectPolicy<Runnable> rejectPolicy){this.coreSize = coreSize;this.timeout = timeout;this.timeUnit = timeUnit;this.taskQueue = new BlockedQueue<>(queueCapacity);this.rejectPolicy = rejectPolicy;}class Worker implements Runnable{private Runnable task;public Worker(Runnable task){this.task = task;}@Overridepublic void run() {while(task!=null || (task = taskQueue.poll(timeout,timeUnit))!=null){System.out.println("取出的任务是"+task);try {task.run();}catch (Exception e){e.printStackTrace();}finally {task = null;}synchronized (workers){workers.remove(this);}}}}
}

这个代码我自己觉得是有些问题，因为如果我的任务队列大小有10的时候，我给出了13个任务，两个交给核心线程不占任务队列大小，另外10个任务正好占满，剩下一个放不进去，这时就会卡住不输出。---------未解决

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.hqwc.cn/news/661610.html

如若内容造成侵权/违法违规/事实不符，请联系编程知识网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

Murphy：优维大模型运维数智人上线服役

Murphy：优维大模型运维数智人上线服役

01 墨菲定律&#xff1a; 幽默又严肃的NASA军规 「墨菲定律」 &#xff08;Murphy’s law&#xff09; 任何可能出错的事情最终都会出错 墨菲定律强调了在复杂系统和人类活动中&#xff0c;事情往往会出现意想不到的差错。虽然这只是一种对人类经验和偶然性的幽默概括与启…
阅读更多...
iA Writer for Mac：简洁强大的写作软件

iA Writer for Mac：简洁强大的写作软件

在追求高效写作的今天&#xff0c;iA Writer for Mac凭借其简洁而强大的功能&#xff0c;成为了许多作家、记者和学生的首选工具。这款专为Mac用户打造的写作软件&#xff0c;以其独特的设计理念和实用功能&#xff0c;助你轻松打造高质量的文章。 iA Writer for Mac v7.1.2中文…
阅读更多...
前端性能优化知识梳理

前端性能优化知识梳理

1.重要性 当我们面试的时候&#xff0c;前端性能优化方面算是必考的知识点&#xff0c;但是工作中我们又很少会重点的对项目进行前端优化&#xff0c;它真的不重要吗&#xff1f; 如果我们可以将后端响应时间缩短一半&#xff0c;整体响应时间只能减少5%~10%。而如果关注前端…
阅读更多...
ZYNQ启动流程分析之BOOT.BIN头

ZYNQ启动流程分析之BOOT.BIN头

BOOT.BIN头简介 BOOT.BIN头其实就是BOOT.BIN文件前面的一段头部数据&#xff0c;并且这个头部数据是按照一定的格式组织在一起的&#xff0c;并且该头部数据能够被BootROM代码解析。   在boot.bin文件中从地址0-0x8FF可以分为17个部分&#xff0c;每个部分都有一定的含义。…
阅读更多...
深入解析yolov5，为什么算法都是基于yolov5做改进的？（一）

深入解析yolov5，为什么算法都是基于yolov5做改进的？（一）

YOLOv5简介 YOLOv5是一种单阶段目标检测算法&#xff0c;它在YOLOv4的基础上引入了多项改进&#xff0c;显著提升了检测的速度和精度。YOLOv5的设计哲学是简洁高效&#xff0c;它有四个版本&#xff1a;YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x&#xff0c;分别对应不同的模型大小…
阅读更多...
Dockerfile实战（SSH、Systemctl、Nginx、Tomcat）

Dockerfile实战（SSH、Systemctl、Nginx、Tomcat）

目录 一、构建SSH镜像 1.1 dockerfile文件内容 1.2 生成镜像 1.3 启动容器并修改root密码 二、构建Systemctl镜像 2.1 编辑dockerfile文件 ​编辑2.2 生成镜像 2.3 启动容器&#xff0c;并挂载宿主机目录挂载到容器中&#xff0c;然后进行初始化 2.4 进入容器验证 三、…
阅读更多...
基本STL使用

基本STL使用

一 、关于vector 在STL中有一个称为vector的数据结构&#xff0c;可以用来代替数组。 定义Book特性 private:vector<string> shelf_books;Notic : 类中不能使用类似的定义&#xff1a;vector<sttring> shelf_boos( 10 ); 定义Book方法 public:void setName(str…
阅读更多...
PDF 正确指定页码后，挂载的书签页码对不上

PDF 正确指定页码后，挂载的书签页码对不上

这个问题与我的另一篇中方法一样 如何让一个大几千页的打开巨慢的 PDF 秒开-CSDN博客 https://blog.csdn.net/u013669912/article/details/138166922 另作一篇的原因 一篇文章附带一个与该文章主题不相关的问题时&#xff0c;不利于被遇到该问题的人快速搜索发现以解决其遇到…
阅读更多...
OpenCV如何实现背投(58)

OpenCV如何实现背投(58)

返回:OpenCV系列文章目录&#xff08;持续更新中......&#xff09; 上一篇&#xff1a;OpenCV直方图比较(57) 下一篇&#xff1a;OpenCV如何模板匹配(59) 目标 在本教程中&#xff0c;您将学习&#xff1a; 什么是背投以及它为什么有用如何使用 OpenCV 函数 cv::calcBackP…
阅读更多...
Windows之隐藏特殊文件夹(自定义快捷桌面程序)

Windows之隐藏特殊文件夹(自定义快捷桌面程序)

作者主页&#xff1a;点击&#xff01; 创作时间&#xff1a;2024年5月1日12点55分 祝大家劳动节快乐~ Windows中的特殊文件夹是指一些预定义的文件夹&#xff0c;用于存储特定类型的数据或文件。这些文件夹通常由操作系统或应用程序使用&#xff0c;但用户也可以访问和管理它…
阅读更多...
数字旅游引领未来智慧之旅：科技应用深度重塑旅游生态，智慧服务全面升级打造极致高品质旅游体验

数字旅游引领未来智慧之旅：科技应用深度重塑旅游生态，智慧服务全面升级打造极致高品质旅游体验

随着信息技术的飞速发展&#xff0c;数字旅游作为旅游业与科技融合的新兴业态&#xff0c;正以其独特的魅力和优势&#xff0c;引领着旅游业迈向智慧之旅的新时代。数字旅游不仅通过科技应用重塑了旅游生态&#xff0c;更通过智慧服务为游客带来了高品质的旅游体验。本文将深入…
阅读更多...
bun 换源 国内阿里源 npmmirror 加速下载

bun 换源 国内阿里源 npmmirror 加速下载

Github https://github.com/oven-sh/bun 版本号 bun 1.1.5 windows 安装 bun 如果本机有 nodejs 环境, 可以 npm install -g bun 安装 ( 官方把 exe 已经传到了 npm 仓库, 走的国内 npm 镜像, 下载速度会很快) 没有 nodejs, 可以用 powershell 脚本安装 具体操作 全局 …
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023