温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

java中自定义线程池的原理是什么

发布时间:2021-06-11 15:51:51 来源:亿速云 阅读:119 作者:Leah 栏目:编程语言

java中自定义线程池的原理是什么,很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。

1.如何管理线程

2.如何定义工作线程以及工作线程如何持续的保持运行状态

3.如何定义线程池大小及队列大小

4.如何提供接口给调用者使用

5.如何关闭线程池中的线程

接下来我们就一一的实现这几个问题。

1.我们需要定义一个队列来来管理线程,这里使用了LinkedBlockingQueue

// 1.定义一个存储线程队列 
private LinkedBlockingQueue<Runnable> queue;

2.因为是一个简单的测试,所以我们可以先定义一个内部类来实现工作线程

// 2.定义工作线程进行线程的执行 
  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("准备消费线程"); 
          task = threadPoolExecutor.queue.take(); 
          if (task != null) { 
            task.run(); 
            System.out.println("消费线程"); 
          } 
        } catch (InterruptedException e) { 
          e.printStackTrace(); 
        } 
      } 
    } 
  }

SelfThreadPoolExecutor是外部定义的整体类名

3.使用有参的构造方法进行线程池大小的管理

// 3.存放工作线程的集合 
  private List<Worker> workerList;  
  // 4.线程池初始化 
  public SelfThreadPoolExecutor(int coreSize, int queueSize) { 
    if (coreSize <= 0 || queueSize <= 0) { 
      throw new IllegalArgumentException("参数不正确"); 
    } 
    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.定义阻塞和非阻塞的方式提供对应的接口

// 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.进行线程池的关闭

// 7.线程池的关闭 
  private boolean receiveTask = true;  
  public void shutdown() { 
    // 7.1.队列不再接收线程 
    receiveTask = false; 
    // 7.2.关闭处于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(); 
  }

运行结果是:

准备消费线程 
准备消费线程 
准备消费线程 
准备消费线程 
准备消费线程 
开启线程 
消费线程 
准备消费线程 
开启线程 
消费线程 
准备消费线程 
开启线程 
消费线程 
准备消费线程 
。。。。。。

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注亿速云行业资讯频道,感谢您对亿速云的支持。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI