ThreadLocal线程内共享变量源码详解

1.作用 :线程内共享变量,也就是说该ThreadLocal中的数据作用域是线程内的.
2.源码解析
ThreadLocal 有三个自己的方法
分别是set(),get(),remove()
我们就根据这三个方法来看下ThreadLocal的具体用法,以及原理
set(T value):
先从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);
}
可以看出实际上我们真正的数据是存储在ThreadLocal的一个静态内部类ThreadLocalMap中的,该类是一个Map,以当前的ThreadLocal对象作为key的,那么线程又是怎样与之联系的呢?我们接下来看getMap(t)
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
可以看到,实际上我们得到的ThreadLocalMap是Thread对象的一个变量,但是我们在Thread类中并没有看到给其赋值的代码,那这个变量是如何初始化的呢?接下来
我们可以看到如果map是null则调用creatMap方法,如下
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
这时候可以知道 本线程也就是t中的ThreadLocalMap类型的threadLocals变量是在这里进行了初始化
并且是以本ThreadLocal对象作为key的.
以上就可以得出结论:
当ThreadLocal对象调用set方法时,也就是存储值的时候,是将本对象作为key储存在一个类似的Map中(其实是Entry实例),并将其副本寄存在本线程中,本线程就维护了一个这样的map,其实该map只有一个键值对,但是我们可以将value存储为map的类型这样,就可以存储多个数据了.

T 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();
}
我们可以看到其实先取的就是线程中的threadLocals变量,然后再取threadLocals的Entry中的以本ThreadLocal对象作为key的value值,如果threadLocals是null,则调用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;
}
可以看到除了第一行 T value = initialValue();剩余的逻辑与set方法一样,也就是先去取默认值initialValue();但是看其源码:
protected T initialValue() {
return null;
}
默认值是null;这里有两点需要注意:
一是ThreadLocal有初始化的默认值的方法的,需要重写;
二是为什么setInitialValue()方法不调用set()方法去代替除了第一行之后的逻辑,个人认为完全可以用其替换.

remove():
public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null)
m.remove(this);
}
该方法相对简单,没什么可以研究的.
欢迎各位大牛来原文基地视察 http://www.poppyflower.cn/?p=103