java 阻塞线程池_JAVA自定义阻塞型线程池

package com.montnets.task;

import java.util.concurrent.ArrayBlockingQueue;

import java.util.concurrent.RejectedExecutionHandler;

import java.util.concurrent.ThreadFactory;

import java.util.concurrent.ThreadPoolExecutor;

import java.util.concurrent.TimeUnit;

import java.util.concurrent.atomic.AtomicInteger;/**

* 自定义阻塞型线程池 当池满时会阻塞任务提交

*

* @ClassName: BlockThreadPool

* @Description: TODO

* @author: wangs

* @date: 2018-1-24 下午5:24:54*/

public classBlockThreadPool {private ThreadPoolExecutor pool = null;public BlockThreadPool(intpoolSize) {

pool= new ThreadPoolExecutor(poolSize, poolSize, 0, TimeUnit.MILLISECONDS, new ArrayBlockingQueue(5), newCustomThreadFactory(),newCustomRejectedExecutionHandler());

}public voiddestory() {if (pool != null) {

pool.shutdownNow();

}

}private classCustomThreadFactory implements ThreadFactory {private AtomicInteger count = new AtomicInteger(0);

@OverridepublicThread newThread(Runnable r) {

Thread t= newThread(r);

String threadName= BlockThreadPool.class.getSimpleName() + count.addAndGet(1);

t.setName(threadName);returnt;

}

}private classCustomRejectedExecutionHandler implements RejectedExecutionHandler {

@Overridepublic voidrejectedExecution(Runnable r, ThreadPoolExecutor executor) {try{//核心改造点，由blockingqueue的offer改成put阻塞方法

executor.getQueue().put(r);

}catch(InterruptedException e) {

e.printStackTrace();

}

}

}public voidexecute(Runnable runnable) {this.pool.execute(runnable);

}//测试构造的线程池

public static voidmain(String[] args) {

BlockThreadPool pool= new BlockThreadPool(3);for (int i = 1; i < 100; i++) {

System.out.println("提交第" + i + "个任务!");

pool.execute(newRunnable() {

@Overridepublic voidrun() {try{

System.out.println(Thread.currentThread().getId() + "=====开始");

TimeUnit.SECONDS.sleep(10);

System.out.println(Thread.currentThread().getId() + "=====【结束】");

}catch(InterruptedException e) {

e.printStackTrace();

}

}

});

System.out.println("【提交第" + i + "个任务成功！】");

}//2.销毁----此处不能销毁,因为任务没有提交执行完,如果销毁线程池,任务也就无法执行了//exec.destory();

try{

Thread.sleep(10000);

}catch(InterruptedException e) {

e.printStackTrace();

}

}

}