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 taskName = this.name;
                SubThread st = new SubThread(taskName, mainT);
                Thread thread = new Thread(st);
                thread.start();
                try {
                    long millis = 1000;
                    if (Integer.valueOf(taskName) % 2 == 0) {
                        millis = Integer.valueOf(taskName) * 2 * millis; //讓任務執行慢點
                    }
                    System.out.println(System.currentTimeMillis() + "  " + this.toString() + " 開始運行! 運行線程 " + Thread.currentThread().getName() + " 需要(秒):" + millis);
                    Thread.sleep(millis); //讓任務執行慢點--這邊還是有點問題,如果這個 sleep 過長的話,還是需要等它執行完,並不能解決線程超時殺掉它的目的
                    System.out.println(System.currentTimeMillis() + "  " + this.toString() + " 運行結束! 運行線程 " + Thread.currentThread().getName());
                } catch (InterruptedException e) {
                    System.out.println(System.currentTimeMillis() + "  " + this.toString() + " 被打斷 運行線程 " + Thread.currentThread().getName());
                } finally {
                    st.setDone(); //通知監控線程,主線程運行結束
                    //TODO thread.interrupt(); //父線程也可以中止子線程的任務執行
                }

            } 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(System.currentTimeMillis() + " 子線程 " + Thread.currentThread().getName() + "【正在】監控主進程 " + mainT.getName() + " 任務" + name);
                    Thread.sleep(5000);
                    //時間到了,主線程還沒有完成就幹掉。
                    if (!isDone) {
                        mainT.interrupt();
                        System.out.println(System.currentTimeMillis() + " 子線程 " + Thread.currentThread().getName() + "【殺掉】主進程(超時了) " + mainT.getName() + " 任務" + name);
                    }
                } catch (InterruptedException e) {
                    System.out.println(System.currentTimeMillis() + " 子線程 " + Thread.currentThread().getName() + "【被打斷】" + mainT.getName() + " 任務" + name);
                } finally {
                    System.out.println(System.currentTimeMillis() + " 子線程 " + Thread.currentThread().getName() + "【完成】監控主進程 " + mainT.getName() + " 任務" + name);
                }
            }

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


}

 

 日誌排序後:

1655083181083  Thread-1 has been created
1655083181083  Thread-2 has been created
1655083181083  開始 下發任務:1 當前線程總數:2
1655083181084  完成 下發任務:1 當前線程總數:2 隊列中的線程數量:1
1655083181085  MyTask [name=1] 開始運行! 運行線程 Thread-2 需要(秒):1000
1655083181085 子線程 Thread-0【正在】監控主進程 Thread-2 任務1
1655083181086  MyTask [name=2] 開始運行! 運行線程 Thread-1 需要(秒):4000
1655083181086  開始 下發任務:2 當前線程總數:2
1655083181086  完成 下發任務:2 當前線程總數:2 隊列中的線程數量:1
1655083181086 子線程 Thread-1【正在】監控主進程 Thread-1 任務2
1655083181088  開始 下發任務:3 當前線程總數:2
1655083181088  完成 下發任務:3 當前線程總數:2 隊列中的線程數量:1
1655083181089  開始 下發任務:4 當前線程總數:2
1655083181089  完成 下發任務:4 當前線程總數:2 隊列中的線程數量:2
1655083181091  開始 下發任務:5 當前線程總數:2
1655083181091  完成 下發任務:5 當前線程總數:2 隊列中的線程數量:3
1655083181093  MyTask [name=6] 開始運行! 運行線程 Thread-3 需要(秒):12000
1655083181093  Thread-3 has been created
1655083181093  開始 下發任務:6 當前線程總數:2
1655083181093  完成 下發任務:6 當前線程總數:3 隊列中的線程數量:3
1655083181093 子線程 Thread-2【正在】監控主進程 Thread-3 任務6
1655083181095  MyTask [name=7] 開始運行! 運行線程 Thread-4 需要(秒):1000
1655083181095  Thread-4 has been created
1655083181095  開始 下發任務:7 當前線程總數:3
1655083181095  完成 下發任務:7 當前線程總數:4 隊列中的線程數量:3
1655083181095 子線程 Thread-3【正在】監控主進程 Thread-4 任務7
1655083181097  MyTask [name=8] 開始運行! 運行線程 Thread-5 需要(秒):16000
1655083181097  Thread-5 has been created
1655083181097  開始 下發任務:8 當前線程總數:4
1655083181097  完成 下發任務:8 當前線程總數:5 隊列中的線程數量:3
1655083181097 子線程 Thread-4【正在】監控主進程 Thread-5 任務8
1655083181099  MyTask [name=9] rejected from java.util.concurrent.ThreadPoolExecutor@4b1210ee[Running, pool size = 5, active threads = 5, queued tasks = 3, completed tasks = 0]
1655083181099  開始 下發任務:9 當前線程總數:5
1655083181099  完成 下發任務:9 當前線程總數:5 隊列中的線程數量:3
1655083181101  MyTask [name=10] rejected from java.util.concurrent.ThreadPoolExecutor@4b1210ee[Running, pool size = 5, active threads = 5, queued tasks = 3, completed tasks = 0]
1655083181101  開始 下發任務:10 當前線程總數:5
1655083181101  完成 下發任務:10 當前線程總數:5 隊列中的線程數量:3
1655083182085  MyTask [name=1] 運行結束! 運行線程 Thread-2
1655083182085  MyTask [name=3] 開始運行! 運行線程 Thread-2 需要(秒):1000
1655083182085 子線程 Thread-5【正在】監控主進程 Thread-2 任務3
1655083182095  MyTask [name=4] 開始運行! 運行線程 Thread-4 需要(秒):8000
1655083182095  MyTask [name=7] 運行結束! 運行線程 Thread-4
1655083182095 子線程 Thread-6【正在】監控主進程 Thread-4 任務4
1655083183090  MyTask [name=3] 運行結束! 運行線程 Thread-2
1655083183090  MyTask [name=5] 開始運行! 運行線程 Thread-2 需要(秒):1000
1655083183090 子線程 Thread-7【正在】監控主進程 Thread-2 任務5
1655083184100  MyTask [name=5] 運行結束! 運行線程 Thread-2
1655083185087  MyTask [name=2] 運行結束! 運行線程 Thread-1
1655083186085 子線程 Thread-0【完成】監控主進程 Thread-2 任務1
1655083186086 子線程 Thread-1【完成】監控主進程 Thread-1 任務2
1655083186094  MyTask [name=6] 被打斷 運行線程 Thread-3
1655083186094 子線程 Thread-2【殺掉】主進程(超時了) Thread-3 任務6
1655083186094 子線程 Thread-2【完成】監控主進程 Thread-3 任務6
1655083186096 子線程 Thread-3【完成】監控主進程 Thread-4 任務7
1655083186098  MyTask [name=8] 被打斷 運行線程 Thread-5
1655083186098 子線程 Thread-4【殺掉】主進程(超時了) Thread-5 任務8
1655083186098 子線程 Thread-4【完成】監控主進程 Thread-5 任務8
1655083187087 子線程 Thread-5【完成】監控主進程 Thread-2 任務3
1655083187103  MyTask [name=4] 被打斷 運行線程 Thread-4
1655083187103 子線程 Thread-6【殺掉】主進程(超時了) Thread-4 任務4
1655083187103 子線程 Thread-6【完成】監控主進程 Thread-4 任務4
1655083188098 子線程 Thread-7【完成】監控主進程 Thread-2 任務5
預啓動線程(備戰)
預啓動線程數(備戰):2

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章