ThreadLocal源码及存在的内存泄漏分析

ThreadLocal 是一个特殊的对象，用于每个线程存储自己的对象，线程安全，使用很简单。

ThreadLocal local = new ThreadLocal();
T t = new T();
local.set(t); 
T t1 = local.get();

实现也很简单，本文从set和get方法来分析ThreadLocal内部源码实现

// set 源码
    public void set(T value) {
        // 第一步 获取当前线程 然后获取当前线程的ThreadLocalMap对象
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
    }

set方法实现很简单，从上面源码中可以看出，每个线程对象都有一个ThreadLocalMap对象以ThreadLocal对象为key来存储值。
ThreadLocalMap是ThreadLocal内部的一个实现，类似于map的实现，也是键值对存储。从上面源码可以看出threadLocal的set就是ThreadLocalMap的set实现。同样 threadLocal的get方法也是ThreadLocalMap的get来获取值，源码不一一展示了。

继续分析ThreadLocalMap的实现。内部也是Entry实现，重点分析Entry接口，源码如下

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

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

从源码中可以看出，该Entry和Map.Entry实现有点不一样，ThreadLocalMap的Entry的key是ThreadLocal的弱引用，至于这里为什么是弱引用，在下面逐一分析。
然后继续分析ThreadLocalMap的get，set方法。
下面下分析set方法，先贴源码

private void set(ThreadLocal<?> key, Object value) {

            // We don't use a fast path as with get() because it is at
            // least as common to use set() to create new entries as
            // it is to replace existing ones, in which case, a fast
            // path would fail more often than not.

            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) {
                    //如果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();
        }