ThreadLocal源码解读

在公司的工程中，我们用ThreadLocal来存储请求中的分页参数，今天对ThreadLocal进行一下分析。

ThreadLocal的本质是线程Thread类的一个成员（threadLocals，类型是ThreadLocalMap）的元素。之所以能对不同的线程往同一个ThreadLocal中设置不同的值，是因为这个不同的值本来就是存在Thread类的threadLocals中。线程Thread与ThreadLocal关系如下：

现在来分析ThreadLocal的set与get方法。set方法如下：

public void set(T value) {
        //获取当前线程
        Thread t = Thread.currentThread();
        //拿到当前线程的成员threadLocals
        ThreadLocalMap map = getMap(t);
        if (map != null)
            //map存在则直接将<key,value>(<this, value>)保存到ThreadLocalMap中
            map.set(this, value);
        else
            //map不存在则初始化该map
            createMap(t, value);
    }

每个线程都有自己的ThreadLocalMap，上面createMap(t, value)执行如下：

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

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

INITIAL_CAPACITY的大小是16，i是该ThreadLocal映射到Entry上的hash值。value是保存在table的i位置。setThreshold是设置ThreadLocalMap的扩张阈值。策略是table填满了长度的2/3就进行扩张。再来看看map的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();
                //如果key存在，则覆盖原来的值
                if (k == key) {
                    e.value = value;
                    return;
                }
               	//如果该位置ThreadLocal被gc清除了，则占用该位置
                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }
            //新存一个Entry
            tab[i] = new Entry(key, value);
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
        }

现在分析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();
    }

get方法就比较简单了，首先获取当前线程的ThreadLocalMap,然后拿到包装类型元素Entry，如果存在则返回value，不存在返回setInitialValue()。

以上就是ThreadLocal分析，写的可能过于简单，如果没能太理解，可以给我留言。