`
redsky008
  • 浏览: 346625 次
  • 性别: Icon_minigender_1
  • 来自: 成都正在项目
社区版块
存档分类
最新评论

另一个线程池ThreadPool,感觉比之前发的那个JDK自带的好用

阅读更多
import java.util.LinkedList;

public class ThreadPool extends ThreadGroup {
  private boolean isClosed=false;  //线程池是否关闭
  public LinkedList<Runnable> workQueue;  //表示工作队列
  private static int threadPoolID;  //表示线程池ID
  private int threadID;  //表示工作线程ID

  public ThreadPool(int poolSize) { //poolSize指定线程池中的工作线程数目
    super("ThreadPool-" + (threadPoolID++));
    setDaemon(true);
    workQueue = new LinkedList<Runnable>();  //创建工作队列
    for (int i=0; i<poolSize; i++)
      new WorkThread().start();  //创建并启动工作线程
  }

/** 向工作队列中加入一个新任务,由工作线程去执行该任务 */
  public synchronized void execute(Runnable task) {
    if (isClosed) { //线程池被关则抛出IllegalStateException异常
      throw new IllegalStateException();
    }
    if (task != null) {
      workQueue.add(task);
      notify();  //唤醒正在getTask()方法中等待任务的工作线程
    }
  }

  /** 从工作队列中取出一个任务,工作线程会调用此方法 */
  protected synchronized Runnable getTask()throws InterruptedException{
    while (workQueue.size() == 0) {
      if (isClosed) return null;
      wait();  //如果工作队列中没有任务,就等待任务
    }
    return workQueue.removeFirst();
  }

  /** 关闭线程池 */
  public synchronized void close() {
    if (!isClosed) {
      isClosed = true;
      workQueue.clear(); //清空工作队列
      interrupt();  //中断所有的工作线程,该方法继承自ThreadGroup类
    }
}

  /** 等待工作线程把所有任务执行完 */
  public void join() {
    synchronized (this) {
      isClosed = true;
      notifyAll();  //唤醒还在getTask()方法中等待任务的工作线程
    }

    Thread[] threads = new Thread[activeCount()];
    //enumerate()方法继承自ThreadGroup类,获得线程组中当前所有活着的工作线程
    int count = enumerate(threads); 
    for (int i=0; i<count; i++) { //等待所有工作线程运行结束
      try {
        threads[i].join();  //等待工作线程运行结束
        System.out.println("已完成一个任务!!");
      }catch(InterruptedException ex) { }
    }
  }

  /**  内部类:工作线程  */
  private class WorkThread extends Thread {
    public WorkThread() {
      //加入到当前ThreadPool线程组中
      super(ThreadPool.this,"WorkThread-" + (threadID++));
    }

    public void run() {
      while (!isInterrupted()) {  //isInterrupted()方法继承自Thread类,判断线程是否被中断
      Runnable task = null;
        try { //得到任务
          task = getTask();
        }catch (InterruptedException ex){}

        // 如果getTask()返回null或者线程执行getTask()时被中断,则结束此线程
        if (task == null) return;
       
        try { //运行任务,捕获异常
          task.run();
        } catch (Throwable t) {
          t.printStackTrace();
        }
      }//#while
    }//#run()
  }//#WorkThread类
}
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics