本篇内容介绍了“Java怎么实现手写线程池并测试”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!
线程池的核心流程:
在线程池核心原理的源码中,涉及到了一系列的流程,包括线程池队列数量是否已满,运用什么样的拒绝策略等。在我们手写线程池的代码中,不需要考虑那么多因素,只需要模拟简单的情景和过程,因此整体来讲还是比较简单的。
手写线程池,必不可少的组件有任务队列,任务的消费者线程池,线程池创建等。我们也可定义构造方法,创建指定大小的线程池线程个数。当然在使用完线程池中的线程后,我们需要考虑将其销毁或关闭。
相关代码和解释如下:
package XIAOWEI;
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 WorkQueueSIZE = 5;
private BlockingQueue<Runnable> workQueue;
private List<WorkThread> workThreads=new ArrayList<WorkThread>();
/**
* @author xiaowei
* @param poolSize
* @param workQueue
* 构造方法,传线程池的大小和阻塞队列
*/
public ThreadPool(int poolSize,BlockingQueue<Runnable> workQueue) {
this.workQueue = workQueue;
IntStream.range(0,poolSize).forEach((i)->{
WorkThread workThread=new WorkThread();
workThread.start();
workThreads.add(workThread);
});
}
/**
* @param poolSize
* 在ThreadPool的构造方法中传入线程池的大小
*/
public ThreadPool(int poolSize){
this(poolSize,new LinkedBlockingQueue<>(WorkQueueSIZE));
}
/**
* @param task
* 通过线程池执行任务
*/
public void extcute(Runnable task){
try {
workQueue.put(task);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public void shutdown(){
if(workThreads!=null && workThreads.size()>0){
workThreads.stream().forEach((workThread) -> {
workThread.interrupt();
});
}
}
/**
* 内部类 源源不断的消耗workQueue中的任务
*/
class WorkThread extends Thread {
@Override
public void run() {
Thread currentThread = Thread.currentThread();
// 死循环 不断一直消费队列中的任务 直到任务被消费完全
while (true) {
try {
if (currentThread.isInterrupted()) {
break;
}
Runnable workTasK = workQueue.take();
workTasK.run();
} catch (InterruptedException e) {
currentThread.interrupt();
}
}
}
}
}
相关测试代码如下:
package XIAOWEI;
import java.util.stream.IntStream;
public class ThreadPoolTest {
public static void main(String[] args) {
ThreadPool threadPool = new ThreadPool(5);
IntStream.range(0,10).forEach((i)->{
threadPool.extcute(()->{
System.out.println(Thread.currentThread().getName()+"2023一起加油");
});
});
threadPool.shutdown();
}
}
一共在线程池中加入了五个线程和十个任务,因此每个线程会执行两个任务。