ThreadLocal原理分析

什么是ThreadLocal

ThreadLocal提供线程局部变量。这些变量与普通的变量不同之处在于，每个访问这种变量的线程（通过它的get或set方法）都有自己的、独立初始化的变量副本。

ThreadLocal的基本使用方法

        ThreadLocal<Object> thread = new ThreadLocal<>();
        thread.set("1111");
        System.out.println(thread.get());

ThreadLocal的原理

赋值的代码块

  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 中底层是一个Entry

  private Entry[] table;
  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();
        }

获取ThreadLocalMap 对象

ThreadLocalMap getMap(Thread t) {
        return t.threadLocals;
    }
ThreadLocal.ThreadLocalMap threadLocals = null;
/**/创建一个新的map对象，且赋值
void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

获取当前线程的局部变量

public T get() {
        Thread t = Thread.currentThread();
        /**/获取当前ThreadLocalMap对象
        ThreadLocalMap map = getMap(t);
        if (map != null) {
             //从map中获取局部变量的值
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        return setInitialValue();
    }

ThreadLocal相关问题解析

为什么当前线程缓存的是ThreadLocalMap 对象
在不同的场景中，为了防止局部变量互不影响，我们可以设置多个ThreadLoacl

ThreadLocal<Object> thread = new ThreadLocal<>();
        ThreadLocal<Object> thread1 = new ThreadLocal<>();
        thread.set("1111");
        thread1.set("2222");
        System.out.println(thread.get());
        System.out.println(thread1.get());

ThreadLocal的内存泄露问题：
什么是内存泄漏：申请了内存，一直无法释放
什么是内存溢出：申请了内存，发现申请内存无法满足需求
hreadLocalMap是一个比较特殊的Map，它的每个Entry的key都是一个弱引用。看代码，Entry 继承了
WeakReference

   static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;

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

如何避免内存泄露

既然已经发现有内存泄露的隐患，自然有应对的策略，在调用ThreadLocal的get()、set()可能会清除ThreadLocalMap中key为null的Entry对象，这样对应的value就没有GC Roots可达了，下次GC的时候就可以被回收，当然如果调用remove方法，肯定会删除对应的Entry对象。
如果使用ThreadLocal的set方法之后，没有显示的调用remove方法，就有可能发生内存泄露，所以养成良好的编程习惯十分重要，使用完ThreadLocal之后，记得调用remove方法。

如有错误青留言指出谢谢！！！