【巨人的肩膀】通透~~ThreadLocal夺命11连问

前言

前一段时间,有同事使用 ThreadLocal 踩坑了,正好引起了我的兴趣

所以近期,我抽空把 ThreadLocal 的源码再研究了一下,越看越有意思,发现里面的东西还真不少

我把精华浓缩了一下,汇集成了下面 11 个问题,看看你能顶住第几个?
在这里插入图片描述

1. 为什么要用ThreadLocal?

并发编程是一项非常重要的技术,它让我们的程序变得更加高效

但在并发的场景中,如果有多个线程同时修改公共变量,可能会出现线程安全问题,即该变量最终结果可能出现异常

为了解决线程安全问题,JDK出现了很多技术手段,比如:使用 synchronized 或 Lock,给访问公共资源的代码上锁,保证了代码的原子性

但在高并发的场景中,如果多个线程同时竞争一把锁,这时会存在大量的锁等待,可能会浪费很多时间,让系统的响应时间一下子变慢

因此,JDK 还提供了另外一种用空间换时间的新思路:ThreadLocal

它的核心思想是:共享变量在每个线程都有一个副本,每个线程操作的都是自己的副本,对另外的线程没有影响

@Service
public class ThreadLocalService {
    private static final ThreadLocal<Integer> threadLocal = new ThreadLocal<>();

    public void add() {
        threadLocal.set(1);
        doSamething();
        Integer integer = threadLocal.get();
    }
}

2. ThreadLocal的原理是什么?

为了搞清楚 ThreadLocal 的底层实现原理,我们不得不扒一下源码

ThreadLocal 的内部有一个静态的内部类叫:ThreadLocalMap

public class ThreadLocal<T> {
     ...
     public T get() {
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取当前线程的成员变量ThreadLocalMap对象
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            //根据threadLocal对象从map中获取Entry对象
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                //获取保存的数据
                T result = (T)e.value;
                return result;
            }
        }
        //初始化数据
        return setInitialValue();
    }
    
    private T setInitialValue() {
        //获取要初始化的数据
        T value = initialValue();
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取当前线程的成员变量ThreadLocalMap对象
        ThreadLocalMap map = getMap(t);
        //如果map不为空
        if (map != null)
            //将初始值设置到map中,key是this,即threadLocal对象,value是初始值
            map.set(this, value);
        else
           //如果map为空,则需要创建新的map对象
            createMap(t, value);
        return value;
    }
    
    public void set(T value) {
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取当前线程的成员变量ThreadLocalMap对象
        ThreadLocalMap map = getMap(t);
        //如果map不为空
        if (map != null)
            //将值设置到map中,key是this,即threadLocal对象,value是传入的value值
            map.set(this, value);
        else
           //如果map为空,则需要创建新的map对象
            createMap(t, value);
    }
    
     static class ThreadLocalMap {
        ...
     }
     ...
}

ThreadLocal 的 get 方法、set 方法和 setInitialValue 方法,其实最终操作的都是 ThreadLocalMap 类中的数据

其中 ThreadLocalMap 类的内部如下

static class ThreadLocalMap {
    static class Entry extends WeakReference<ThreadLocal<?>> {
        Object value;

        Entry(ThreadLocal<?> k, Object v) {
            super(k);
            value = v;
        }
   }
   ...
   private Entry[] table;
   ...
}

ThreadLocalMap 里面包含一个静态的内部类 Entry,该类继承于 WeakReference 类,说明 Entry 是一个弱引用

ThreadLocalMap 内部还包含了一个 Entry 数组,其中:Entry = ThreadLocal + value

而 ThreadLocalMap 被定义成了 Thread 类的成员变量

public class Thread implements Runnable {
    ...
    ThreadLocal.ThreadLocalMap threadLocals = null;
}

下面用一张图从宏观上,认识一下 ThreadLocal 的整体结构:
在这里插入图片描述
从上图中看出,在每个 Thread 类中,都有一个 ThreadLocalMap 的成员变量,该变量包含了一个 Entry 数组,该数组真正保存了 ThreadLocal 类 set 的数据

Entry 是由 threadLocal 和 value 组成,其中 threadLocal 对象是弱引用,在 GC 的时候,会被自动回收。而 value 就是 ThreadLocal 类 set 的数据

下面用一张图总结一下引用关系:
在这里插入图片描述
上图中除了 Entry 的 key 对 ThreadLocal 对象是弱引用,其他的引用都是强引用

需要特别说明的是,上图中 ThreadLocal 对象我画到了堆上,其实在实际的业务场景中不一定在堆上。因为如果 ThreadLocal 被定义成了 static 的,ThreadLocal 的对象是类共用的,可能出现在方法区

3. 为什么用ThreadLocal做key?

不知道你有没有思考过这样一个问题:ThreadLocalMap 为什么要用 ThreadLocal 做 key,而不是用 Thread 做 key?

如果在你的应用中,一个线程中只使用了一个 ThreadLocal 对象,那么使用 Thread 做 key 也未尝不可

@Service
public class ThreadLocalService {
    private static final ThreadLocal<Integer> threadLocal = new ThreadLocal<>();
}    

但实际情况中,你的应用,一个线程中很有可能不只使用了一个 ThreadLocal 对象。这时使用 Thread 做 key 不就出有问题?

@Service
public class ThreadLocalService {
    private static final ThreadLocal<Integer> threadLocal1 = new ThreadLocal<>();
    private static final ThreadLocal<Integer> threadLocal2 = new ThreadLocal<>();
    private static final ThreadLocal<Integer> threadLocal3 = new ThreadLocal<>();
}   

假如使用 Thread 做 key 时,你的代码中定义了 3 个 ThreadLocal 对象,那么,通过 Thread 对象,它怎么知道要获取哪个 ThreadLocal 对象呢?

如下图所示:
在这里插入图片描述
因此,不能使用 Thread 做 key,而应该改成用 ThreadLocal 对象做 key,这样才能通过具体 ThreadLocal 对象的 get 方法,轻松获取到你想要的 ThreadLocal 对象

如下图所示:
在这里插入图片描述

4. Entry的key为什么设计成弱引用?

前面说过,Entry 的 key,传入的是 ThreadLocal 对象,使用了 WeakReference 对象,即被设计成了弱引用

那么,为什么要这样设计呢?

假如 key 对 ThreadLocal 对象的弱引用,改为强引用
在这里插入图片描述
我们都知道 ThreadLocal 变量对 ThreadLocal 对象是有强引用存在的

即使 ThreadLocal 变量生命周期完了,设置成 null 了,但由于 key 对 ThreadLocal 还是强引用

此时,如果执行该代码的线程使用了线程池,一直长期存在,不会被销毁

就会存在这样的强引用链:Thread变量 -> Thread对象 -> ThreadLocalMap -> Entry -> key -> ThreadLocal对象

那么,ThreadLocal 对象和 ThreadLocalMap 都将不会被 GC 回收,于是产生了内存泄露问题

为了解决这个问题,JDK 的开发者们把 Entry 的 key 设计成了弱引用

弱引用的对象,在 GC 做垃圾清理的时候,就会被自动回收了

如果 key 是弱引用,当 ThreadLocal 变量指向 null 之后,在 GC 做垃圾清理的时候,key 会被自动回收,其值也被设置成 null

如下图所示:
在这里插入图片描述
接下来,最关键的地方来了

由于当前的 ThreadLocal 变量已经被指向 null 了,但如果直接调用它的 get、set 或 remove 方法,很显然会出现空指针异常。因为它的生命已经结束了,再调用它的方法也没啥意义

此时,如果系统中还定义了另外一个 ThreadLocal 变量 b,调用了它的 get、set 或 remove,三个方法中的任何一个方法,都会自动触发清理机制,将 key 为 null 的 value 值清空

如果 key 和 value 都是 null,那么 Entry 对象会被 GC 回收。如果所有的 Entry 对象都被回收了,ThreadLocalMap 也会被回收了

这样就能最大程度的解决内存泄露问题

需要特别注意的地方是:

  1. key 为 null 的条件是,ThreadLocal 变量指向 null,并且 key 是弱引用。如果 ThreadLocal 变量没有断开对 ThreadLocal 的强引用,即 ThreadLocal 变量没有指向 null,GC 就贸然的把弱引用的 key 回收了,不就会影响正常用户的使用?
  2. 如果当前 ThreadLocal 变量指向 null 了,并且 key 也为 null 了,但如果没有其他 ThreadLocal 变量触发 get、set 或 remove 方法,也会造成内存泄露

下面看看弱引用的例子:

public static void main(String[] args) {
    WeakReference<Object> weakReference0 = new WeakReference<>(new Object());
    System.out.println(weakReference0.get());
    System.gc();
    System.out.println(weakReference0.get());
}

打印结果:

java.lang.Object@1ef7fe8e
null

传入 WeakReference 构造方法的是直接 new 处理的对象,没有其他引用,在调用 gc 方法后,弱引用对象会被自动回收

但如果出现下面这种情况:

public static void main(String[] args) {
    Object object = new Object();
    WeakReference<Object> weakReference1 = new WeakReference<>(object);
    System.out.println(weakReference1.get());
    System.gc();
    System.out.println(weakReference1.get());
}

执行结果:

java.lang.Object@1ef7fe8e
java.lang.Object@1ef7fe8e

先定义了一个强引用 object 对象,在 WeakReference 构造方法中将 object 对象的引用作为参数传入。这时,调用 gc 后,弱引用对象不会被自动回收

我们的 Entry 对象中的 key 不就是第二种情况吗?在 Entry 构造方法中传入的是 ThreadLocal 对象的引用

如果将 object 强引用设置为 null:

public static void main(String[] args) {
    Object object = new Object();
    WeakReference<Object> weakReference1 = new WeakReference<>(object);
    System.out.println(weakReference1.get());
    System.gc();
    System.out.println(weakReference1.get());

    object=null;
    System.gc();
    System.out.println(weakReference1.get());
}

执行结果:

java.lang.Object@6f496d9f
java.lang.Object@6f496d9f
null

第二次 gc 之后,弱引用能够被正常回收

由此可见,如果强引用和弱引用同时关联一个对象,那么这个对象是不会被 GC 回收。也就是说这种情况下 Entry 的 key,一直都不会为 null,除非强引用主动断开关联

此外,你可能还会问这样一个问题:Entry 的 value 为什么不设计成弱引用?

答:Entry 的 value 假如只是被 Entry 引用,有可能没被业务系统中的其他地方引用。如果将 value 改成了弱引用,被 GC 贸然回收了(数据突然没了),可能会导致业务系统出现异常

而相比之下,Entry 的 key,管理的地方就非常明确了

这就是 Entry 的 key 被设计成弱引用,而 value 没被设计成弱引用的原因

5. ThreadLocal真的会导致内存泄露?

通过上面的 Entry 对象中的 key 设置成弱引用,并且使用 get、set 或 remove 方法清理 key 为 null 的 value 值,就能彻底解决内存泄露问题?

答案是否定的

如下图所示:
在这里插入图片描述
假如 ThreadLocalMap 中存在很多 key 为 null 的 Entry,但后面的程序,一直都没有调用过有效的 ThreadLocal 的 get、set 或 remove 方法

那么,Entry 的 value 值一直都没被清空

所以会存在这样一条强引用链:Thread变量 -> Thread对象 -> ThreadLocalMap -> Entry -> value -> Object

其结果就是:Entry 和 ThreadLocalMap 将会长期存在下去,会导致内存泄露

6. 如何解决内存泄露问题?

前面说过的 ThreadLocal 还是会导致内存泄露的问题,我们有没有解决办法呢?

答:有办法,调用 ThreadLocal 对象的 remove 方法

不是在一开始就调用 remove 方法,而是在使用完 ThreadLocal 对象之后。列如:

先创建一个 CurrentUser 类,其中包含了 ThreadLocal 的逻辑

public class CurrentUser {
    private static final ThreadLocal<UserInfo> THREA_LOCAL = new ThreadLocal();
    
    public static void set(UserInfo userInfo) {
        THREA_LOCAL.set(userInfo);
    }
    
    public static UserInfo get() {
       THREA_LOCAL.get();
    }
    
    public static void remove() {
       THREA_LOCAL.remove();
    }
}

然后在业务代码中调用相关方法:

public void doSamething(UserDto userDto) {
   UserInfo userInfo = convert(userDto);
   
   try{
     CurrentUser.set(userInfo);
     ...
     
     // 业务代码
     UserInfo userInfo = CurrentUser.get();
     ...
   } finally {
      CurrentUser.remove();
   }
}

需要我们特别注意的地方是:一定要在 finally 代码块中,调用 remove 方法清理没用的数据。如果业务代码出现异常,也能及时清理没用的数据

remove 方法中会把 Entry 中的 key 和 value 都设置成 null,这样就能被 GC 及时回收,无需触发额外的清理机制,所以它能解决内存泄露问题

7. ThreadLocal是如何定位数据的?

前面说过 ThreadLocalMap 对象底层是用 Entry 数组保存数据的

那么问题来了,ThreadLocal 是如何定位 Entry 数组数据的?

在 ThreadLocal 的 get、set、remove 方法中都有这样一行代码:

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

通过 key 的 hashCode 值,与数组的长度减 1。其中 key 就是 ThreadLocal 对象,与数组的长度减 1,相当于除以数组的长度减 1,然后取模

这是一种 hash 算法

接下来给大家举个例子:假设len=16,key.threadLocalHashCode=31,

于是: int i = 31 & 15 = 15

相当于:int i = 31 % 16 = 15

计算的结果是一样的,但是使用与运算效率跟高一些 (这里有个前提,length 必须是 2 的倍数)

为什么与运算效率更高?

答:因为 ThreadLocal 的初始大小是 16,每次都是按 2 倍扩容,数组的大小其实一直都是 2 的 n 次方。这种数据有个规律就是高位是 0,低位都是 1。在做与运算时,可以不用考虑高位,因为与运算的结果必定是 0。只需考虑低位的与运算,所以效率更高

如果使用 hash 算法定位具体位置的话,就可能会出现 hash 冲突的情况,即两个不同的 hashCode 取模后的值相同

ThreadLocal 是如何解决 hash 冲突的呢?

我们看看 getEntry 是怎么做的

private Entry getEntry(ThreadLocal<?> key) {
    // 通过hash算法获取下标值
    int i = key.threadLocalHashCode & (table.length - 1);
    Entry e = table[i];
    // 如果下标位置上的key正好是我们所需要寻找的key
    if (e != null && e.get() == key)
        //说明找到数据了,直接返回
        return e;
    else
        //说明出现hash冲突了,继续往后找
        return getEntryAfterMiss(key, i, e);
}

再看看 getEntryAfterMiss 方法:

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

    // 判断Entry对象如果不为空,则一直循环
    while (e != null) {
        ThreadLocal<?> k = e.get();
        // 如果当前Entry的key正好是我们所需要寻找的key
        if (k == key)
            // 说明这次真的找到数据了
            return e;
        if (k == null)
            // 如果key为空,则清理脏数据
            expungeStaleEntry(i);
        else
            // 如果还是没找到数据,则继续往后找
            i = nextIndex(i, len);
        e = tab[i];
    }
    return null;
}

关键看看 nextIndex 方法:

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

当通过 hash 算法计算出的下标小于数组大小,则将下标值加 1。否则,即下标大于等于数组大小,下标变成 0 了。下标变成 0 之后,则循环一次,下标又变成 1

寻找的大致过程如下图所示:
在这里插入图片描述
如果找到最后一个,还是没有找到,则再从头开始找
在这里插入图片描述
不知道你有没有发现,它构成了一个:环形。

ThreadLocal 从数组中找数据的过程大致是这样的:

  1. 通过 key 的 hashCode 取余计算出一个下标
  2. 通过下标,在数组中定位具体 Entry,如果 key 正好是我们所需要的 key,说明找到了,则直接返回数据
  3. 如果第 2 步没有找到我们想要的数据,则从数组的下标位置,继续往后面找
  4. 如果第 3 步中找 key 的正好是我们所需要的 key,说明找到了,则直接返回数据
  5. 如果还是没有找到数据,再继续往后面找。如果找到最后一个位置,还是没有找到数据,则再从头,即下标为 0 的位置,继续从前往后找数据
  6. 直到找到第一个 Entry 为空为止

8. ThreadLocal是如何扩容的?

从上面得知,ThreadLocal 的初始大小是 16。那么问题来了,ThreadLocal 是如何扩容的?

在 set 方法中会调用 rehash 方法:

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();
}

注意一下,其中有个判断条件是:sz (之前的size+1)如果大于或等于 threshold 的话,则调用 rehash 方法

threshold 默认是 0,在创建 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);
}

调用 setThreshold 方法给 threshold 设置一个值,而这个值 INITIAL_CAPACITY 是默认的大小 16

private void setThreshold(int len) {
    threshold = len * 2 / 3;
}

也就是第一次设置的 threshold = 16 * 2 / 3, 取整后的值是:10

换句话说当 sz 大于等于 10 时,就可以考虑扩容了

rehash 代码如下

private void rehash() {
    //先尝试回收一次key为null的值,腾出一些空间
    expungeStaleEntries();

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

在真正扩容之前,先尝试回收一次 key 为 null 的值,腾出一些空间

如果回收之后的 size 大于等于 threshold 的 3/4 时,才需要真正的扩容

计算公式如下

16 * 2 * 4 / 3 * 4 - 16 * 2 / 3 * 4 = 8

也就是说添加数据后,新的 size 大于等于老 size 的 1/2 时,才需要扩容

private void resize() {
    Entry[] oldTab = table;
    int oldLen = oldTab.length;
    //按2倍的大小扩容
    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; // Help the GC
            } 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;
}

resize 中每次都是按 2 倍的大小扩容

扩容的过程如下图所示
在这里插入图片描述
扩容的关键步骤如下:

  1. 老 size + 1 = 新size
  2. 如果新 size 大于等于老 size 的 2/3 时,需要考虑扩容。
  3. 扩容前先尝试回收一次key为null的值,腾出一些空间。
  4. 如果回收之后发现size还是大于等于老size的1/2时,才需要真正的扩容。
  5. 每次都是按2倍的大小扩容

9. 父子线程如何共享数据?

前面介绍的 ThreadLocal 都是在一个线程中保存和获取数据的

但在实际工作中,有可能是在父子线程中共享数据的。即在父线程中往 ThreadLocal 设置了值,在子线程中能够获取到

例如:

public class ThreadLocalTest {

    public static void main(String[] args) {
        ThreadLocal<Integer> threadLocal = new ThreadLocal<>();
        threadLocal.set(6);
        System.out.println("父线程获取数据:" + threadLocal.get());

        new Thread(() -> {
            System.out.println("子线程获取数据:" + threadLocal.get());
        }).start();
    }
}

执行结果:

父线程获取数据:6
子线程获取数据:null

你会发现,在这种情况下使用 ThreadLocal 是行不通的。main 方法是在主线程中执行的,相当于父线程。在 main 方法中开启了另外一个线程,相当于子线程

显然通过 ThreadLocal,无法在父子线程中共享数据

那么,该怎么办呢?

答:使用 InheritableThreadLocal,它是 JDK 自带的类,继承了 ThreadLocal 类

修改代码之后:

public class ThreadLocalTest {

    public static void main(String[] args) {
        InheritableThreadLocal<Integer> threadLocal = new InheritableThreadLocal<>();
        threadLocal.set(6);
        System.out.println("父线程获取数据:" + threadLocal.get());

        new Thread(() -> {
            System.out.println("子线程获取数据:" + threadLocal.get());
        }).start();
    }
}

执行结果:

父线程获取数据:6
子线程获取数据:6

果然,在换成 InheritableThreadLocal 之后,在子线程中能够正常获取父线程中设置的值

其实,在 Thread 类中除了成员变量 threadLocals 之外,还有另一个成员变量:inheritableThreadLocals

Thread 类的部分代码如下:

ThreadLocal.ThreadLocalMap threadLocals = null;
ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;

最关键的一点是,在它的init方法中会将父线程中往 ThreadLocal 设置的值,拷贝一份到子线程中

感兴趣的小伙伴,可以找我私聊。或者看看我后面的文章,后面还会有专栏

10. 线程池中如何共享数据?

在真实的业务场景中,一般很少用单独的线程,绝大多数,都是用的线程池

那么,在线程池中如何共享 ThreadLocal 对象生成的数据呢?

因为涉及到不同的线程,如果直接使用 ThreadLocal,显然是不合适的

我们应该使用 InheritableThreadLocal,具体代码如下:

private static void fun1() {
    InheritableThreadLocal<Integer> threadLocal = new InheritableThreadLocal<>();
    threadLocal.set(6);
    System.out.println("父线程获取数据:" + threadLocal.get());

    ExecutorService executorService = Executors.newSingleThreadExecutor();

    threadLocal.set(6);
    executorService.submit(() -> {
        System.out.println("第一次从线程池中获取数据:" + threadLocal.get());
    });

    threadLocal.set(7);
    executorService.submit(() -> {
        System.out.println("第二次从线程池中获取数据:" + threadLocal.get());
    });
}

执行结果:

父线程获取数据:6
第一次从线程池中获取数据:6
第二次从线程池中获取数据:6

由于这个例子中使用了单例线程池,固定线程数是1

第一次 submit 任务的时候,该线程池会自动创建一个线程。因为使用了 InheritableThreadLocal,所以创建线程时,会调用它的 init 方法,将父线程中的 inheritableThreadLocals 数据复制到子线程中。所以我们看到,在主线程中将数据设置成 6,第一次从线程池中获取了正确的数据 6

之后,在主线程中又将数据改成 7,但在第二次从线程池中获取数据却依然是 6

因为第二次 submit 任务的时候,线程池中已经有一个线程了,就直接拿过来复用,不会再重新创建线程了。所以不会再调用线程的 init 方法,所以第二次其实没有获取到最新的数据 7,还是获取的老数据 6

那么,这该怎么办呢?

答:使用 TransmittableThreadLocal,它并非 JDK 自带的类,而是阿里巴巴开源 jar 包中的类

可以通过如下 pom 文件引入该 jar 包:

<dependency>
   <groupId>com.alibaba</groupId>
   <artifactId>transmittable-thread-local</artifactId>
   <version>2.11.0</version>
   <scope>compile</scope>
</dependency>

代码调整如下:

private static void fun2() throws Exception {
    TransmittableThreadLocal<Integer> threadLocal = new TransmittableThreadLocal<>();
    threadLocal.set(6);
    System.out.println("父线程获取数据:" + threadLocal.get());

    ExecutorService ttlExecutorService = TtlExecutors.getTtlExecutorService(Executors.newFixedThreadPool(1));

    threadLocal.set(6);
    ttlExecutorService.submit(() -> {
        System.out.println("第一次从线程池中获取数据:" + threadLocal.get());
    });

    threadLocal.set(7);
    ttlExecutorService.submit(() -> {
        System.out.println("第二次从线程池中获取数据:" + threadLocal.get());
    });

}

执行结果:

父线程获取数据:6
第一次从线程池中获取数据:6
第二次从线程池中获取数据:7

我们看到,使用了 TransmittableThreadLocal 之后,第二次从线程中也能正确获取最新的数据 7 了

nice

如果你仔细观察这个例子,你可能会发现,代码中除了使用 TransmittableThreadLocal 类之外,还使用了 TtlExecutors.getTtlExecutorService 方法,去创建 ExecutorService 对象

这是非常重要的地方,如果没有这一步,TransmittableThreadLocal 在线程池中共享数据将不会起作用

创建 ExecutorService 对象,底层的 submit 方法会 TtlRunnable 或 TtlCallable 对象

以 TtlRunnable 类为例,它实现了 Runnable 接口,同时还实现了它的 run 方法:

public void run() {
    Map<TransmittableThreadLocal<?>, Object> copied = (Map)this.copiedRef.get();
    if (copied != null && (!this.releaseTtlValueReferenceAfterRun || this.copiedRef.compareAndSet(copied, (Object)null))) {
        Map backup = TransmittableThreadLocal.backupAndSetToCopied(copied);

        try {
            this.runnable.run();
        } finally {
            TransmittableThreadLocal.restoreBackup(backup);
        }
    } else {
        throw new IllegalStateException("TTL value reference is released after run!");
    }
}

这段代码的主要逻辑如下:

  1. 把当时的 ThreadLocal 做个备份,然后将父类的 ThreadLocal 拷贝过来
  2. 执行真正的 run 方法,可以获取到父类最新的 ThreadLocal 数据
  3. 从备份的数据中,恢复当时的 ThreadLocal 数据

11. ThreadLocal有哪些用途?

最后,一起聊聊 ThreadLocal 有哪些用途?

老实说,使用 ThreadLocal 的场景挺多的

下面列举几个常见的场景:

  1. 在 spring 事务中,保证一个线程下,一个事务的多个操作拿到的是一个 Connection
  2. 在 hiberate 中管理 session
  3. 在 JDK8 之前,为了解决 SimpleDateFormat 的线程安全问题
  4. 获取当前登录用户上下文
  5. 临时保存权限数据
  6. 使用 MDC 保存日志信息
    等等,还有很多业务场景,这里就不一一列举了

由于篇幅有限,今天的内容先分享到这里。希望你看了这篇文章,会有所收获

接下来留几个问题给大家思考一下:

ThreadLocal 变量为什么建议要定义成 static 的?(为了避免重复创建线程对象)
Entry 数组为什么要通过 hash 算法计算下标,即直线寻址法,而不直接使用下标值?
强引用和弱引用有什么区别?(强不会被 GC 回收,弱引用在 GC 时自动被回收)
Entry 数组大小,为什么是 2 的 N 次方?(2 的 N 次方可以使用取余算法,hash更简单)
使用 InheritableThreadLocal 时,如果父线程中重新 set 值,在子线程中能够正确的获取修改后的新值吗?(不能)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值