全链路追踪traceId,ThreadLocal与ExecutorService

关于全链路追踪traceId遇到线程池的问题，做过架构的估计都遇到过，现在以写个demo，总体思想就是获取父线程traceId,给子线程，子线程用完移除掉。

mac上的chrome时不时崩溃，写了一大半的博客没了，直接贴源码和注释吧

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ThreadPoolTracing {

    private static final Logger logger = LoggerFactory.getLogger(ThreadPoolTracing.class);
    public static ThreadLocal<String>  threadLocalTraceId = new ThreadLocal<>();

    static class Task implements Runnable {

        @Override
        public void run() {
            String traceId=threadLocalTraceId.get();
            logger.info("traceId={}",traceId);
        }
    }
}

public class MyTest2 {

    private static final Logger loger = LoggerFactory.getLogger(MyTest2.class);
    //线程池大小设置为一，保证是同一个线程run之前获取traceId，run后删除，便于测试
    private static ExecutorService executorService = Executors.newFixedThreadPool(1);

    @Test
    public void test1() {

        String traceId = UUID.randomUUID().toString().replace("-", "");
        ThreadPoolTracing.threadLocalTraceId.set(traceId);
        loger.info("父线程={}；traceId={}",Thread.currentThread().getName(),traceId);

        Runnable runnable=new Runnable() {
            @Override
            public void run() {
                //ThreadLocal 拿不到值；如果是InheritableThreadLocal，可以拿到值
                String id0 = ThreadPoolTracing.threadLocalTraceId.get();
                loger.info("子线程={},traceId={}",Thread.currentThread().getName(),id0);
            }
        };
        executorService.execute(runnable);//结果为空
        executorService.execute(new ThreadPoolTracing.Task());//结果为空

        Runnable wrap=  wrap( runnable);
        executorService.execute(wrap);//可以获取traceId
        try {
            Thread.sleep(100);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        traceId = UUID.randomUUID().toString().replace("-", "");
        //traceId 重新复制
        ThreadPoolTracing.threadLocalTraceId.set(traceId);
        loger.info("父线程={}；traceId={}",Thread.currentThread().getName(),traceId);
        //线程池中的traceId跟着变更
        wrap=  wrap( runnable);
        executorService.execute(wrap);

        try {
            Thread.sleep(1000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }


   
    public Runnable wrap(Runnable task) {
        //获取父线程中的Trace
        String id0 = ThreadPoolTracing.threadLocalTraceId.get();
        class CurrentTraceContextRunnable implements Runnable {
            @Override
            public void run() {
                //traceId 给子线程
                ThreadPoolTracing.threadLocalTraceId.set(id0);
                task.run();
                //子线程用完删除
                ThreadPoolTracing.threadLocalTraceId.remove();
            }
        }
        return new CurrentTraceContextRunnable();

    }


}