快捷導(dǎo)航

java自定義線程池的原理簡介

更新時間：2019年08月22日 10:09:51 作者：357029540

這篇文章主要介紹了java自定義線程池的原理詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

線程池的相關(guān)概念就不在這里說明了，百度一下有很多，這里簡單表述一下如何實(shí)現(xiàn)一個自定義的線程池就行線程管理，我們?nèi)绻獙?shí)現(xiàn)一個線程池對線程的管理，那么需要實(shí)現(xiàn)一下幾點(diǎn)的思路:

1.如何管理線程

2.如何定義工作線程以及工作線程如何持續(xù)的保持運(yùn)行狀態(tài)

3.如何定義線程池大小及隊列大小

4.如何提供接口給調(diào)用者使用

5.如何關(guān)閉線程池中的線程

接下來我們就一一的實(shí)現(xiàn)這幾個問題。

1.我們需要定義一個隊列來來管理線程，這里使用了LinkedBlockingQueue

// 1.定義一個存儲線程隊列 
private LinkedBlockingQueue<Runnable> queue;

2.因?yàn)槭且粋€簡單的測試，所以我們可以先定義一個內(nèi)部類來實(shí)現(xiàn)工作線程

// 2.定義工作線程進(jìn)行線程的執(zhí)行 
  class Worker extends Thread {  
    private SelfThreadPoolExecutor threadPoolExecutor; 
 
    public Worker(SelfThreadPoolExecutor poolExecutor) { 
      this.threadPoolExecutor = poolExecutor; 
    }  
    @Override 
    public void run() { 
      Runnable task; 
      while (threadPoolExecutor.receiveTask || threadPoolExecutor.queue.size() > 0) { 
        try { 
          // 有線程則取出來，否則等待 
          System.out.println("準(zhǔn)備消費(fèi)線程"); 
          task = threadPoolExecutor.queue.take(); 
          if (task != null) { 
            task.run(); 
            System.out.println("消費(fèi)線程"); 
          } 
        } catch (InterruptedException e) { 
          e.printStackTrace(); 
        } 
      } 
    } 
  }

SelfThreadPoolExecutor是外部定義的整體類名

3.使用有參的構(gòu)造方法進(jìn)行線程池大小的管理

// 3.存放工作線程的集合 
  private List<Worker> workerList;  
  // 4.線程池初始化 
  public SelfThreadPoolExecutor(int coreSize, int queueSize) { 
    if (coreSize <= 0 || queueSize <= 0) { 
      throw new IllegalArgumentException("參數(shù)不正確"); 
    } 
    this.queue = new LinkedBlockingQueue<>(queueSize); 
    // 線程安全的集合 
    this.workerList = Collections.synchronizedList(new ArrayList<>()); 
    for (int i = 0; i < coreSize; i++) { 
      Worker worker = new Worker(this); 
      worker.start(); 
      workerList.add(worker); 
    } 
  }

4.定義阻塞和非阻塞的方式提供對應(yīng)的接口

// 5.非阻塞的方法接口 
  public boolean offer(Runnable task) { 
    if (receiveTask) { 
      return queue.offer(task); 
    } else { 
      return false; 
    } 
  } 
  // 6.阻塞的方法接口 
  public void put(Runnable task) { 
    try { 
      if (receiveTask) { 
        queue.put(task); 
      } 
    } catch (InterruptedException e) { 
      e.printStackTrace(); 
    } 
  }

6.進(jìn)行線程池的關(guān)閉

// 7.線程池的關(guān)閉 
  private boolean receiveTask = true;  
  public void shutdown() { 
    // 7.1.隊列不再接收線程 
    receiveTask = false; 
    // 7.2.關(guān)閉處于wait或block的線程 
    for (Thread thread : workerList) { 
      if (Thread.State.BLOCKED.equals(thread.getState()) 
      || Thread.State.WAITING.equals(thread.getState()) 
      || Thread.State.TIMED_WAITING.equals(thread.getState())){ 
        thread.interrupt(); 
      } 
    } 
  }

我們測試的方法如下：

public static void main(String [] args){ 
    SelfThreadPoolExecutor selfThreadPoolExecutor = new SelfThreadPoolExecutor(5,10); 
    for(int i = 0;i < 20;i++){ 
      Runnable task = () ->{ 
        System.out.println("開啟線程"); 
      }; 
      selfThreadPoolExecutor.put(task); 
    } 
    selfThreadPoolExecutor.shutdown(); 
  }

運(yùn)行結(jié)果是：

準(zhǔn)備消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
開啟線程 
消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
開啟線程 
消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
開啟線程 
消費(fèi)線程 
準(zhǔn)備消費(fèi)線程 
。。。。。。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: