一、ThreadLocal的实现原理
在Thread类中,定义了一个变量ThreadLocal.ThreadLocalMap threadLocals = null;用于保存与此线程有关的ThreadLocal值,也就是说在ThreadLocal存在一个Map被用来保存和当前线程有关的变量值。
1、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;
}
setInitialValue用来设置线程的初始值。
2、initialValue()
返回线程初始值,可以被子类覆盖,用于线程的初始化。
protected T initialValue() {
return null;
}
3、get()
首先获取当前线程对象,然后通过getMap()拿到线程ThreadLocalMap,通过ThreadLocalMap获取到线程中的变量值。如果ThreadLocalMap是null,则先调用setInitialValue初始化线程的变量值后返回。
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();
}
4、set(T value)
设置当前线程中变量的值,首先获取当前线程,通过当前线程获取到线程中的变量Map,通过ThreadLocalMap 保存线程的变量值。
public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
5、remove()
移除当前线程中的变量,以便在线程退出时,不再持有其他对象的引用。
public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null)
m.remove(this);
}
6、exit()
线程退出时调用的方法,可以看出,因为这些变量是维护在Thread类内部的,所以意味着只要线程不退出,对象的引用将一直存在。而当线程退出时,Thread类会进行清理。
private void exit() {
if (group != null) {
group.threadTerminated(this);
group = null;
}
/* Aggressively null out all reference fields: see bug 4006245 */
target = null;
/* Speed the release of some of these resources */
threadLocals = null;
inheritableThreadLocals = null;
inheritedAccessControlContext = null;
blocker = null;
uncaughtExceptionHandler = null;
}
需要注意的是,如果使用固定大小的线程池,将一些比较大的对象设置在ThreadLocal中,则可能会使系统出现内存泄漏。如果希望可以及时回收,则需要调用ThreadLocal.remove()方法将变量移除。或者直接通过设置ThreadLocal为null。让系统进行垃圾回收。
二、使用场景
ThreadLocal 适用于每个线程需要自己独立的实例且该实例需要在多个方法中被使用,也即变量在线程间隔离而在方法或类间共享的场景。