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

来源:互联网 发布:python股票大数据分析 编辑:程序博客网 时间:2024/06/07 02:39

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();        }
原创粉丝点击