ThreadPoolExecutor 線程執行超時

 

package com.vipsoft.Thread;

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

public class ThreadPoolExecutorTest {

    public static void main(String[] args) throws Exception {
        int corePoolSize = 2;
        int maximumPoolSize = 5;
        long keepAliveTime = 10;
        TimeUnit unit = TimeUnit.SECONDS;
        BlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<>(3);  //定義一個大小爲2的隊列，只等有一個任務在排隊等，多出來的需要開新線程
        ThreadFactory threadFactory = new MyTreadFactory();
        RejectedExecutionHandler handler = new MyPolicy();
        ThreadPoolExecutor executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory, handler);
        System.out.println("預啓動線程（備戰）");
        executor.prestartAllCoreThreads(); // 預啓動所有核心線程，處於備戰
        System.out.println("預啓動線程數（備戰）：" + executor.getPoolSize());
        for (int i = 1; i <= 10; i++) {
            System.out.println(System.currentTimeMillis() + "  " + "開始 下發任務：" + i + " 當前線程總數：" + executor.getPoolSize());
            MyTask task = new MyTask(String.valueOf(i));
            executor.execute(task);
            System.out.println(System.currentTimeMillis() + "  " + "完成 下發任務：" + i + " 當前線程總數：" + executor.getPoolSize() + " 隊列中的線程數量：" + workQueue.size());
            Thread.sleep(1); //停1毫秒，日誌記錄，時間後方便分析
            if (i == 9) {
                //TODO Thread.sleep(3000); //任務9下發後【會被拒絕】，停3秒，等隊列或線程釋放後，再下發任務10，這時候任務10不會被拒絕
            }
        }
        System.in.read(); //阻塞主線程
    }

    static class MyTreadFactory implements ThreadFactory {

        private final AtomicInteger mThreadNum = new AtomicInteger(1);

        @Override
        public Thread newThread(Runnable r) {
            Thread t = new Thread(r, "Thread-" + mThreadNum.getAndIncrement());
            System.out.println(System.currentTimeMillis() + "  " + t.getName() + " has been created");
            return t;
        }
    }

    public static class MyPolicy implements RejectedExecutionHandler {

        public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {
            // 可做日誌記錄等
            System.err.println(System.currentTimeMillis() + "  " + r.toString() + " rejected from " + e.toString());
        }
    }

    static class MyTask implements Runnable {
        private String name;

        public MyTask(String name) {
            this.name = name;
        }

        @Override
        public void run() {
            try {
                Thread mainT = Thread.currentThread();
                String threadName = this.name;
                SubThread st = new SubThread(threadName, mainT);
                Thread thread = new Thread(st);
                thread.start();
                try {
                    System.out.println(System.currentTimeMillis() + "  " + this.toString() + " 開始運行! " + Thread.currentThread().getName());
                    Thread.sleep(Integer.valueOf(threadName) * 1000); //讓任務執行慢點
                    st.setDone();
                    System.out.println(System.currentTimeMillis() + "  " + this.toString() + " 運行結束! " + Thread.currentThread().getName());
                } catch (InterruptedException e) {
                    System.out.println("主線程，被打斷 " + threadName);
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        @Override
        public String toString() {
            return "MyTask [name=" + this.name + "]";
        }

        static class SubThread implements Runnable {
            private String name;
            private Thread mainT;

            //如果完成了就不用殺了。
            private boolean isDone;

            public void setDone() {
                isDone = true;
            }

            public SubThread(String name, Thread mainT) {
                this.name = name;
                this.mainT = mainT;
            }

            @Override
            public void run() {
                try {
                    System.out.println("子線程，正在監控主進程 " + name);
                    Thread.sleep(5000);
                    //時間到了，主線程還沒有完成就幹掉。
                    if (!isDone) {
                        mainT.interrupt();
                        System.out.println("子線程，幹掉了主進程（超時了） " + name);
                    }
                } catch (InterruptedException e) {
                    System.out.println("子線程，被打斷");
                    return;
                }
            }

            @Override
            public String toString() {
                return "MyTask [name=" + this.name + "]";
            }
        }
    }


}