设为首页 加入收藏

TOP

10分钟带你徒手做个Java线程池(二)
2023-07-25 21:30:57 】 浏览:49
Tags:10分 钟带你 Java 程池
il.stream.IntStream;
/** * @author binghe * @version 1.0.0 * @description 自定义线程池 */ public class ThreadPool { //默认阻塞队列大小 private static final int DEFAULT_WORKQUEUE_SIZE = 5; //模拟实际的线程池使用阻塞队列来实现生产者-消费者模式 private BlockingQueue<Runnable> workQueue; //模拟实际的线程池使用List集合保存线程池内部的工作线程 private List<WorkThread> workThreads = new ArrayList<WorkThread>(); //在ThreadPool的构造方法中传入线程池的大小和阻塞队列 public ThreadPool(int poolSize, BlockingQueue<Runnable> 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;
/**
 * @author binghe
 * @version 1.0.0
 * @description 测试自定义线程池
 */
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线程池就开发完成了。

总结

线程池的核心原理其实并不复杂,只要我们耐心的分析,深入其源码理解线程池的核心本质,你就会发现线程池的设计原来是如此的优雅。希望通过这个手写线程池的小例子,能够让你更好的理解线程池的核心原理。

 

点击关注,第一时间了解华为云新鲜技术~

首页 上一页 1 2 下一页 尾页 2/2/2
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇Leetcode刷题第二周 下一篇Java的值传递和引用传递

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目