實(shí)現(xiàn)java簡單的線程池
請看下面這張圖
首先我們得對線程池進(jìn)行一個功能拆分
Thread Pool 就是我們的線程池,t1,t2,t3代表三個線程 Blocking Queue代表阻塞隊(duì)列 main代表main方法的線程 task1,task2,task3代表要執(zhí)行的每個任務(wù)現(xiàn)在我們梳理一下執(zhí)行的流程,注意這里是簡略版的,文章后面我會給出詳細(xì)版的
所以此時,我們發(fā)現(xiàn)了需要創(chuàng)建幾個類,或者說幾個角色,分別是
線程池 工作線程 阻塞隊(duì)列 拒絕策略(干嘛的?就是當(dāng)線程數(shù)已經(jīng)滿了,并且阻塞隊(duì)列也滿了,還有任務(wù)想進(jìn)入阻塞隊(duì)列的時候,就可以拒絕這個任務(wù))實(shí)現(xiàn)方式1.拒絕策略/** * 拒絕策略 */@FunctionalInterfaceinterface RejectPolicy<T>{//queue就是我們自己實(shí)現(xiàn)的阻塞隊(duì)列,task是任務(wù) void reject(BlockingQueue<T> queue,T task);}2.阻塞隊(duì)列
我們需要實(shí)現(xiàn)四個方法,獲取和添加,超時獲取和超時添加,至于方法實(shí)現(xiàn)的細(xì)節(jié),我都備注了大量的注釋進(jìn)行解釋。
/** * 阻塞隊(duì)列 */class BlockingQueue<T>{ //阻塞隊(duì)列 private Deque<T> queue = new ArrayDeque<>(); //鎖 private ReentrantLock lock = new ReentrantLock(); //生產(chǎn)者條件變量 private Condition fullWaitSet = lock.newCondition(); //消費(fèi)者條件變量 private Condition emptyWaitSet = lock.newCondition(); //容量 private int capacity; public BlockingQueue(int capacity){this.capacity = capacity; } //帶有超時阻塞獲取 public T poll(long timeout, TimeUnit timeUnit){lock.lock();try { //將timeout統(tǒng)一轉(zhuǎn)換為納秒 long nanos = timeUnit.toNanos(timeout); while(queue.isEmpty()){try { if(nanos <= 0){//小于0,說明上次沒有獲取到,代表已經(jīng)超時了return null; } //返回值是剩余的時間 nanos = emptyWaitSet.awaitNanos(nanos);} catch (InterruptedException e) { e.printStackTrace();} } T t = queue.removeFirst(); //通知生產(chǎn)者 fullWaitSet.signal(); return t;}finally { lock.unlock();} } //阻塞獲取 public T take(){lock.lock();try{ while(queue.isEmpty()){ //如果任務(wù)隊(duì)列為空,代表線程池沒有可以執(zhí)行的內(nèi)容try { /* 也就說此時進(jìn)來的線程是執(zhí)行不了任務(wù)的,所以此時emptyWaitSet消費(fèi)者要進(jìn)行阻塞狀態(tài) 等待下一次喚醒,然后繼續(xù)判斷隊(duì)列是否為空 */ emptyWaitSet.await();} catch (InterruptedException e) { e.printStackTrace();} } /* 代碼執(zhí)行到這里。說明任務(wù)隊(duì)列不為空,線程池就從任務(wù)隊(duì)列拿出一個任務(wù)出來執(zhí)行 也就是說把阻塞隊(duì)列的一個任務(wù)出隊(duì) */ T t = queue.removeFirst(); /* 然后喚醒之前存放在生成者Condition休息室,因?yàn)橛捎谥白枞?duì)列已滿,fullWaitSet才會進(jìn)入阻塞狀態(tài) 所以當(dāng)阻塞隊(duì)列刪除了任務(wù),就要喚醒之前進(jìn)入阻塞狀態(tài)的fullWaitSet */ fullWaitSet.signal(); //返回任務(wù) return t;}finally { lock.unlock();} } //阻塞添加 public void put(T task){lock.lock();try { while(queue.size() == capacity){ //任務(wù)隊(duì)列滿了try { System.out.println('等待加入任務(wù)隊(duì)列'+task); /* 也就說此時進(jìn)來的任務(wù)是進(jìn)不了阻塞隊(duì)列的,已經(jīng)滿了,所以此時生產(chǎn)者Condition要進(jìn)入阻塞狀態(tài) 等待下一次喚醒,然后繼續(xù)判斷隊(duì)列是否為空 */ fullWaitSet.await();} catch (InterruptedException e) { e.printStackTrace();} } //任務(wù)隊(duì)列還未滿 System.out.println('加入任務(wù)隊(duì)列'+task); //把任務(wù)加入阻塞隊(duì)列 queue.addLast(task); /* 然后喚醒之前存放在消費(fèi)者Condition休息室,因?yàn)橛捎谥白枞?duì)列為空,emptyWaitSet才會進(jìn)入阻塞狀態(tài) 所以當(dāng)阻塞隊(duì)列加入了任務(wù),就要喚醒之前進(jìn)入阻塞狀態(tài)的emptyWaitSet */ emptyWaitSet.signal();}finally { lock.unlock();} } //帶超時阻塞時間添加 public boolean offer(T task,long timeout,TimeUnit timeUnit){lock.lock();try { long nanos = timeUnit.toNanos(timeout); while(queue.size() == capacity){try { if(nanos < 0){return false; } System.out.println('等待加入任務(wù)隊(duì)列'+task); //不會一直阻塞,超時就會繼續(xù)向下執(zhí)行 nanos = fullWaitSet.awaitNanos(nanos);} catch (InterruptedException e) { e.printStackTrace();} } System.out.println('加入任務(wù)隊(duì)列'+task); queue.addLast(task); emptyWaitSet.signal(); return true;}finally { lock.unlock();} } //獲取任務(wù)數(shù)量 public int size(){lock.lock();try{ return queue.size();}finally { lock.unlock();} } //嘗試添加任務(wù),如果阻塞隊(duì)列已經(jīng)滿了,就使用拒絕策略 public void tryPut(RejectPolicy<T> rejectPolicy, T task){lock.lock();try { //判斷隊(duì)列是否已滿 if(queue.size() == capacity){rejectPolicy.reject(this,task); }else{ //有空閑System.out.println('加入任務(wù)隊(duì)列'+task);queue.addLast(task);emptyWaitSet.signal(); }}finally { lock.unlock();} }}3.線程池和工作線程
我把工作線程當(dāng)成線程池的內(nèi)部類去實(shí)現(xiàn)。方便調(diào)用變量。
/** * 線程池 */class ThreadPool{ //阻塞隊(duì)列 private BlockingQueue<Runnable> taskQueue; //線程集合 private HashSet<Worker> workers = new HashSet<>(); //核心線程數(shù) private int coreSize; //獲取任務(wù)的超時時間 private long timeout; private TimeUnit timeUnit; private RejectPolicy<Runnable> rejectPolicy; public ThreadPool(int coreSize, long timeout, TimeUnit timeUnit, int queueCapacity,RejectPolicy<Runnable> rejectPolicy) {this.coreSize = coreSize;this.timeout = timeout;this.timeUnit = timeUnit;this.taskQueue = new BlockingQueue<>(queueCapacity);this.rejectPolicy = rejectPolicy; } //執(zhí)行任務(wù) public void execute(Runnable task){synchronized (workers){ if(workers.size() <= coreSize){ //當(dāng)前的線程數(shù)小于核心線程數(shù)Worker worker = new Worker(task);workers.add(worker);//讓線程開始工作,執(zhí)行它的run方法worker.start(); }else{// 1) 死等// 2) 帶超時等待// 3) 讓調(diào)用者放棄任務(wù)執(zhí)行// 4) 讓調(diào)用者拋出異常// 5) 讓調(diào)用者自己執(zhí)行任務(wù)taskQueue.tryPut(rejectPolicy,task); }} } /** * 工作線程,也就是線程池里面的線程 */ class Worker extends Thread{private Runnable task;public Worker(Runnable task){ this.task = task;}@Overridepublic void run() { //執(zhí)行任務(wù) // 1) 當(dāng) task 不為空,執(zhí)行任務(wù) // 2) 當(dāng) task 執(zhí)行完畢,再接著從任務(wù)隊(duì)列獲取任務(wù)并執(zhí)行 while (task != null || (task = taskQueue.poll(timeout, timeUnit)) != null) {try { System.out.println('正在執(zhí)行的任務(wù)' + task); task.run();} catch (Exception e) { e.printStackTrace();} finally { //代表這個任務(wù)已經(jīng)執(zhí)行完了 task = null;} } synchronized (workers) {System.out.println('worker 被移除' + this);workers.remove(this); }} }}策略模式
細(xì)心的小伙伴已經(jīng)發(fā)現(xiàn),我在拒絕策略這里使用了23種設(shè)計模式的策略模式,因?yàn)槲覜]有將拒絕的方式寫死,而是交給了調(diào)用者去實(shí)現(xiàn)。
對比JDK的線程池下面是JDK自帶的線程池
經(jīng)典的七大核心參數(shù)
corePoolSize:核心線程數(shù) queueCapacity:任務(wù)隊(duì)列容量(阻塞隊(duì)列) maxPoolSize:最大線程數(shù) keepAliveTime:線程空閑時間 TimeUnit unit:超時時間單位 ThreadFactory threadFactory:線程工程 rejectedExecutionHandler:任務(wù)拒絕處理器實(shí)際上我們自己實(shí)現(xiàn)的也大同小異,只不過JDK官方的更為復(fù)雜。
JDK線程執(zhí)行的流程圖
線程我們知道在操作系統(tǒng)層面有5種狀態(tài)
線程在Java API層面有6種狀態(tài)
線程池有5種狀態(tài)
RUNNING:能接受新任務(wù),并處理阻塞隊(duì)列中的任務(wù) SHUTDOWN:不接受新任務(wù),但是可以處理阻塞隊(duì)列中的任務(wù) STOP:不接受新任務(wù),并且不處理阻塞隊(duì)列中的任務(wù),并且還打斷正在運(yùn)行任務(wù)的線程,就是直接不干了! TIDYING:所有任務(wù)都終止,并且工作線程也為0,處于關(guān)閉之前的狀態(tài) TERMINATED:已關(guān)閉。本篇文章就到這里了,希望能給你帶來幫助,也希望您能夠多多關(guān)注好吧啦網(wǎng)的更多內(nèi)容!
相關(guān)文章:
