JUC-ThreadLocal

一、ThreadLocal简介

        针对共享资源的线程安全问题，ThreadLocal可以实现资源对象的线程隔离，让每个线程各用各的资源对象，避免争用引发的线程安全问题，并且还可以实现线程内资源共享。具体原理如下图：

二、ThreadLocal原理

        每个线程内又一个ThreadLocalMap类型的成员变量，用来存储资源对象。可看出ThreadLocalMap是ThreadLocal的内部静态类，而它的构成主要是用Entry来保存数据 ，而且还是继承的弱引用。在Entry内部使用ThreadLocal作为key，使用我们设置的value作为value。

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

 问题一：为什么ThreadLocalMap中的key是弱引用？

       Thread可能需要长时间运行(如线程池中的线程)，如果没有良好的编程习惯，ThreadLocalMap中的键值对将会越来越多，如果key不再使用，需要在内存不足(GC)时释放其占用的内存，如果是key是强引用，即使是垃圾回收，也无法释放其所占用的内存，因此key是弱引用，只要别的地方没有再引用这个key，就可以释放掉内存。

        GC仅是让key的内存释放，后续还要根据key是否为null来进一步释放值的内存。

问题二：ThreadLocalMap中的value回收时机？

        首先，当ThreadLocalMap中的key已经被回收，value仍然存在。

• 当执行get()方法时，发现key=null，则把value清掉，并设置新的key值。

• 当执行set()方法时，发现key=null，则会使用启发式扫描，清除临近的nullkey，启发次数与map集合中元素个数有关，个数多则清除的nullkey多，反之则少。同时还和是否发现nullkey有关，如果发现nullkey就多扫描几次，反之就少。

•  一般使用ThreadLocal时都会把它作为静态变量，静态变量是强引用，因此GC是无法回收的。所以推荐使用remove()方法回收。

 三、源码分析

1、get()方法

    public T get() {
        //得到当前线程
        Thread t = Thread.currentThread();
        //得到当前线程中的ThreadLocalMap
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            //得到当前线程的ThreadLocalMap所储存的键值对
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        //初始化的结果就是value=null
        return setInitialValue();
    }

        private T setInitialValue() {
        //这里的value=null
        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;
    }

2、set()方法 

    public void set(T value) {
        //获取当前线程
        Thread t = Thread.currentThread();
        //获取当前线程的ThreadLocalMap
        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);
    }

由以上代码可以看出，set方法首先获取到当前线程的ThreadLocalMap，如果不为空则更新，为空则初始化。