ThreadLocal 源码解读

一、引入

public class Thread implements Runnable {
    /* 前面略  */

    /* ThreadLocal values pertaining to this thread. This map is maintained
     * by the ThreadLocal class. */
    ThreadLocal.ThreadLocalMap threadLocals = null;
    /* 后面略  */
 }

首先我们看到的是 Thread 中有一个属性 threadLocals,它的类型是 ThreadLocalMap,封装类型是 default(表示它只能在包内可见),jdk 是这么介绍它的:与此线程有关的 ThreadLocal 值,该映射由 ThreadLocal 类维护。 啥意思呢?那就来看看 ThreadLocalMap 是啥玩意!

public class ThreadLocal<T> {
   /* 前面略  */

    static class ThreadLocalMap {

        /**
         * The entries in this hash map extend WeakReference, using
         * its main ref field as the key (which is always a
         * ThreadLocal object).  Note that null keys (i.e. entry.get()
         * == null) mean that the key is no longer referenced, so the
         * entry can be expunged from table.  Such entries are referred to
         * as "stale entries" in the code that follows.
         */
        static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }
   /* 后面略  */
   }
}

从类定义上可以看出 ThreadLocal 是支持泛型的,而 ThreadLocalMap 是 ThreadLocal 的一个内部类,封装类型也是 default(表示只能在包内可见),jdk 是这么介绍它的:ThreadLocalMap 是自定义的哈希映射,仅适用于维护线程局部值。并且为了存储容量可控,不至于内存泄漏,哈希表条目使用弱引用作为键(弱引用的对象的生命周期直到下一次垃圾回收之前被回收),ThreadLocalMap 使用静态内部类 Entry(可以类比 Map 中的 entry)来存储实际的 key 和 value。

从上面这些介绍,我们可以大致想到,ThreadLocal 是一个与线程相关的类,用来存储维护线程局部值。

二、set(T value) 方法解读

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

可以看到 set(T value) 方法做的事情很简单,就是维护 Thread 的 threadLocals 属性,如果该属性不存在的话,就以当前 ThreadLocal 实例为 key 创建一个;该属性存在的话,则直接赋值。

三、get() 方法解读

    public T get() {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            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 map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
        return value;
    }
    protected T initialValue() {
        return null;
    }

get() 的方法同样很简单,就是从 Thread 的 threadLocals 属性获取值,如果获取不到,则把 initialValue() 的值赋值给线程的 threadLocals 属性并返回。initialValue() 方法是一个 protected 类型的方法,默认返回 null,我们可以在创建 ThreadLocal 的时候重写它,表示所有线程的默认值。

    // java8 的方式
    ThreadLocal<Boolean> threadLocal1 = ThreadLocal.withInitial(() -> false);
    // 
    ThreadLocal<Boolean> threadLocal2 = new ThreadLocal<Boolean>() {
        @Override
        protected Boolean initialValue() {
            return false;
        }
    };

四、总结

  • ThreadLocal 用于存储维护线程的局部值。
  • 和 ThreadLocal 类似的还有一个叫 InheritableThreadLocal, InheritableThreadLocal 继承自 ThreadLocal,用于父子线程间共享共同的值,父线程中设置的值,子线程中可以访问到。
  • 上面看 ThreadLocalMap 的时候,我们知道 key 是弱引用,gc 的时候 key 会被回收,但是 value 和 ThreadLocalMap 的引用不会被回收,如果这种情况的 Thread 很多,而且一直没有执行完,就可能会出现内存泄漏,因此使用完 ThreadLocal 的时候尽量调用 ThreadLocal 的 remove() 方法。
  • 当使用线程池的时候,在调用 ThreadLocal 的 set 方法后,却没有调用 remove 的方法,如果同一个线程再去调用 get 方法可能拿到的值并不是当时 set 进去的(因为线程池的线程是复用的),可能导致程序数据异常之类的,因此使用完 ThreadLocal 的时候尽量调用 ThreadLocal 的 remove() 方法。
posted @ 2019-12-08 22:35  JMCui  阅读(...)  评论(...编辑  收藏