ThreadLocal学习（二）

ThreadLocal学习（二）

了解了ThreadLocal的使用方法后，我们继续探究ThreadLocal的底层实现原理。

如果我们去翻阅ThreadLocal的源码的话可以发现ThreadLocal有四种方法：

方法	作用
protected T initialValue()	返回当前线程局部变量的初始值
public void set(T value)	设置当前线程绑定的局部变量
public T get()	获取当前线程绑定的局部变量
public void remove()	移除当前线程绑定的局部变量

那么我们逐步对上面的方法进行解析

• set方法

源码：

/**
         * 设置当前线程对应的ThreadLocal的值
         * value:将要保存在当前线程的ThreadLocal的值
         */
        public void set(T value) {
            //获取当前线程对象
            Thread t = Thread.currentThread();
            //获取此线程中的ThreadLocalMap对象
            ThreadLocal.ThreadLocalMap map = getMap(t);
            //判断map是否为空
            if (map != null)
                map.set(this, value);
            else
                createMap(t, value);
        }

/**
         * this是指当前ThreadLocal对象
         */
		void createMap(Thread t, T firstValue) {
		        t.threadLocals = new ThreadLocalMap(this, firstValue);
		    }

总结set方法：

1. 首先获取当前线程，并根据当前线程获取一个map
2. 判断获取map是否为空，不为空，则将参数设置到map中（当前ThreadLoacl的引用作为key）
3. 为空，则创建map并设置初始值

• get方法

源码：

/**
     * 返回当前线程中保存的ThreadLocal的值
     * 如果线程中没有ThreadLocal变量则会调用setInitialValue（）方法进行初始化值
     * @return
     */
    
    public T get() {
        //获取当前线程对象
        Thread t = Thread.currentThread();
        //获取此线程对象中的ThreadLocalMap对象
        ThreadLocal.ThreadLocalMap map = getMap(t);
        //判断map是否为空
        if (map != null) {
            //以当前ThreadLocal为key，调用getEntry获取对应的存储实体e
            ThreadLocal.ThreadLocalMap.Entry e = map.getEntry(this);
            //判断e是否为空
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        //两种情况执行返回：
        //1.map不存在，表示此线程没有维护的ThreadLocalMap对象
        //2.map存在，但是没有与当前ThreadLocal关联的entry
        return setInitialValue();
    }

    private T setInitialValue() {
        //此方法可以被重写，如果不重写默认返回null
        T value = initialValue();
        //剩下与set方法类似
        Thread t = Thread.currentThread();
        ThreadLocal.ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
        return value;
    }

总结get方法：

1. 首先获取当前线程，根据当前线程获取一个map。
2. 如果获取的map不为空，则在map中医ThreadLocal的引用作为key来在map中获取对应的entrye,否则转到4。
3. 如果e不为空，则返回e.value，否则转到4。
4. map为空或者e为空，则通过InitialValue()函数获取初始值value，然后用ThreadLocal个value作为firstKey和firstValue创建一个新的Map。

• remove方法

源码：

public void remove() {
         ThreadLocalMap m = getMap(Thread.currentThread());
         if (m != null)
             m.remove(this);
     }

很简单的移除方法。

关于源码的介绍就到这里啦！