ThreadLocal-从源码看设计

ThreadLocal-从源码看设计

一、ThreadLocal定义

ThreadLocal提供线程局部变量。这些变量与它们在每个线程中访问，每个线程有自己的、独立初始化的变量副本，实例通常是私有的类中希望将状态与线程关联的静态字段。即在每个显示访问当前实例对象，都会获取各自线程对应的值，不会相互影响，不存在线程安全问题，也不会影响程序执行的性能。由于在各自线程存储对应的值，所以内存消耗会比直接使用实际类对象要大。

二、从源码看设计

1.get方法

    /**
      *返回当前线程的副本中的值线程局部变量。 
      *如果该变量对于当前线程，它首先被初始化为返回的值
      *通过调用{@link #initialValue}方法。
      *
      * @return 此本地线程的当前线程值
      **/
    public T get() {
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取当前线程的ThreadLocalMap成员变量
        ThreadLocalMap map = getMap(t);
        //如果当前线程ThreadLocal已经初始化
        if (map != null) {
            //获取当前ThreadLocal实例对应的Entry实体
            ThreadLocalMap.Entry e = map.getEntry(this);
            //如果不为null,取出Entry对应值
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        //如果当前线程ThreadLocal未初始化
        return setInitialValue();
    }

    /**
     * 获取与ThreadLocal关联的Map。 在InheritableThreadLocal中重写。
     *
     * @param  t 当前线程
     * @return 当前线程的ThreadLocalMap
     */
    ThreadLocalMap getMap(Thread t) {
        return t.threadLocals;
    }

   /**
     * set()的变体来建立initialValue。如果用户已覆盖set()方法，请使用它代替set()。
     *
     * @return 初始化的值
     */
    private T setInitialValue() {
        //默认返回null
        T value = initialValue();
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取线程对应map
        ThreadLocalMap map = getMap(t);
        //map已经实例化，则设置value返回;否则创建ThreadLocalMap
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
        return value;
    }       

    /**
     * 创建ThreadLocal关联的map对象，并赋值给当前线程。在InheritableThreadLocal中被重写
     *
     * @param t 当前线程
     * @param firstValue 初始化map的实体值
     */
    void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

2. set方法

   /**
     * 将此线程局部变量的当前线程副本设置为指定值。 大多数子类将不需要重写此方法，而仅依靠{@link #initialValue}方法来设置线程局部变量的值。
     *
     * @param value 要存储在此本地线程的当前线程副本中的值。
     */
    public void set(T value) {
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取当前线程对应的ThreadLocalMap
        ThreadLocalMap map = getMap(t);
        //map不为空，以当前对象为key,值为value设置进map；否则创建map
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
    }

3.总结

get()方法

获取的值是当前线程对象里的ThreadLocalMap变量，然后从ThreadLocalMap变量中获取到对应ThreadLocal对象的值。

set()方法

将要设置的值，保存到从当前线程获取到的ThreadLocalMap中。

通过这两个方法，就达到了多线程多个变量副本的效果。

4.ThreadLocalMap

是ThreadLocal的内部类，用于存储不同ThreadLocal对象和对应的值。内部采用一个初始容量为16的数组存储ThreadLocal对应的Entry实体.

static class ThreadLocalMap {

        /**
         * 此哈希映射中的条目使用其主引用字段作为键（始终是ThreadLocal对象）扩展WeakReference。 
         请注意，空键（即entry.get（）== null）意味着不再引用该键，因此可以从表中删除该条目。 
         
         在下面的代码中，此类条目称为“陈旧条目”。
         */
        static class Entry extends WeakReference<ThreadLocal<?>> {
            /** 这个ThreadLocal关联的值 */
            Object value;

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }

        /**
         * 初始容量-必须是2的幂
         */
        private static final int INITIAL_CAPACITY = 16;

        /**
         * 数组表，根据需要调整大小。 table.length必须始终为2的幂。
         */
        private Entry[] table;

        /**
         * 表格中实体数量
         */
        private int size = 0;

        /**
         * 下一个要调整大小的大小值。
         */
        private int threshold; // 默认为0
    }

扩容条件：当数组大小，超过容量的2/3扩容

    /**
      * 设置调整大小阈值以保持最坏的2/3负载系数。
      */
     private void setThreshold(int len) {
         threshold = len * 2 / 3;
     }

ThreadLocalMap在这里只简介，核心重点是分析同步机制，需要深入了解具体数据结构的，请查看源码分析

ThreadLocal和同步机制区别

ThreadLocal：
1.空间换时间
2.在不同线程空间存储多份实例

同步机制:
1.时间换空间
2.一个实例，不同线程排队访问