码迷,mamicode.com
首页 > 编程语言 > 详细

Java线程池的实现

时间:2016-05-14 16:45:17      阅读:213      评论:0      收藏:0      [点我收藏+]

标签:

1. 线程池类

import java.util.LinkedList;
import java.util.List;

public class MyThreadPool {
    // 线程池中默认线程的个数为5个
    private static int worker_num = 5;
    // 工作的线程
    private WorkThread[] workThreads;
    // 未处理的任务
    private static volatile int finished_task = 0;
    // 任务队列,作为一个缓冲,List线程不安全所以需要在使用的过程中对它进行同步。
    private List<Runnable> taskQueue = new LinkedList<Runnable>();
    
    // 单例模式
    private static MyThreadPool threadPool;
    // 私有化构造方法
    private MyThreadPool(){
        this(5);
    }
    // 创建线程池,num为线程池工作线程的个数
    private MyThreadPool(int num) {
        MyThreadPool.worker_num = num;
        workThreads = new WorkThread[num];
        for (int i = 0; i < num; i++) {
            workThreads[i] = new WorkThread();
            workThreads[i].start();
        }
    }
    
    // 获得一个线程池,默认线程数
    public static MyThreadPool getThreadPool(){
        return getThreadPool(MyThreadPool.worker_num);
    }
    // 单例模式,获得一个线程池
    public static MyThreadPool getThreadPool(int num) {
        if(num <= 0){
            num = MyThreadPool.worker_num;
        }
        if(threadPool == null){
            synchronized(MyThreadPool.class){
                if(threadPool == null)
                    threadPool = new MyThreadPool(num);
            }
        }
        return threadPool;
    }

    /**
     * 执行任务
     * 将该任务加入到任务队列的末尾,等待工作线程的调度
     * @param task
     */
    public void execute(Runnable task){
        synchronized (taskQueue){
            taskQueue.add(task);
            taskQueue.notify();
        }
    }
    
    /**
     * 批量执行任务
     * 将任务放到任务队列的末尾,等待工作线程的调度
     * @param task
     */
    public void execute(Runnable[] task){
        synchronized (taskQueue){
            for (Runnable runnable : task) {
                taskQueue.add(runnable);
            }
            taskQueue.notify();
        }
    }
    
    /**
     * 批量执行任务
     * 将任务放到任务队列的末尾,等待工作线程的调度
     * @param task
     */
    public void execute(List<Runnable> task){
        synchronized (taskQueue){
            for (Runnable runnable : task) {
                taskQueue.add(runnable);
            }
            taskQueue.notify();
        }
    }
    
    /**
     * 销毁线程池
     * 在所有任务都完成的情况下才销毁所有线程,否则等待任务队列的任务全部完成才销毁
     */
    public void destroy(){
        while(!taskQueue.isEmpty()){// 如果还有任务没执行完成,就等会再看看
            try {
                Thread.sleep(10);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        
        // 工作线程停止工作,且置为null
        for (int i = 0; i < MyThreadPool.worker_num; i++) {
            workThreads[i].stopWorker();
            workThreads[i] = null;
        }
        threadPool = null;
        taskQueue.clear();
    }
    
    // 返回工作线程的个数
    public int getWorkThreadNumber() {
        return worker_num;
    }
    
    // 返回已完成任务的个数
    public int getFinishedTaskNumber(){
        return finished_task;
    }
    
    // 返回任务队列的长度,即还没处理的任务个数
    public int getWaitTaskNumber(){
        return taskQueue.size();
    }
    
    // 覆盖toString方法,返回线程池信息:工作线程个数和已完成任务个数  
    @Override  
    public String toString() {  
        return "WorkThread number:" + worker_num + "  finished task number:"  
                + finished_task + "  wait task number:" + getWaitTaskNumber();  
    }
    
    /**
     * 内部类,工作线程用来执行任务线程
     * @author liu
     *
     */
    private class WorkThread extends Thread{
        // 该工作线程是否有效,用于自然结束该工作线程
        private boolean isRunning = true;
        
        /* 
         * 工作线程的关键之处,如果任务队列不空,则取出任务执行,若任务队列空,则等待。直到任务队列有任务时才取出执行
         */
        @Override
        public void run() {
            Runnable r = null;
            while( isRunning ){// 队列为空
                synchronized (taskQueue){
                    while(isRunning && taskQueue.isEmpty()){
                        try {
                            taskQueue.wait(20);
                        } catch (InterruptedException e) {
                            e.printStackTrace();
                        }
                    }
                    
                    if(!taskQueue.isEmpty()){
                        // 取出任务
                        r = taskQueue.remove(0);
                    }
                }
                
                if(r != null){
                    r.run();
                    // 完成的任务加一
                    finished_task += 1;
                }
                
                r = null;
            }
        }
        
        // 停止工作,让该线程自然执行完run方法,自然结束 
        public void stopWorker() {
            this.isRunning = false;
        }
        
    }
}

 

 2. 测试代码

public class TestThreadPool {

    public static void main(String[] args) {
        // 创建5个线程的线程池  
        MyThreadPool t = MyThreadPool.getThreadPool(5);
        Runnable[] r = new Runnable[] { new Task(), new Task(), new Task()};
        t.execute(r);
        r = new Runnable[] {new Task(), new Task(), new Task()};
        t.execute(r);  
        System.out.println(t);
        t.destroy();// 所有线程都执行完成才destory  
        System.out.println(t); 
    }
    
    // 任务类
    static class Task implements Runnable {  
        private static volatile int i = 1;
  
        @Override  
        public void run() {// 执行任务  
            System.out.println("任务 " + (i++) + " 完成");  
        }
    }
}

3. 运行结果

技术分享

注:参照网上的代码和思路,通过自己的修改和调试完成

Java线程池的实现

标签:

原文地址:http://www.cnblogs.com/liuleicode/p/5492724.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!