ThreadLocal源码分析

set过程

1、ThreadLocal.set(T value)方法

 /**
     * Sets the current thread's copy of this thread-local variable
     * to the specified value.  Most subclasses will have no need to
     * override this method, relying solely on the {@link #initialValue}
     * method to set the values of thread-locals.
     *
     * @param value the value to be stored in the current thread's copy of
     *        this thread-local.
     */
    public void set(T value) {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
    }

当调用set值方法时，首先从当前线程获取ThreadLocalMap对象map，判断map是否为空，不为空则直接set值（key为threadlocal对象）,为空则调用createMap方法。

2、ThreadLocalMap map = getMap(t)中的getMap(t)源码。

/**
     * Get the map associated with a ThreadLocal. Overridden in
     * InheritableThreadLocal.
     *
     * @param  t the current thread
     * @return the map
     */
    ThreadLocalMap getMap(Thread t) {
        return t.threadLocals;
    }

3、t.threadLocals;的源码

public
class Thread implements Runnable {
     ....
     ....

    /* ThreadLocal values pertaining to this thread. This map is maintained
     * by the ThreadLocal class. */
    ThreadLocal.ThreadLocalMap threadLocals = null;
}

返回当前线程的ThreadLocalMap对象(默认为null)，ThreadLocalMap为ThreadLocal的静态内部类，由ThreadLocalMap中的enty对象存储value值。

4、createMap()方法

    /**
     * Create the map associated with a ThreadLocal. Overridden in
     * InheritableThreadLocal.
     *
     * @param t the current thread
     * @param firstValue value for the initial entry of the map
     */
    void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

createMap方法时创建一个新的ThreadLocalMap并向这个新的ThreadLocalMap中存储数据。

5、new ThreadLocalMap(this，firstValue)方法

        /**
         * Construct a new map initially containing (firstKey, firstValue).
         * ThreadLocalMaps are constructed lazily, so we only create
         * one when we have at least one entry to put in it.
         */
        ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
            table = new Entry[INITIAL_CAPACITY];
            int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
            table[i] = new Entry(firstKey, firstValue);
            size = 1;
            setThreshold(INITIAL_CAPACITY);
        }

new一个Entry数组（默认大小为INITIAL_CAPACITY = 16），并将this和value的值初始化到数组中。

get过程

/**
     * Returns the value in the current thread's copy of this
     * thread-local variable.  If the variable has no value for the
     * current thread, it is first initialized to the value returned
     * by an invocation of the {@link #initialValue} method.
     *
     * @return the current thread's value of this thread-local
     */
    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();
    }

如果已经set了值，则返回，如果未set值则返回null。

ps：为什么不直接向Thread类中的ThreadLocalMap对象存取数据呢？

threadLocals默认是包级别访问，所以不能直接从外部访问该变量，只能同包下的类可以访问threadlocals变量，而ThreadLocal和Thread恰好在同一个包中。

InheritableThreadLocal的使用