package com.zxz.pgdemo.controller;

import java.util.concurrent.*;

public class RejectHandlerDemo {
    public static class MyTask implements Runnable {
        public Object data;
        public CountDownLatch countDownLatch;
        public MyTask(Object data, CountDownLatch countDownLatch) {
            this.data=data;
            this.countDownLatch=countDownLatch;
        }

        @Override
        public void run() {
            doWork(data);
            countDownLatch.countDown();
        }

        public void doWork(Object data) {
            System.out.println(System.currentTimeMillis() + ":Thread ID:" + Thread.currentThread().getId() + ";data=" + data.toString());
        }
    }


    public static void main(String args[]) throws InterruptedException {

        LinkedBlockingDeque<Runnable> queue = new LinkedBlockingDeque<>(10);
        ExecutorService executorService = new ThreadPoolExecutor(5, 5,
                0L, TimeUnit.SECONDS,
                queue,
                Executors.defaultThreadFactory()
                , (r, executor) -> {
                    try {
                        queue.put(r);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                        System.out.println(r.toString() + " is discard");
                    }

                });

        CountDownLatch countDownLatch = new CountDownLatch(100);
        for (int i = 0; i < 100; i++) {
            executorService.submit(new MyTask(i, countDownLatch));
            Thread.sleep(10);
        }
        countDownLatch.await();
        System.out.println("---------done--------");
        executorService.shutdown();
    }
}