Java并发编程---自定义线程池

古城微笑少年丶 2022-06-02 07:06 147阅读 0赞

**一.概念**

**若Executors工厂类无法满足我们的需求,可以自己去创建自定义的线程池,其实Executors工厂类里面的创建线程方法,其内部实现均是用了ThreadPoolExecutor这个类,这个类可以自定义线程.构造方法如下:**

public ThreadPoolExecutor(int corePoolSize,
                                  int maximumPoolSize,
                                  long keepAliveTime,
                                  TimeUnit unit,
                                  BlockingQueue<Runnable> workQueue,
                                  RejectedExecutionHandler handler) {
            this(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue,
                 Executors.defaultThreadFactory(), handler);

**二.代码示例**

package com.thread.executors;
    
    public class MyTask implements Runnable {
    
    	private int taskId;
    	private String taskName;
    
    	public int getTaskId() {
    		return taskId;
    	}
    
    	public void setTaskId(int taskId) {
    		this.taskId = taskId;
    	}
    
    	public String getTaskName() {
    		return taskName;
    	}
    
    	public void setTaskName(String taskName) {
    		this.taskName = taskName;
    	}
    
    	public MyTask(int taskId, String taskName) {
    		super();
    		this.taskId = taskId;
    		this.taskName = taskName;
    	}
    
    	@Override
    	public void run() {
    		// TODO Auto-generated method stub
             try {
            	System.out.println("run taskId = " + this.taskId);
    			Thread.sleep(5*1000);
    		} catch (InterruptedException e) {
    			// TODO Auto-generated catch block
    			e.printStackTrace();
    		}
    	}
    
    	@Override
    	public String toString() {
    		return Integer.toString(this.taskId);
    	}
    }

package com.thread.executors;
    
    import java.util.concurrent.RejectedExecutionHandler;
    import java.util.concurrent.ThreadPoolExecutor;
    
    /**
     * 自定义的线程拒绝策略
     * @author lhy
     * @time   2017.01.07
     */
    public class MyRejected implements RejectedExecutionHandler {
    	
    	public MyRejected() {
    		super();
    		// TODO Auto-generated constructor stub
    	}
    
    	@Override
    	public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
    		// TODO Auto-generated method stub
            System.out.println("自定义处理...");
            System.out.println("当前被拒绝的任务为: " + r.toString());
    	}
    
    }

package com.thread.executors;
    
    import java.util.concurrent.ArrayBlockingQueue;
    import java.util.concurrent.ThreadPoolExecutor;
    import java.util.concurrent.TimeUnit;
    
    import com.thread.safety.MyThread;
    
    /**
     * 自定义线程池(使用有界队列)
     * @author lhy
     * @time  2017.01.05
     *
     */
    public class UseThreadPoolExecutor1 {
    
    	public static void main(String[] args) {
    		// TODO Auto-generated method stub
    
    		/**
    		 * 在使用有界队列时,若有新的任务需要执行,如果线程池实际线程数小于corePoolSize,则优先创建线程,
    		 * 若大于corePoolSize,则会将任务加入队列,
    		 * 若队列已满,则在总线程数不大于MaxNumPoolSize的前提下,创建新的线程,
    		 * 若线程数大于MaxNumPoolSize,则执行拒绝策略,或其他自定义方式
    		 */
    		ThreadPoolExecutor pool = new ThreadPoolExecutor(
    				1,              //coreSize
    				2,              //MaxSize
    				60,             //60
    				TimeUnit.SECONDS,    //失效时间,缓存时间
    				new ArrayBlockingQueue<Runnable>(3),       //指定一种队列,有界队列
    				new MyRejected()                           //自定义的拒绝策略
    				);
    		
    		MyTask mt1 = new MyTask(1, "任务1");
    		MyTask mt2 = new MyTask(2, "任务2");
    		MyTask mt3 = new MyTask(3, "任务3");
    		MyTask mt4 = new MyTask(4, "任务4");
    		MyTask mt5 = new MyTask(5, "任务5");
    		MyTask mt6 = new MyTask(6, "任务6");
    		
    		pool.execute(mt1);
    		pool.execute(mt2);
    		pool.execute(mt3);
    		pool.execute(mt4);
    		pool.execute(mt5);
    		pool.execute(mt6);
    		
    		pool.shutdown();
    	}
    
    }

package com.thread.executors;
    
    import java.util.concurrent.ArrayBlockingQueue;
    import java.util.concurrent.BlockingQueue;
    import java.util.concurrent.ExecutorService;
    import java.util.concurrent.LinkedBlockingQueue;
    import java.util.concurrent.ThreadPoolExecutor;
    import java.util.concurrent.TimeUnit;
    import java.util.concurrent.atomic.AtomicInteger;
    
    /**
     * 自定义线程池(使用无界队列)
     * @author lhy
     * @time   2017.01.05
     *
     */
    public class UseThreadPoolExecutor2 implements Runnable {
    
    	private static AtomicInteger count = new AtomicInteger(0);
    
    	@Override
    	public void run() {
    		// TODO Auto-generated method stub
    		try {
    			int temp = count.incrementAndGet();    //原子性操作,任务+1
    			System.out.println("任务" + temp);
    			Thread.sleep(2000);
    		} catch (InterruptedException e) {
    			// TODO Auto-generated catch block
    			e.printStackTrace();
    		}
    	}
    
    	public static void main(String[] args) throws Exception {
    		/**
    		 * 无界的任务队列时: LinkedBlockingQueue,
    		 * 与有界的任务队列相比,除非系统资源耗尽,否则无界的任务队列不存在任务入队失败的情况.
    		 * 当有新任务到来时,系统的线程数小于CorePoolSize时,则新建线程执行任务,
    		 * 当达到CorePoolSize后.就不会继续增加,若后续仍有新的任务加入,而又没有空闲的线程资源,则任务直接进入队列等待.
    		 * 当任务的创建和处理的速度差异很大,无界队列会保持快速增长,直到耗尽系统内存
    		 */
    		// TODO Auto-generated method stub
    		BlockingQueue<Runnable> queue = 
    				new LinkedBlockingQueue<Runnable>(10);    //无界队列,扔到线程池
    //				new ArrayBlockingQueue<Runnable>(10);
    		ExecutorService executor = new ThreadPoolExecutor(
    				5, // core
    				10,      //无意义,怎么设都行
    				120L, // 120秒
    				TimeUnit.SECONDS,
    				queue);
             
    		for(int i=0; i<20; i++){
    			executor.execute(new UseThreadPoolExecutor2());
    		}
    		Thread.sleep(1000);
    		System.out.println("queue Size:" + queue.size());
    		Thread.sleep(2000);
    	}
    
    }

**三.代码分析(自定义线程池使用说明)**

**这个构造方法对于队列是什么类型的比较关键:**

**3.1 在使用有界队列时,若有新的任务需要执行,如果线程池实际线程数小于corePoolSize,则优先创建线程,若大于corePoolSize,则会将任务加入队列,若队列已满,则在总线程数不大于MaxNumPoolSize的前提下,创建新的线程,若线程数大于MaxNumPoolSize,则执行拒绝策略.或其他自定义方式**

**3.2无界的任务队列时: LinkedBlockingQueue,与有界的任务队列相比,除非系统资源耗尽,否则无界的任务队列不存在任务入队失败的情况.当有新任务到来时,系统的线程数小于CorePoolSize时,则新建线程执行任务,当达到CorePoolSize后.就不会继续增加,若后续仍有新的任务加入,而又没有空闲的线程资源,则任务直接进入队列等待.当任务的创建和处理的速度差异很大,无界队列会保持快速增长,直到耗尽系统内存**

**3.3 JDK拒绝策略:**

**AbortPolicy:直接抛出异常,组织系统正常工作**

**CallerRunsPolicy:只要线程池未关闭,该策略直接在调用者线程中,运行当前被丢弃的任务**

**DiscardOldestPolicy:丢弃最老的一个请求,尝试再次提交当前任务**

**DiscardPolicy:丢弃无法处理的任务,不给予任何处理**

**如果需要自定义拒绝策略可以实现RejectedExecutionHandler接口**

**四.运行结果**

**4.1 有界队列**

run taskId = 1
    自定义处理...
    当前被拒绝的任务为: 6
    run taskId = 5
    run taskId = 2
    run taskId = 3
    run taskId = 4

**4.2 无界队列**

任务1
    任务3
    任务4
    任务2
    任务5
    任务6
    任务7
    任务8
    任务9
    任务10
    queue Size:10
    任务11
    任务12
    任务13
    任务14
    任务15
    任务16
    任务17
    任务18
    任务19
    任务20