import com.alibaba.ttl.TransmittableThreadLocal;
import com.alibaba.ttl.TtlRunnable;
import com.alibaba.ttl.threadpool.TtlExecutors;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;

import java.util.Date;
import java.util.concurrent.Executor;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * TransmittableThreadLocal 測(cè)試
 */
@Slf4j
class TTLest {

    @Test
    void test1() throws InterruptedException {
        doTest();
    }

    /**
     * TTL 測(cè)試
     *
     * @throws InterruptedException
     */
    public void doTest() throws InterruptedException {
        TransmittableThreadLocal<String> context = new TransmittableThreadLocal<>();
//        ThreadLocal<String> context = new ThreadLocal<>();

        // 任務(wù)的具體方法
        Runnable runnable = () -> {
            System.out.println("當(dāng)前任務(wù)被執(zhí)行,執(zhí)行時(shí)間:" + new Date() + " 執(zhí)行線(xiàn)程:" + Thread.currentThread().getName() + "thread2 父線(xiàn)程數(shù)據(jù){}" + context.get());
            context.set("value-set-in-son");
            System.out.println("當(dāng)前任務(wù)被執(zhí)行,執(zhí)行時(shí)間:" + new Date() + " 執(zhí)行線(xiàn)程:" + Thread.currentThread().getName() + "thread2 父線(xiàn)程數(shù)據(jù){}" + context.get());
            try {
                // 等待 1s
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        };
        // 創(chuàng)建線(xiàn)程,線(xiàn)程的任務(wù)隊(duì)列的長(zhǎng)度為 1
        ThreadPoolExecutor threadPool = new ThreadPoolExecutor(1, 1,
                100, TimeUnit.SECONDS, new LinkedBlockingQueue<>(1),
                new ThreadPoolExecutor.CallerRunsPolicy());

        log.error("錯(cuò)誤使用方式:1-----------------------------------");
        context.set("value-set-in-parent");

        // 添加并執(zhí)行 4 個(gè)任務(wù),模擬拒絕策略CallerRunsPolicy
        threadPool.execute(runnable);
        threadPool.execute(runnable);
        threadPool.execute(runnable);
        threadPool.execute(runnable);
        threadPool.execute(runnable);
        TimeUnit.SECONDS.sleep(5);

        log.error("正確使用方式:1-----------------------------------");
        context.set("value-set-in-parent");

        // 添加并執(zhí)行 4 個(gè)任務(wù),模擬拒絕策略CallerRunsPolicy
        threadPool.execute(TtlRunnable.get(runnable));
        threadPool.execute(TtlRunnable.get(runnable));
        threadPool.execute(TtlRunnable.get(runnable));
        threadPool.execute(TtlRunnable.get(runnable));
        threadPool.execute(TtlRunnable.get(runnable));
        TimeUnit.SECONDS.sleep(5);

        log.error("正確使用方式:2-----------------------------------");
        context.set("value-set-in-parent");

        Executor ttlThreadPool = TtlExecutors.getTtlExecutor(threadPool);
        // 添加并執(zhí)行 4 個(gè)任務(wù),模擬拒絕策略CallerRunsPolicy
        ttlThreadPool.execute(runnable);
        ttlThreadPool.execute(runnable);
        ttlThreadPool.execute(runnable);
        ttlThreadPool.execute(runnable);
        ttlThreadPool.execute(runnable);

        // 線(xiàn)程池執(zhí)行完任務(wù)谍婉，關(guān)閉線(xiàn)程池
        threadPool.shutdown();
    }
}