多线程访问同一个共享变量可能出现并发问题,尤其是使用线程写入的时候。为了保证线程安全,往往需要进行加锁,但是加锁操作往往是一件耗时的工作,在有些情况(例如数据库连接、session管理等)可以采取另外的方式,将共享变量存入线程本地解决竞争问题,每个线程只操作本地变量。
ThreadLocal就可以做到这件事:创建了一个ThreadLocal变量,就为线程创建了那个共享变量的本地副本。但是这个本地变量不是存在ThreadLocal里,而是存在Thread类的threadLocals变量里,这个是一个ThreadLocalMap(一种HashMap),key存放的是当前线程的ThreadLocal实例,value是共享变量的本地副本。
ThreadLocal只要有三个方法:
1、void set(T value)
在当前线程里设置value的一个本地副本
public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
首先获取当前线程Thread t = Thread.currentThread()
,通过t得到当前线程的threadLocals,
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
如果threadLocals不是空,则存入当前的共享变量副本,否则,新建一个threadLocals,
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
2、T get()
用来获取ThreadLocal在当前线程中保存的变量副本
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,
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
如果threadLocals不为空,则返回对应的本地变量,否则表示threadLocals还没有创建,需要初始化
return 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;
}
3、void remove()
删除ThreadLocal中当前线程的本地变量
public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null)
m.remove(this);
}
ThreadLocal使用不当会导致内存泄漏
关键是ThreadLocalMap的key是弱引用类型
static class ThreadLocalMap {
static class Entry extends WeakReference<ThreadLocal<?>> {
Object value;
Entry(ThreadLocal<?> k, Object v) {
super(k);
value = v;
}
}
...
}
public class WeakReference<T> extends Reference<T> {
public WeakReference(T referent) {
super(referent);
}
public WeakReference(T referent, ReferenceQueue<? super T> q) {
super(referent, q);
}
}
Reference(T referent) {
this(referent, null);
}
Reference(T referent, ReferenceQueue<? super T> queue) {
this.referent = referent;
this.queue = (queue == null) ? ReferenceQueue.NULL : queue;
}
由上述代码可以看到,加入到ThreadLocalMap的key最终作为Reference的referent。
因此当下一次GC时,ThreadLocalMap的key一定会回收,而ThreadLocalMap实际上是就是threadLocals变量,与线程生命周期相同,这时就会存在key为null而value不为null的map,这些value无法被访问也不能被释放,所以造成内存泄漏。
但是如果ThreadLocalMap的key设计成强引用,外部已经没有对ThreadLocal对象的引用,但是ThreadLocalMap仍然在强引用它,ThreadLocal对象应该被回收而不能被回收,引起内存泄漏
所以,不管ThreadLocalMap的key是强引用还是弱引用,都可能引起内存泄漏,问题的根源在ThreadLocalMap的生命周期和Thread一样。解决方法是手动删除ThreadLocalMap的key,手动调用remove方法。