Java線程池擴展

有時需要對線程池進行擴展,如在監(jiān)控每個任務的開始和結束時間,或者自定義一些其他增強的功能。

ThreadPoolExecutor線程池提供了兩個方法：

● protected void afterExecute(Runnable r, Throwable t)

● protected void beforeExecute(Thread t, Runnable r)

在線程池執(zhí)行某個任務前會調用beforeExecute()方法,在任務結束后(任務異常退出)會執(zhí)行afterExecute()方法。

查看ThreadPoolExecutor源碼,在該類中定義了一個內部類Worker, ThreadPoolExecutor線程池中的工作線程就是Worker類的實例, Worker實例在執(zhí)行時會調用beforeExecute()與 afterExecute()方法。

package com.wkcto.threadpool;

import com.wkcto.produerstack.MyStack;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * 擴展線程池
 */
public class Test06 {
    //定義任務類
    private static class  MyTask implements  Runnable{
         String name;

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

        @Override
        public void run() {
            System.out.println(name + "任務正在被線程 " + Thread.currentThread().getId() + " 執(zhí)行");
            try {
                Thread.sleep(1000);     //模擬任務執(zhí)行時長
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    public static void main(String[] args) {
        //定義擴展線程池, 可以定義線程池類繼承ThreadPoolExecutor,在子類中重寫beforeExecute()/afterExecute()方法
        //也可以直接使用ThreadPoolExecutor的內部類
        ExecutorService executorService = new ThreadPoolExecutor(5, 5, 0, TimeUnit.SECONDS, new LinkedBlockingQueue<>() ){
            //在內部類中重寫任務開始方法
            @Override
            protected void beforeExecute(Thread t, Runnable r) {
                System.out.println(t.getId() + "線程準備執(zhí)行任務: " + ((MyTask)r).name);
            }

            @Override
            protected void afterExecute(Runnable r, Throwable t) {
                System.out.println( ((MyTask)r).name + "任務執(zhí)行完畢");
            }

            @Override
            protected void terminated() {
                System.out.println("線程池退出");
            }
        };

        //向線程池中添加任務
        for (int i = 0; i < 5; i++) {
            MyTask task = new MyTask("task-" + i);
            executorService.execute(task);
        }

        //關閉線程池
        executorService.shutdown();     //關閉線程池僅僅是說線程池不再接收新的任務 , 線程池中已接收的任務正常執(zhí)行完畢
    }
}