在Java编程学习过程中,一定会学习如何保证线程安全,提到线程安全,也就不得不提到ThreadLocal了。
ThreadLocal 是线程的局部变量, 是每一个线程所单独持有的,其他线程不能对其进行访问, 通常是类中字段修饰为 private static ,是对该字段初始值的一个拷贝,它们希望将状态与某一个线程(例如,用户 ID 或事务 ID)相关联。
当使用ThreadLocal维护变量的时候 为每一个使用该变量的线程提供一个独立的变量副本,即每个线程内部都会有一个该变量,这样同时多个线程访问该变量并不会彼此相互影响,因此他们使用的都是自己从内存中拷贝过来的变量的副本, 这样就不存在线程安全问题,也不会影响程序的执行性能。但是由于在每个线程中都创建了副本,所以要考虑它对资源的消耗,比如内存的占用会比不使用ThreadLocal要大。
Thread 在内部是通过ThreadLocalMap来维护ThreadLocal变量表, 在Thread类中有一个threadLocals 变量,是ThreadLocalMap类型的,它就是为每一个线程来存储自身的ThreadLocal变量的, ThreadLocalMap是ThreadLocal类的一个内部类,这个Map里面的最小的存储单位是一个Entry, 它使用ThreadLocal作为key, 变量作为 value,这是因为在每一个线程里面,可能存在着多个ThreadLocal变量。
初始时,在Thread里面,threadLocals为空,当通过ThreadLocal变量调用get()方法或者set()方法,就会对Thread类中的threadLocals进行初始化,并且以当前ThreadLocal变量为键值,以ThreadLocal要保存的副本变量为value,存到threadLocals。
然后在当前线程里面,如果要使用副本变量,就可以通过get方法在threadLocals里面查找。
下面我们一起来简单研究一下TheadLocal的源码。(本文基于JDK1.8)
1、get()方法
/**
* 用来获取ThreadLocal在当前线程中保存的变量副本
* @return
*/
public T get() {
Thread t = Thread.currentThread();
// 获取当前线程的ThreadLocalMap,ThreadLocalMap在后面简单进行介绍
ThreadLocalMap map = getMap(t);
// 若当前线程有TheadLocalMap,则得到当前theadLocal所对应的值。
// 若当前线程不存在ThreadLocalMap,则调用setInitialValue()方法。
if (map != null) {
ThreadLocalMap.Entry e = map.getEntry(this);
if (e != null) {
@SuppressWarnings("unchecked")
T result = (T)e.value;
return result;
}
}
return setInitialValue();
}
若当前线程不存在TheadLocalMap则调用setInitialValue方法进行设置,并返回null,下面简单看一下其源码。
protected T initialValue() {
return null;
}
private T setInitialValue() {
// initialValue()方法直接返回了null
T value = initialValue();
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
// 若当前线程存在ThreadLocalMap,则进行set操作,否则创建新的Map,并且设置当前线程的值。
if (map != null)
map.set(this, value);
else
createMap(t, value);
return value;
}
2、set()方法
public void set(T value) {
// 首先获取当前线程的ThreadLocalMap,保证操作器自己的TheadLocalMap
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
// 若当前线程存在ThreadLocalMap,则进行set操作,否则创建新的Map,并且设置当前线程的值。
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
3、remove方法,即将此threadLocal从当前线程所持有的TheadLocalMap中移除,较简单。
4、getMap()方法
/**
* 返回为指定线程的TheadLocalMap
* @param t
* @return
*/
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
5、creatMap()方法
/**
* 创建新的ThreadLocalMap
* @param t
* @param firstValue
*/
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
在ThreadLocal类中并没有TheadLocalMap成员,那么它是持有ThreadLocalMap的呢?
其实TheadLocalMap是Thread类的一个成员,TheadLocal通过Thread来持有当前线程的ThreadLocalMap,从而进一步管理当前线程的局部变量。而ThreadLocalMap是ThreadLocal的内部类,定义了初始化容量INITIAL_CAPACITY、存放值的Entry数组、Map的大小size以及负载因子threshold。
ThreadLocalMap的键为当前threadLocal,而值为真正存储的Object。下面简单看一下getEntry方法和set方法的源码。
/**
* 从Map中获取某一个threadLocal所所对应的值。
* @param key
* @return
*/
private Entry getEntry(ThreadLocal<?> key) {
// 计算指定threadLocal的位置
int i = key.threadLocalHashCode & (table.length - 1);
Entry e = table[i];
// 获取计算位置的元素,并且比较是否为所指定的threadLocal,否则调用getEntryAfterMiss
if (e != null && e.get() == key)
return e;
else
return getEntryAfterMiss(key, i, e);
}
/**
* 设置某一个threadLocal对应的值Object
* @param key
* @param value
*/
private void set(ThreadLocal<?> key, Object value) {
Entry[] tab = table;
int len = tab.length;
int i = key.threadLocalHashCode & (len-1);
// 获取计算位置的元素entry
for (Entry e = tab[i];
e != null;
e = tab[i = nextIndex(i, len)]) {
ThreadLocal<?> k = e.get();
// 若循环的当前位置就是所执行的threadLocal的key,则将value值进行替换
if (k == key) {
e.value = value;
return;
}
if (k == null) {
replaceStaleEntry(key, value, i);
return;
}
}
// 若没有找到相同的threadLocal,则生成新的entry,并且自增size
tab[i] = new Entry(key, value);
int sz = ++size;
if (!cleanSomeSlots(i, sz) && sz >= threshold)
rehash();
}
在需resize()扩容时,其所进行的操作同hashMap所做的操作大体相同,这里不咋赘述,如有需要可以去博主的HashMap源码解析中去看看。
6、Entry
Entry作为存储数据的数据结构,是ThreadLocalMap的一个内部类,其内部只有一个成员,即Object类型的成员,用来存放各类值信息。具体如下所示:
static class Entry extends WeakReference<ThreadLocal<?>> {
/** The value associated with this ThreadLocal. */
Object value;
Entry(ThreadLocal<?> k, Object v) {
super(k);
value = v;
}
}