线程内部存储---TheadLocal从使用到源码分析

ThreadLocal是什么？

ThreadLocalThreadLocal是一个线程内部用于存储数据的类，通过它可以在指定的线程中存储数据，数据存储以后，只有在该线程中可以获取到存储的数据，对于其它线程来说无法获取到数据。个人认为是一个线程内部的存储机制。

如何使用？

ThreadLocal<Boolean> threadLocal = new ThreadLocal<>();

这样就在一个线程创建了ThreadLocal这个对象，这个对象支持范性的，也就是说我们可以存储的自己想存的任意类型。

threadLocal.set(true);

这样我们就在当前所在线程，存储了true。

new Thread("Thread#1") {

@Override

public void run() {

System.out.println(threadLocal.get());

};

}.start();

我们在一个分线程去打印这个刚才存储的值会发现是null，因为这是在两个线程操作的。如果在当前线程获取则为正确刚才存的值。

如果想移除这个数据也很简单：

threadLocal.remove();

这就是threadLocal的使用，其实很简单。 主要记得是区分线程的就ok。

内部源码是如何实现的呢？

这就是treadLocal中的所有方法了，其实我们最关心的就是set和get方法。

我们先看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);

}

这就是set方法，对就这么几行。

首先会通过Thread.currentThread();这个方法获取当前线程的Thread。然后通过getMap()获取 ThreadLocalMap对象。

ThreadLocalMap getMap(Thread t) {

return t.threadLocals;

}

getMap中也很简单，就是返回这个线程的threadLocals, 这个ThreadLocal就是 Thread中的一个变量ThreadLocal.ThreadLocalMap threadLocals = null;

其实还是ThreadLocal中的内部类ThreadLocalMap；这里一会在分析，先顺着思路往下走。

获取map之后会根据判断如果不是null就进行set,也就是存储，如果是null就会调用createMap()方法进行创建这个ThreadMap。接下来在看是如何set和createMap的。

创建其实很简单,直接就是new一个。

void createMap(Thread t, T firstValue) {

t.threadLocals = new ThreadLocalMap(this, firstValue);

}

set方法就稍微复杂了些，因为这也是核心内容。set方法在内部类ThreadLocalMap中，所以接下来分析下这个ThreadLocalMap类。

ThreadLocalMap

这个类在构造中创建了一个数组， new Entry[INITIAL_CAPACITY]; ,Entry里面就是一个object的对象,然后里面主要getEntry和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) {//这里判断这个k和传进来的key是否相等
e.value = value;//进行存储后return
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();
}

““

看主要代码 判断k和传进来的key相等话就存储这个value

get方法：

“`

private Entry getEntry(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)

return (T)e.value;

}

return setInitialValue();

}

get里面也是要先获取当前的Thread这也就是为什么ThreadLocal获取和存储的都是只在当前线程的。

然后也是getMap方法获取这个ThreadLocalMap，这也就是为什么里面就一行代码，也要写成一个方来，因为这是中思路，其他地方获取直接调用就行，日后扩展的话一样方便。

再然后就是通过这个map去调用上面说的getEntry方法。

至此，我们就知道ThreadLocal的总体工作流程和思路了。