package com.wkcto.threadpool;

import java.util.Random;
import java.util.concurrent.*;

/**
 * 自定義拒絕策略
 */
public class Test03 {
    public static void main(String[] args) {
        //定義任務(wù)
        Runnable r = new Runnable() {
            @Override
            public void run() {
                int num = new Random().nextInt(5);
                System.out.println(Thread.currentThread().getId() + "--" + System.currentTimeMillis() + "開始睡眠" + num + "秒");
                try {
                    TimeUnit.SECONDS.sleep(num);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        };

        //創(chuàng)建線程池, 自定義拒絕策略
        ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(5, 5, 0, TimeUnit.SECONDS, new LinkedBlockingQueue<>(10), Executors.defaultThreadFactory(), new RejectedExecutionHandler(){
            @Override
            public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
                //r就是請求的任務(wù), executor就是當(dāng)前線程池
                System.out.println(r + " is discarding..");
            }
        });

        //向線程池提交若干任務(wù)
        for (int i = 0; i < Integer.MAX_VALUE; i++) {
            threadPoolExecutor.submit(r);
        }
    }
}