Java 并发编程之自定义线程池 ThreadPoolExecutor

1）定义一个任务线程

public class Task implements Runnable {

    private String name;

    Task(String name) {
        this.name = name;
    }

    @Override
    public void run() {
        System.out.println(Thread.currentThread().getName() + " Start." + name);
        try {
            TimeUnit.SECONDS.sleep(5);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println(Thread.currentThread().getName() + " End.");
    }

    @Override
    public String toString() {
        return this.name;
    }
}

2）拒绝策略

public class CustomizedRejectedExecutionHandler implements RejectedExecutionHandler {
    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        System.out.println(r.toString() + " is rejected!");
    }
}

3）自定义线程池

public class ThreadPool {
    public static void main(String[] args) {
        CustomizedRejectedExecutionHandler rejectedExecutionHandler = new CustomizedRejectedExecutionHandler();
        ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(
                // 核心线程数
                3
                // 最大线程数
                ,6
                // 生存时间
                ,10
                // 生产时间单位
                ,TimeUnit.SECONDS
                // 任务队列，最多容纳线程个数
                ,new ArrayBlockingQueue<Runnable>(4)
                // 线程工厂
                , Executors.defaultThreadFactory()
                // 拒绝策略，线程池装满并且队列也装满的情况下，任务线程会被如何处理
                , rejectedExecutionHandler);

        for (int i=0; i<10; i++) {
            Task task = new Task(i + "");
            threadPoolExecutor.execute(task);
        }

        try {
            Thread.sleep(30000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        //shut down the pool
        threadPoolExecutor.shutdown();
    }
}