Java ThreadLocal

ThreadLocal的作用是提供线程内局部变量，这种变量在线程生命周期内起作用。

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);
    }

    ThreadLocalMap getMap(Thread t) {
        return t.threadLocals;
    }

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

Thread

    ThreadLocal.ThreadLocalMap threadLocals = null;

ThreadLocal.set()通过对应的线程Thread获取Thread成员变量threadLocals。threadLocals是ThreadLocal的内部类ThreadLocalMap。map为null，创建ThreadLocalMap并给赋值给Thread的成员变量threadLocals。然后调用map.set(this, value)，这里this--->ThreadLocal。

ThreadLocalMap.set()

        private void set(ThreadLocal key, Object value) {
            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) {
                    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();
        }

ThreadLocalMap很明显是个map。这里不纠结其具体实现，只需知道map的key是ThreadLocal对象，value是set()的值。

ThreadLocal.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();
    }

    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);
        return value;
    }

    protected T initialValue() {
        return null;
    }

get()方法从ThreadLocalMap中取值

• ThreadLocal通过其内部类ThreadLocalMap来维护变量值。
• ThreadLocalMap存储在Thread对象中。
• ThreadLocalMap的key是ThreadLocal对象，value是变量值。