手写简易版线程池

我是陈皮，一个在互联网 Coding 的 ITer，个人微信公众号「陈皮的JavaLib」关注第一时间阅读最新文章。

简易版线程池定义类如下：

package com.chenpi;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;

/**
 * @author 陈皮
 * @version 1.0
 * @description 自定义线程池, 本实现没考虑空闲线程处理，拒绝策略等.
 * @date 2020/06/02
 */
public class MyExecutorService {

  // 线程池内一直运行的线程
  private List<WorkThread> workThreads;

  // 存放未执行任务的队列容器
  private final BlockingQueue<Runnable> workQueue;

  // 标志线程池关闭与否
  private volatile boolean isWorking = true;

  // 构造函数，初始化线程池
  public MyExecutorService(int corePoolSize, int queueCapacity) {
    workThreads = new ArrayList<>(corePoolSize);
    workQueue = new LinkedBlockingQueue<>(queueCapacity);
    // 提前预热创建固定数量的线程，保持运行状态
    for (int i = 0; i < corePoolSize; i++) {
      WorkThread workThread = new WorkThread();
      workThread.start();
      workThreads.add(workThread);
    }
  }

  // 执行任务，实际上先放入任务队列中
  public boolean execute(Runnable task) {
    if (null != task) {
      System.out.println("执行任务，此时队列中任务数:" + workQueue.size());
      return workQueue.offer(task);
    }
    return false;
  }

  // 优雅关闭线程池
  public void shutdown() {
    isWorking = false;
    workThreads.clear();
    workThreads = null;
  }

  // 工作线程类
  class WorkThread extends Thread {

    @Override
    public void run() {
      // 如果线程池未关闭，并且任务队列还有未执行任务，则线程不断从任务队列取任务，如果能取到则执行任务的run方法
      while (isWorking || !workQueue.isEmpty()) {
        Runnable task = workQueue.poll();
        if (null != task) {
          task.run();
        }
      }
    }
  }
}

测试类如下所示。

package com.chenpi;

/**
 * @author 陈皮
 * @version 1.0
 * @description 测试线程池
 * @date 2020/06/02
 */
public class Main {

  public static void main(String[] args) {

    // 1.创建线程池
    MyExecutorService myExecutorService = new MyExecutorService(5, 100);

    // 2.创建15个任务
    for (int i = 0; i < 15; i++) {
      final int finalI = i;
      myExecutorService.execute(() -> {
        System.out.println(Thread.currentThread().getName() + " 执行... i=" + finalI);
        try {
          Thread.sleep((long) (Math.random() * 5000));
        } catch (InterruptedException e) {
          e.printStackTrace();
        }
      });
    }

    // 3.关闭线程池
    myExecutorService.shutdown();
  }
}

测试结果如下所示。

执行任务，此时队列中任务数:0
执行任务，此时队列中任务数:1
执行任务，此时队列中任务数:1
执行任务，此时队列中任务数:1
Thread-4 执行... i=1
执行任务，此时队列中任务数:1
Thread-1 执行... i=2
Thread-2 执行... i=0
执行任务，此时队列中任务数:1
执行任务，此时队列中任务数:2
执行任务，此时队列中任务数:3
执行任务，此时队列中任务数:4
执行任务，此时队列中任务数:5
执行任务，此时队列中任务数:6
执行任务，此时队列中任务数:7
执行任务，此时队列中任务数:8
Thread-0 执行... i=3
Thread-3 执行... i=4
执行任务，此时队列中任务数:9
执行任务，此时队列中任务数:9
Thread-4 执行... i=5
Thread-3 执行... i=6
Thread-3 执行... i=7
Thread-0 执行... i=8
Thread-4 执行... i=9
Thread-0 执行... i=10
Thread-1 执行... i=11
Thread-4 执行... i=12
Thread-2 执行... i=13
Thread-3 执行... i=14

本次分享到此结束啦~~

如果觉得文章对你有帮助，点赞、收藏、关注、评论，您的支持就是我创作最大的动力！