Bootstrap

面试官:小伙子,听说你看过ThreadLocal源码?(万字图文深度解析ThreadLocal)

前言

本文所有内容及图片皆为原创,作者:一枝花算不算浪漫

原创不易,如若转载请标注出处,感谢!

(高清无损原图.pdf关注公众号后回复 获取,可以搜索个人公众号:壹枝花算不算浪漫)

前几天写了一篇相关的文章:[我画了35张图就是为了让你深入 AQS][1],反响不错,这次趁热打铁再写一篇的文章,同样是深入原理,图文并茂。

全文共10000+字,31张图,这篇文章同样耗费了不少的时间和精力才创作完成,原创不易,请大家点点关注+在看,感谢。

对于,大家的第一反应可能是很简单呀,线程的变量副本,每个线程隔离。那这里有几个问题大家可以思考一下:

  • ThreadLocal的key是弱引用,那么在 threadLocal.get()的时候,发生GC之后,key是否为null

  • ThreadLocal中*ThreadLocalMap*的数据结构

  • ThreadLocalMap的*Hash算法*?

  • ThreadLocalMap中*Hash冲突*如何解决?

  • ThreadLocalMap扩容机制?

  • ThreadLocalMap中过期key的清理机制?探测式清理和*启发式清理*流程?

  • ThreadLocalMap.set()方法实现原理?

  • ThreadLocalMap.get()方法实现原理?

  • 项目中ThreadLocal使用情况?遇到的坑?

  • ......

上述的一些问题你是否都已经掌握的很清楚了呢?本文将围绕这些问题使用图文方式来剖析的点点滴滴

全文目录

7.1 ThreadLocalMap.set()原理图解

7.2 ThreadLocalMap.set()源码详解

10.1 ThreadLocalMap.get()图解

10.2 ThreadLocalMap.get()源码详解

13.1 ThreadLocal使用场景

13.2 分布式TraceId解决方案

注明: 本文源码基于

ThreadLocal代码演示

我们先看下使用示例:

public class ThreadLocalTest {
    private List messages = Lists.newArrayList();

    public static final ThreadLocal holder = ThreadLocal.withInitial(ThreadLocalTest::new);

    public static void add(String message) {
        holder.get().messages.add(message);
    }

    public static List clear() {
        List messages = holder.get().messages;
        holder.remove();

        System.out.println("size: " + holder.get().messages.size());
        return messages;
    }

    public static void main(String[] args) {
        ThreadLocalTest.add("一枝花算不算浪漫");
        System.out.println(holder.get().messages);
        ThreadLocalTest.clear();
    }
}

打印结果:

[一枝花算不算浪漫]
size: 0

对象可以提供线程局部变量,每个线程拥有一份自己的副本变量,多个线程互不干扰。

ThreadLocal的数据结构

类有一个类型为的实例变量,也就是说每个线程有一个自己的。

有自己的独立实现,可以简单地将它的视作,为代码中放入的值(实际上并不是本身,而是它的一个弱引用)。

每个线程在往里放值的时候,都会往自己的里存,读也是以作为引用,在自己的里找对应的,从而实现了线程隔离

有点类似的结构,只是是由**数组+链表**实现的,而中并没有链表结构。

我们还要注意, 它的是 ,继承自, 也就是我们常说的弱引用类型。

GC 之后key是否为null?

回应开头的那个问题, 的是弱引用,那么在的时候,发生之后,是否是?

为了搞清楚这个问题,我们需要搞清楚的四种引用类型

  • 强引用:我们常常new出来的对象就是强引用类型,只要强引用存在,垃圾回收器将永远不会回收被引用的对象,哪怕内存不足的时候

  • 软引用:使用SoftReference修饰的对象被称为软引用,软引用指向的对象在内存要溢出的时候被回收

  • 弱引用:使用WeakReference修饰的对象被称为弱引用,只要发生垃圾回收,若这个对象只被弱引用指向,那么就会被回收

  • 虚引用:虚引用是最弱的引用,在 Java 中使用 PhantomReference 进行定义。虚引用中唯一的作用就是用队列接收对象即将死亡的通知

接着再来看下代码,我们使用反射的方式来看看后中的数据情况:(下面代码来源自:https://blog.csdn.net/thewindkee/article/details/103726942,本地运行演示GC回收场景)

public class ThreadLocalDemo {

    public static void main(String[] args) throws NoSuchFieldException, IllegalAccessException, InterruptedException {
        Thread t = new Thread(()->test("abc",false));
        t.start();
        t.join();
        System.out.println("--gc后--");
        Thread t2 = new Thread(() -> test("def", true));
        t2.start();
        t2.join();
    }

    private static void test(String s,boolean isGC)  {
        try {
            new ThreadLocal<>().set(s);
            if (isGC) {
                System.gc();
            }
            Thread t = Thread.currentThread();
            Class clz = t.getClass();
            Field field = clz.getDeclaredField("threadLocals");
            field.setAccessible(true);
            Object threadLocalMap = field.get(t);
            Class tlmClass = threadLocalMap.getClass();
            Field tableField = tlmClass.getDeclaredField("table");
            tableField.setAccessible(true);
            Object[] arr = (Object[]) tableField.get(threadLocalMap);
            for (Object o : arr) {
                if (o != null) {
                    Class entryClass = o.getClass();
                    Field valueField = entryClass.getDeclaredField("value");
                    Field referenceField = entryClass.getSuperclass().getSuperclass().getDeclaredField("referent");
                    valueField.setAccessible(true);
                    referenceField.setAccessible(true);
                    System.out.println(String.format("弱引用key:%s,值:%s", referenceField.get(o), valueField.get(o)));
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

结果如下:

弱引用key:java.lang.ThreadLocal@433619b6,值:abc
弱引用key:java.lang.ThreadLocal@418a15e3,值:java.lang.ref.SoftReference@bf97a12
--gc后--
弱引用key:null,值:def

如图所示,因为这里创建的并没有指向任何值,也就是没有任何引用:

new ThreadLocal<>().set(s);

所以这里在之后,就会被回收,我们看到上面中的, 如果改动一下代码:

这个问题刚开始看,如果没有过多思考,弱引用,还有垃圾回收,那么肯定会觉得是。

其实是不对的,因为题目说的是在做 操作,证明其实还是有强引用存在的,所以 并不为 ,如下图所示,的强引用仍然是存在的。

如果我们的强引用不存在的话,那么 就会被回收,也就是会出现我们 没被回收, 被回收,导致 永远存在,出现内存泄漏。

ThreadLocal.set()方法源码详解

中的方法原理如上图所示,很简单,主要是判断是否存在,然后使用中的方法进行数据处理。

代码如下:

public void set(T value) {
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}

void createMap(Thread t, T firstValue) {
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}

主要的核心逻辑还是在中的,一步步往下看,后面还有更详细的剖析。

ThreadLocalMap Hash算法

既然是结构,那么当然也要实现自己的算法来解决散列表数组冲突问题。

int i = key.threadLocalHashCode & (len-1);

中算法很简单,这里就是当前key在散列表中对应的数组下标位置。

这里最关键的就是值的计算,中有一个属性为

public class ThreadLocal {
    private final int threadLocalHashCode = nextHashCode();

    private static AtomicInteger nextHashCode = new AtomicInteger();

    private static final int HASH_INCREMENT = 0x61c88647;

    private static int nextHashCode() {
        return nextHashCode.getAndAdd(HASH_INCREMENT);
    }
    
    static class ThreadLocalMap {
        ThreadLocalMap(ThreadLocal firstKey, Object firstValue) {
            table = new Entry[INITIAL_CAPACITY];
            int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);

            table[i] = new Entry(firstKey, firstValue);
            size = 1;
            setThreshold(INITIAL_CAPACITY);
        }
    }
}

每当创建一个对象,这个 这个值就会增长 。

这个值很特殊,它是斐波那契数 也叫 黄金分割数。增量为 这个数字,带来的好处就是 分布非常均匀

我们自己可以尝试下:

可以看到产生的哈希码分布很均匀,这里不去细纠斐波那契具体算法,感兴趣的可以自行查阅相关资料。

ThreadLocalMap Hash冲突

注明: 下面所有示例图中,绿色块代表正常数据,**灰色块**代表的值为,*已被垃圾回收*。白色块表示为。

虽然中使用了黄金分隔数来作为计算因子,大大减少了冲突的概率,但是仍然会存在冲突。

中解决冲突的方法是在数组上构造一个链表结构,冲突的数据挂载到链表上,如果链表长度超过一定数量则会转化成红黑树

而中并没有链表结构,所以这里不能适用解决冲突的方式了。

如上图所示,如果我们插入一个的数据,通过计算后应该落入第4个槽位中,而槽位4已经有了数据。

此时就会线性向后查找,一直找到为的槽位才会停止查找,将当前元素放入此槽位中。当然迭代过程中还有其他的情况,比如遇到了不为且值相等的情况,还有中的值为的情况等等都会有不同的处理,后面会一一详细讲解。

这里还画了一个中的为的数据(**Entry=2的灰色块数据**),因为值是**弱引用**类型,所以会有这种数据存在。在过程中,如果遇到了过期的数据,实际上是会进行一轮探测式清理操作的,具体操作方式后面会讲到。

ThreadLocalMap.set()详解

ThreadLocalMap.set()原理图解

看完了 hash算法后,我们再来看是如何实现的。

往中数据(新增或者*更新*数据)分为好几种情况,针对不同的情况我们画图来说说明。

第一种情况: 通过计算后的槽位对应的数据为空:

这里直接将数据放到该槽位即可。

第二种情况: 槽位数据不为空,值与当前通过计算获取的值一致:

这里直接更新该槽位的数据。

第三种情况: 槽位数据不为空,往后遍历过程中,在找到为的槽位之前,没有遇到过期的:

遍历散列数组,线性往后查找,如果找到为的槽位,则将数据放入该槽位中,或者往后遍历过程中,遇到了key值相等的数据,直接更新即可。

第四种情况: 槽位数据不为空,往后遍历过程中,在找到为的槽位之前,遇到过期的,如下图,往后遍历过程中,一到了的槽位数据的:

散列数组下标为7位置对应的数据为,表明此数据值已经被垃圾回收掉了,此时就会执行方法,该方法含义是替换过期数据的逻辑,以*index=7*位起点开始遍历,进行探测式数据清理工作。

初始化探测式清理过期数据扫描的开始位置:

以当前开始 向前迭代查找,找其他过期的数据,然后更新过期数据起始扫描下标。循环迭代,直到碰到为结束。

如果找到了过期的数据,继续向前迭代,直到遇到的槽位才停止迭代,如下图所示,slotToExpunge被更新为0

以当前节点()向前迭代,检测是否有过期的数据,如果有则更新值。碰到则结束探测。以上图为例被更新为0。

上面向前迭代的操作是为了更新探测清理过期数据的起始下标的值,这个值在后面会讲解,它是用来判断当前过期槽位之前是否还有过期元素。

接着开始以位置(index=7)向后迭代,如果找到了相同key值的Entry数据:

从当前节点向后查找值相等的元素,找到后更新的值并交换元素的位置(位置为过期元素),更新数据,然后开始进行过期的清理工作,如下图所示:

向后遍历过程中,如果没有找到相同key值的Entry数据:

从当前节点向后查找值相等的元素,直到为则停止寻找。通过上图可知,此时中没有值相同的。

创建新的,替换位置:

替换完成后也是进行过期元素清理工作,清理工作主要是有两个方法:和,具体细节后面会讲到,请继续往后看。

ThreadLocalMap.set()源码详解

上面已经用图的方式解析了实现的原理,其实已经很清晰了,我们接着再看下源码:

:

private void set(ThreadLocal key, Object value) {
    Entry[] tab = table;
    int len = tab.length;
    int i = key.threadLocalHashCode & (len-1);

    for (Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        ThreadLocal k = e.get();

        if (k == key) {
            e.value = value;
            return;
        }

        if (k == null) {
            replaceStaleEntry(key, value, i);
            return;
        }
    }

    tab[i] = new Entry(key, value);
    int sz = ++size;
    if (!cleanSomeSlots(i, sz) && sz >= threshold)
        rehash();
}

这里会通过来计算在散列表中的对应位置,然后以当前对应的桶的位置向后查找,找到可以使用的桶。

Entry[] tab = table;
int len = tab.length;
int i = key.threadLocalHashCode & (len-1);

什么情况下桶才是可以使用的呢?

接着就是执行循环遍历,向后查找,我们先看下、方法实现:

private static int nextIndex(int i, int len) {
    return ((i + 1 < len) ? i + 1 : 0);
}

private static int prevIndex(int i, int len) {
    return ((i - 1 >= 0) ? i - 1 : len - 1);
}

接着看剩下循环中的逻辑:

2.1 如果,说明当前操作是一个替换操作,做替换逻辑,直接返回

2.2 如果,说明当前桶位置的是过期数据,执行方法(核心方法),然后返回

3.1 在为的桶中创建一个新的对象

3.2 执行操作

4.1 如果清理工作完成后,未清理到任何数据,且超过了阈值(数组长度的2/3),进行操作

4.2 中会先进行一轮探测式清理,清理过期,清理完成后如果size >= threshold - threshold / 4,就会执行真正的扩容逻辑(扩容逻辑往后看)

接着重点看下方法,方法提供替换过期数据的功能,我们可以对应上面第四种情况的原理图来再回顾下,具体代码如下:

:

private void replaceStaleEntry(ThreadLocal key, Object value,
                                       int staleSlot) {
    Entry[] tab = table;
    int len = tab.length;
    Entry e;

    int slotToExpunge = staleSlot;
    for (int i = prevIndex(staleSlot, len);
         (e = tab[i]) != null;
         i = prevIndex(i, len))

        if (e.get() == null)
            slotToExpunge = i;

    for (int i = nextIndex(staleSlot, len);
         (e = tab[i]) != null;
         i = nextIndex(i, len)) {

        ThreadLocal k = e.get();

        if (k == key) {
            e.value = value;

            tab[i] = tab[staleSlot];
            tab[staleSlot] = e;

            if (slotToExpunge == staleSlot)
                slotToExpunge = i;
            cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);
            return;
        }

        if (k == null && slotToExpunge == staleSlot)
            slotToExpunge = i;
    }

    tab[staleSlot].value = null;
    tab[staleSlot] = new Entry(key, value);

    if (slotToExpunge != staleSlot)
        cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);
}

表示开始探测式清理过期数据的开始下标,默认从当前的开始。以当前的开始,向前迭代查找,找到没有过期的数据,循环一直碰到为才会结束。如果向前找到了过期数据,更新探测清理过期数据的开始下标为i,即

for (int i = prevIndex(staleSlot, len);
     (e = tab[i]) != null;
     i = prevIndex(i, len)){

    if (e.get() == null){
        slotToExpunge = i;
    }
}

接着开始从向后查找,也是碰到为的桶结束。

如果迭代过程中,碰到k == key,这说明这里是替换逻辑,替换新数据并且交换当前位置。如果,这说明一开始向前查找过期数据时并未找到过期的数据,接着向后查找过程中也未发现过期数据,修改开始探测式清理过期数据的下标为当前循环的index,即。最后调用进行启发式过期数据清理。

if (k == key) {
    e.value = value;

    tab[i] = tab[staleSlot];
    tab[staleSlot] = e;
 
    if (slotToExpunge == staleSlot)
        slotToExpunge = i;

    cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);
    return;
}

和方法后面都会细讲,这两个是和清理相关的方法,一个是过期相关的启发式清理(),另一个是过期相关的探测式清理。

如果k != key则会接着往下走,说明当前遍历的是一个过期数据,说明,一开始的向前查找数据并未找到过期的。如果条件成立,则更新 为当前位置,这个前提是前驱节点扫描时未发现过期数据。

if (k == null && slotToExpunge == staleSlot)
    slotToExpunge = i;

往后迭代的过程中如果没有找到的数据,且碰到为的数据,则结束当前的迭代操作。此时说明这里是一个添加的逻辑,将新的数据添加到 对应的中。

tab[staleSlot].value = null;
tab[staleSlot] = new Entry(key, value);

最后判断除了以外,还发现了其他过期的数据,就要开启清理数据的逻辑:

if (slotToExpunge != staleSlot)
    cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);

ThreadLocalMap过期key的探测式清理流程

上面我们有提及的两种过期数据清理方式:探测式清理和*启发式清理*。

我们先讲下探测式清理,也就是方法,遍历散列数组,从开始位置向后探测清理过期数据,将过期数据的设置为,沿途中碰到未过期的数据则将此数据后重新在数组中定位,如果定位的位置已经有了数据,则会将未过期的数据放到最靠近此位置的的桶中,使后的数据距离正确的桶的位置更近一些。操作逻辑如下:

如上图, 经过hash计算后应该落到的桶中,由于桶已经有了数据,所以往后迭代最终数据放入到的桶中,放入后一段时间后中的数据变为了

如果再有其他数据到中,就会触发探测式清理操作。

如上图,执行探测式清理后,的数据被清理掉,继续往后迭代,到的元素时,经过后发现该元素正确的,而此位置已经已经有了数据,往后查找离最近的的节点(刚被探测式清理掉的数据:index=5),找到后移动的数据到中,此时桶的位置离正确的位置更近了。

经过一轮探测式清理后,过期的数据会被清理掉,没过期的数据经过重定位后所处的桶位置理论上更接近的位置。这种优化会提高整个散列表查询性能。

接着看下具体流程,我们还是以先原理图后源码讲解的方式来一步步梳理:

我们假设 来调用此方法,如上图所示,我们可以看到中的数据情况,接着执行清理操作:

第一步是清空当前位置的数据,位置的变成了。然后接着往后探测:

执行完第二步后,index=4的元素挪到index=3的槽位中。

继续往后迭代检查,碰到正常数据,计算该数据位置是否偏移,如果被偏移,则重新计算位置,目的是让正常数据尽可能存放在正确位置或离正确位置更近的位置

在往后迭代的过程中碰到空的槽位,终止探测,这样一轮探测式清理工作就完成了,接着我们继续看看具体实现源代码

private int expungeStaleEntry(int staleSlot) {
    Entry[] tab = table;
    int len = tab.length;

    tab[staleSlot].value = null;
    tab[staleSlot] = null;
    size--;

    Entry e;
    int i;
    for (i = nextIndex(staleSlot, len);
         (e = tab[i]) != null;
         i = nextIndex(i, len)) {
        ThreadLocal k = e.get();
        if (k == null) {
            e.value = null;
            tab[i] = null;
            size--;
        } else {
            int h = k.threadLocalHashCode & (len - 1);
            if (h != i) {
                tab[i] = null;

                while (tab[h] != null)
                    h = nextIndex(h, len);
                tab[h] = e;
            }
        }
    }
    return i;
}

这里我们还是以 来做示例说明,首先是将槽位的数据清空,然后设置

接着以位置往后迭代,如果遇到的过期数据,也是清空该槽位数据,然后

ThreadLocal k = e.get();

if (k == null) {
    e.value = null;
    tab[i] = null;
    size--;
} 

如果没有过期,重新计算当前的下标位置是不是当前槽位下标位置,如果不是,那么说明产生了冲突,此时以新计算出来正确的槽位位置往后迭代,找到最近一个可以存放的位置。

int h = k.threadLocalHashCode & (len - 1);
if (h != i) {
    tab[i] = null;

    while (tab[h] != null)
        h = nextIndex(h, len);

    tab[h] = e;
}

这里是处理正常的产生冲突的数据,经过迭代后,有过冲突数据的位置会更靠近正确位置,这样的话,查询的时候 效率才会更高。

ThreadLocalMap扩容机制

在方法的最后,如果执行完启发式清理工作后,未清理到任何数据,且当前散列数组中的数量已经达到了列表的扩容阈值,就开始执行逻辑:

if (!cleanSomeSlots(i, sz) && sz >= threshold)
    rehash();

接着看下具体实现:

private void rehash() {
    expungeStaleEntries();

    if (size >= threshold - threshold / 4)
        resize();
}

private void expungeStaleEntries() {
    Entry[] tab = table;
    int len = tab.length;
    for (int j = 0; j < len; j++) {
        Entry e = tab[j];
        if (e != null && e.get() == null)
            expungeStaleEntry(j);
    }
}

这里首先是会进行探测式清理工作,从的起始位置往后清理,上面有分析清理的详细流程。清理完成之后,中可能有一些为的数据被清理掉,所以此时通过判断 也就是 来决定是否扩容。

我们还记得上面进行的阈值是,所以当面试官套路我们扩容机制的时候 我们一定要说清楚这两个步骤:

接着看看具体的方法,为了方便演示,我们以来举例:

扩容后的的大小为,然后遍历老的散列表,重新计算位置,然后放到新的数组中,如果出现冲突则往后寻找最近的为的槽位,遍历完成之后,中所有的数据都已经放入到新的中了。重新计算下次扩容的阈值,具体代码如下:

private void resize() {
    Entry[] oldTab = table;
    int oldLen = oldTab.length;
    int newLen = oldLen * 2;
    Entry[] newTab = new Entry[newLen];
    int count = 0;

    for (int j = 0; j < oldLen; ++j) {
        Entry e = oldTab[j];
        if (e != null) {
            ThreadLocal k = e.get();
            if (k == null) {
                e.value = null;
            } else {
                int h = k.threadLocalHashCode & (newLen - 1);
                while (newTab[h] != null)
                    h = nextIndex(h, newLen);
                newTab[h] = e;
                count++;
            }
        }
    }

    setThreshold(newLen);
    size = count;
    table = newTab;
}

ThreadLocalMap.get()详解

上面已经看完了方法的源码,其中包括数据、清理数据、优化数据桶的位置等操作,接着看看操作的原理。

ThreadLocalMap.get()图解

第一种情况: 通过查找值计算出散列表中位置,然后该位置中的和查找的一致,则直接返回:

第二种情况: 位置中的和要查找的不一致:

我们以为例,通过计算后,正确的位置应该是4,而的槽位已经有了数据,且值不等于,所以需要继续往后迭代查找。

迭代到的数据时,此时,触发一次探测式数据回收操作,执行方法,执行完后,的数据都会被回收,而的数据都会前移,此时继续往后迭代,到的时候即找到了值相等的数据,如下图所示:

ThreadLocalMap.get()源码详解

:

private Entry getEntry(ThreadLocal key) {
    int i = key.threadLocalHashCode & (table.length - 1);
    Entry e = table[i];
    if (e != null && e.get() == key)
        return e;
    else
        return getEntryAfterMiss(key, i, e);
}

private Entry getEntryAfterMiss(ThreadLocal key, int i, Entry e) {
    Entry[] tab = table;
    int len = tab.length;

    while (e != null) {
        ThreadLocal k = e.get();
        if (k == key)
            return e;
        if (k == null)
            expungeStaleEntry(i);
        else
            i = nextIndex(i, len);
        e = tab[i];
    }
    return null;
}

ThreadLocalMap过期key的启发式清理流程

上面多次提及到过期可以的两种清理方式:探测式清理(expungeStaleEntry())、*启发式清理(cleanSomeSlots())*

探测式清理是以当前 往后清理,遇到值为则结束清理,属于线性探测清理

而启发式清理被作者定义为:Heuristically scan some cells looking for stale entries.

具体代码如下:

private boolean cleanSomeSlots(int i, int n) {
    boolean removed = false;
    Entry[] tab = table;
    int len = tab.length;
    do {
        i = nextIndex(i, len);
        Entry e = tab[i];
        if (e != null && e.get() == null) {
            n = len;
            removed = true;
            i = expungeStaleEntry(i);
        }
    } while ( (n >>>= 1) != 0);
    return removed;
}

InheritableThreadLocal

我们使用的时候,在异步场景下是无法给子线程共享父线程中创建的线程副本数据的。

为了解决这个问题,JDK中还有一个类,我们来看一个例子:

public class InheritableThreadLocalDemo {
    public static void main(String[] args) {
        ThreadLocal threadLocal = new ThreadLocal<>();
        ThreadLocal inheritableThreadLocal = new InheritableThreadLocal<>();
        threadLocal.set("父类数据:threadLocal");
        inheritableThreadLocal.set("父类数据:inheritableThreadLocal");

        new Thread(new Runnable() {
            @Override
            public void run() {
                System.out.println("子线程获取父类threadLocal数据:" + threadLocal.get());
                System.out.println("子线程获取父类inheritableThreadLocal数据:" + inheritableThreadLocal.get());
            }
        }).start();
    }
}

打印结果:

子线程获取父类threadLocal数据:null
子线程获取父类inheritableThreadLocal数据:父类数据:inheritableThreadLocal

实现原理是子线程是通过在父线程中通过调用方法来创建子线程,方法在的构造方法中被调用。在方法中拷贝父线程数据到子线程中:

private void init(ThreadGroup g, Runnable target, String name,
                      long stackSize, AccessControlContext acc,
                      boolean inheritThreadLocals) {
    if (name == null) {
        throw new NullPointerException("name cannot be null");
    }

    if (inheritThreadLocals && parent.inheritableThreadLocals != null)
        this.inheritableThreadLocals =
            ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);
    this.stackSize = stackSize;
    tid = nextThreadID();
}

但仍然有缺陷,一般我们做异步化处理都是使用的线程池,而是在中的方法给赋值的,而线程池是线程复用的逻辑,所以这里会存在问题。

当然,有问题出现就会有解决问题的方案,阿里巴巴开源了一个组件就可以解决这个问题,这里就不再延伸,感兴趣的可自行查阅资料。

ThreadLocal项目中使用实战

ThreadLocal使用场景

我们现在项目中日志记录用的是,最后在中进行展示和检索。

现在都是分布式系统统一对外提供服务,项目间调用的关系可以通过traceId来关联,但是不同项目之间如何传递呢?

这里我们使用来实现此功能,内部就是通过来实现的,具体实现如下:

当前端发送请求到服务A时,*服务A*会生成一个类似的字符串,将此字符串放入当前线程的中,在调用**服务B**的时候,将写入到请求的中,**服务B**在接收请求时会先判断请求的中是否有,如果存在则写入自己线程的中。

图中的即为我们各个系统链路关联的,系统间互相调用,通过这个即可找到对应链路,这里还有会有一些其他场景:

针对于这些场景,我们都可以有相应的解决方案,如下所示

Feign远程调用解决方案

服务发送请求:

@Component
@Slf4j
public class FeignInvokeInterceptor implements RequestInterceptor {

    @Override
    public void apply(RequestTemplate template) {
        String requestId = MDC.get("requestId");
        if (StringUtils.isNotBlank(requestId)) {
            template.header("requestId", requestId);
        }
    }
}

服务接收请求:

@Slf4j
@Component
public class LogInterceptor extends HandlerInterceptorAdapter {

    @Override
    public void afterCompletion(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2, Exception arg3) {
        MDC.remove("requestId");
    }

    @Override
    public void postHandle(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2, ModelAndView arg3) {
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        String requestId = request.getHeader(BaseConstant.REQUEST_ID_KEY);
        if (StringUtils.isBlank(requestId)) {
            requestId = UUID.randomUUID().toString().replace("-", "");
        }
        MDC.put("requestId", requestId);
        return true;
    }
}

线程池异步调用,requestId传递

因为是基于去实现的,异步过程中,子线程并没有办法获取到父线程存储的数据,所以这里可以自定义线程池执行器,修改其中的方法:

public class MyThreadPoolTaskExecutor extends ThreadPoolTaskExecutor {
    
    @Override
    public void execute(Runnable runnable) {
        Map context = MDC.getCopyOfContextMap();
        super.execute(() -> run(runnable, context));
    }

    @Override
    private void run(Runnable runnable, Map context) {
        if (context != null) {
            MDC.setContextMap(context);
        }
        try {
            runnable.run();
        } finally {
            MDC.remove();
        }
    }
}

使用MQ发送消息给第三方系统

在MQ发送的消息体中自定义属性,接收方消费消息后,自己解析使用即可。

[1]:https://juejin.im/post/5eacc1c75188256d976df748