import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 自定義拒絕策略Demo：詳細(xì)日志+線程池狀態(tài)監(jiān)控
 */
public class CustomRejectPolicyDemo {

    // 1. 實(shí)現(xiàn)RejectedExecutionHandler，自定義拒絕策略
    static class MyCustomRejectPolicy implements RejectedExecutionHandler {
        @Override
        public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
            // 核心：打印詳細(xì)拒絕日志 + 線程池實(shí)時(shí)狀態(tài)（生產(chǎn)級(jí)排查必備）
            String rejectLog = String.format(
                    "===== 任務(wù)被拒絕 =====\n" +
                    "拒絕時(shí)間：%s\n" +
                    "被拒任務(wù)：%s\n" +
                    "線程池狀態(tài)：運(yùn)行中=%s | 核心線程數(shù)=%d | 最大線程數(shù)=%d\n" +
                    "當(dāng)前狀態(tài)：活躍線程數(shù)=%d | 隊(duì)列任務(wù)數(shù)=%d | 已完成任務(wù)數(shù)=%d | 總?cè)蝿?wù)數(shù)=%d",
                    System.currentTimeMillis(),
                    r.toString(),
                    executor.isRunning(), // 線程池是否處于RUNNING狀態(tài)
                    executor.getCorePoolSize(),
                    executor.getMaximumPoolSize(),
                    executor.getActiveCount(), // 實(shí)時(shí)活躍線程數(shù)
                    executor.getQueue().size(), // 隊(duì)列中等待的任務(wù)數(shù)
                    executor.getCompletedTaskCount(), // 已完成的任務(wù)總數(shù)
                    executor.getTaskCount() // 提交的總?cè)蝿?wù)數(shù)（完成+執(zhí)行中+隊(duì)列中）
            );
            // 實(shí)際生產(chǎn)中可替換為SLF4J/Logback日志框架
            System.err.println(rejectLog);
            // 可選：根據(jù)業(yè)務(wù)需求擴(kuò)展（如任務(wù)持久化到MQ/Redis、觸發(fā)限流告警、主線程兜底執(zhí)行等）
        }
    }

    // 2. 自定義線程工廠（可選，用于設(shè)置線程名，便于排查任務(wù)執(zhí)行線程）
    static class MyThreadFactory implements ThreadFactory {
        private final AtomicInteger threadId = new AtomicInteger(1);
        private final String threadPrefix = "biz-worker-thread-";

        @Override
        public Thread newThread(Runnable r) {
            Thread thread = new Thread(r, threadPrefix + threadId.getAndIncrement());
            thread.setDaemon(false); // 非守護(hù)線程，保證任務(wù)執(zhí)行完成
            thread.setPriority(Thread.NORM_PRIORITY); // 正常優(yōu)先級(jí)
            return thread;
        }
    }

    // 主程序：測試自定義拒絕策略
    public static void main(String[] args) throws InterruptedException {
        // 線程池核心參數(shù)：故意設(shè)置小參數(shù)，方便觸發(fā)拒絕策略
        int corePoolSize = 2;    // 核心線程數(shù)
        int maximumPoolSize = 4; // 最大線程數(shù)
        long keepAliveTime = 1;  // 非核心線程空閑超時(shí)時(shí)間
        int queueCapacity = 2;   // 任務(wù)隊(duì)列容量（有限隊(duì)列，必選，否則無法觸發(fā)拒絕策略）

        // 3. 創(chuàng)建線程池，指定【自定義拒絕策略】
        ThreadPoolExecutor executor = new ThreadPoolExecutor(
                corePoolSize,
                maximumPoolSize,
                keepAliveTime,
                TimeUnit.SECONDS,
                new LinkedBlockingQueue<>(queueCapacity), // 有限容量隊(duì)列，觸發(fā)拒絕策略的關(guān)鍵
                new MyThreadFactory(), // 自定義線程工廠（可選）
                new MyCustomRejectPolicy() // 核心：指定自定義拒絕策略
        );

        try {
            // 提交7個(gè)任務(wù)，觸發(fā)拒絕策略（2核心+2隊(duì)列+2非核心=6，第7個(gè)任務(wù)被拒絕）
            for (int i = 1; i <= 7; i++) {
                int taskId = i;
                executor.submit(() -> {
                    try {
                        // 模擬任務(wù)執(zhí)行耗時(shí)1秒
                        TimeUnit.SECONDS.sleep(1);
                        System.out.printf("任務(wù)%d執(zhí)行成功，執(zhí)行線程：%s%n", taskId, Thread.currentThread().getName());
                    } catch (InterruptedException e) {
                        Thread.currentThread().interrupt();
                        System.out.printf("任務(wù)%d被中斷%n", taskId);
                    }
                });
                System.out.printf("任務(wù)%d已提交%n", taskId);
            }
        } finally {
            // 4. 優(yōu)雅關(guān)閉線程池（必須，避免資源泄漏）
            executor.shutdown();
            // 等待線程池終止，超時(shí)則強(qiáng)制關(guān)閉
            if (!executor.awaitTermination(5, TimeUnit.SECONDS)) {
                executor.shutdownNow();
            }
        }
    }
}