自定义线程池

转自http://sunnylocus.iteye.com/blog/223327

线程池的作用:

     线程池作用就是限制系统中执行线程的数量。
     根据系统的环境情况，可以自动或手动设置线程数量，达到运行的最佳效果；少了浪费了系统资源，多了造成系统拥挤效率不高。用线程池控制线程数量，其他线程 排队等候。一个任务执行完毕，再从队列的中取最前面的任务开始执行。若队列中没有等待进程，线程池的这一资源处于等待。当一个新任务需要运行时，如果线程 池中有等待的工作线程，就可以开始运行了；否则进入等待队列。

为什么要用线程池:

1. 减少了创建和销毁线程的次数，每个工作线程都可以被重复利用，可执行多个任务
2. 可以根据系统的承受能力，调整线程池中工作线线程的数目，防止因为因为消耗过多的内存，而把服务器累趴下(每个线程需要大约1MB内存，线程开的越多，消耗的内存也就越大，最后死机)

线程池类

package com.snow.junit;

import java.util.LinkedList;

public class MyThreadPool extends ThreadGroup{
    private boolean isClose = false;  //线程池是否关闭
    private LinkedList workQueue;     //工作队列
    private static int threadPollID = 1;  //线程池ID
    
    public MyThreadPool(int poolSize) {
        super("threadPoolId" + "");      //调用父类构造方法，指定ThreadGroup的名称
        setDaemon(true);                 //继承的方法，设置是否设置守护线程池
        workQueue = new LinkedList();     //创建工作队列
        for (int i = 0; i < poolSize; i++) {
            new WorkThread(i).start();    //创建并启动工作线程，创建线程池最大容量数的工作线程
        }
    }
    
    //向工作线程中添加一个任务，由工作线程来执行该任务
    public synchronized void execute(Runnable task){
        if (isClose) {
            throw new IllegalStateException();
        }
        if (task != null) {
            workQueue.add(task);    //向消息队列中添加一个任务
            notify();               //唤醒一个正在getTask()方法中等待任务的工作线程
        }
    }
    
    //从工作队列中取出一个任务，工作线程会调用该方法
    private synchronized Runnable getTask(int threadId) throws InterruptedException{
        while (workQueue.size() == 0) {
            if (isClose) {return null;}
            System.out.println("工作线程" + threadId + "等待任务");
            wait();                 //如果工作线程中没有任务，那么就等待着
        }
        System.out.println("工作线程" + threadId + "开始执行任务...");
        return (Runnable) workQueue.removeFirst();          //返回队列中的第一个元素，并从队列中删除
    }
    
    //等待工作线程把任务执行完成
    public void waitFinish(){
        synchronized (this) {
            isClose = true;
            notifyAll();        //唤醒所有还在getTask()方法中等待任务的工作线程
        }
        Thread[] threads = new Thread[activeCount()];   //activeCount()返回该线程组中活动线程的估计值
        int count = enumerate(threads);  //enumerate方法继承自ThreadGroup，根据活动的线程的估计值获得该线程组中当前所有活动的工作线程
        for (int i = 0; i < count; i++) {   //等待所有工作线程结束
            try {
                threads[i].join();    //等待工作线程结束
            } catch (InterruptedException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }
    
    //关闭线程池
    public synchronized void closePool(){
        if (!isClose) {
            waitFinish();   //等待工作线程执行完毕
            isClose = true;  
            workQueue.clear();  //清空工作队列
            interrupt();   //中断线程池所有的工作线程
        }
    }
    
    //工作线程，负责从工作队列中取出任务，并执行
    private class WorkThread extends Thread{
        private int id;
        public WorkThread(int id){
            //父类构造，将线程加入到当前ThreadPool线程组
            super(MyThreadPool.this, id + "");
            this.id = id;
        }
        public void run(){
            while (! isInterrupted()) { //继承自Thread，判断线程是否被中断
                Runnable task = null;
                try {
                    task = getTask(id);    //取出任务
                } catch (InterruptedException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                //如果getTask()返回null或者线程执行getTask()时被中断，则结束此线程
                if (task == null) {
                    return;
                }
                task.run();   //运行任务 
            }
        }
    }
    
}

测试类：

package com.snow.junit;

public class MyThreadPoolTest {

    public static void main(String[] args) {
        MyThreadPool threadPool = new MyThreadPool(4); //创建一个有个4工作线程的线程池 
        try {
            Thread.sleep(1000);   //休眠1秒,以便让线程池中的工作线程全部运行
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        for (int i = 0; i < 7; i++) {  //创建6个任务
            threadPool.execute(createTask(i));
        }
        threadPool.waitFinish();  //等待所有任务执行完毕  
        threadPool.closePool();   //关闭线程池  
    }
    
    private static Runnable createTask(final int taskID){
        return new Runnable() {
            
            @Override
            public void run() {
                System.out.println("Hello world!");
            }
        };
    }

}