java源码阅读之深入理解ThreadLocal

2019-03-31 17:48:15 阅读：301 来源： 互联网

标签：map java Thread ThreadLocalMap value ThreadLocal 源码 Entry

总结：每一个Thread实例中都会保存一个map数据结构，其中用来存储Thread实例用到的每个ThreadLocal的哈希值和其对应的value。使用set和get方法时，利用ThreadLocalHashCode在Thread实例的map中查找和修改其value值。

先看简介

ThreadLocals rely on per-thread linear-probe hash maps attached to each thread (Thread.threadLocals and inheritableThreadLocals). The ThreadLocal objects act as keys, searched via threadLocalHashCode.

本地线程们依赖挂载到每个线程上的哈希表maps。本地线程对象为key，通过threadLocalHashCode搜索。

从初始化函数切入

    private T setInitialValue() {
        T value = initialValue();
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            map.set(this, value);
        } else {
            createMap(t, value);
        }
        if (this instanceof TerminatingThreadLocal) {
            TerminatingThreadLocal.register((TerminatingThreadLocal<?>) this);
        }
        return value;
    }

我们可以看到使用了ThreadLocalMap这个对象，其为ThreadLocal的内部静态类，阅读后得知其中有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;
            }
        }

继续阅读初始化函数，通过getMap获取map对象
ThreadLocal.ThreadLocalMap threadLocals = null;
每一个Thread对象都保存了一个map用来保存ThreadLocal和value副本，如果map为null则在Thread的map中添加一个新的Entry对象（稍后会解释）

    void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

注意：创建map中新的Entry<key,value>对象的key不是Thread实例的hashcode而是ThreadLocalHashCode
这样每个Thread通过维持一个map数据结构就拥有其所有的ThreadLocal中的变量副本了

Set方法

    public void set(T value) {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            map.set(this, value);
        } else {
            createMap(t, value);
        }
    }

依旧从Thread实例中获取map，如果没有该变量则在map中新建一个Entry<ThreadLocalHashCode, value>，如果有的话就修改

get方法

    public T get() {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        return setInitialValue();
    }

如果map不为空则获取LocalThreadMap中的Entry对象，以ThreadLocalHashCode在map中匹配value，如果map为空则重新进行初始化

标签：map,java,Thread,ThreadLocalMap,value,ThreadLocal,源码,Entry
来源： https://blog.csdn.net/weixin_42072543/article/details/88930877