前言
基于引用的学习，来简单的看一下ThreadLocal的实现。ThreadLocal简单来时是线程本地变量，解决共享对象(单个线程内共享)的多线程访问问题的，其不同于synchonized的关键点在于，synchronized是利用锁机制，保证共享对象在某一时刻只能被一个线程访问，但是ThreadLocal是为每个线程提供变量副本。其劣势在于为每个线程并发访问的数据建立了一个副本，会造成内存消耗。 常见的坑：没有进行remove操作，会造成内存泄露问题。（注意并不是ThreadLocal出现的地方就得调用remove，如果ThreadLocal是静态变量等情况，其生命周期和程序一样长，就不是很必要了）
内存泄露原因分析
ThreadLocalMap
static class ThreadLocalMap {
// 继承自弱引用意味着在发生gc时，如果key值ThreadLocal被发现只有一个弱引用，其会被回收，此时如果发现创建ThreadLocal的线程一直在运行的话，那么Entry对象的value值将一直得不到回收，发生内存泄漏。
static class Entry extends WeakReference<ThreadLocal> {
/** The value associated with this ThreadLocal. */
Object value;
/**
* public WeakReference(T referent) {
* super(referent);
* }
*/
Entry(ThreadLocal k, Object v) {
super(k);
value = v;
}
}
...
// 由set方法中createMap调用
ThreadLocalMap(ThreadLocal firstKey, Object firstValue) {
table = new Entry[INITIAL_CAPACITY];
int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
table[i] = new Entry(firstKey, firstValue);
size = 1;
setThreshold(INITIAL_CAPACITY);
}
...
// 由map方法中map.getEntry(this)调用
private Entry getEntry(ThreadLocal key) {
int i = key.threadLocalHashCode & (table.length - 1);
Entry e = table[i];
if (e != null && e.get() == key)
return e;
else
// 此过程中会清理key值为空的Entry对象
return getEntryAfterMiss(key, i, e);
}
}
set方法
public void set(T value) {
// t为正在执行的当先线程的句柄
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
// 此过程也会清理key值为空的Entry对象
map.set(this, value);
else
createMap(t, value);
}
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
/**
* ThreadLocal.ThreadLocalMap threadLocals = null;
*/
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
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();
}
简单示例
public class ThreadLocalTest {
public static void main(String[] args) {
ThreadLocal<String> localName = new ThreadLocal<>();
try {
localName.set("Hello World");
String name = localName.get();
System.out.println(name);
} finally {
// 此处请注意
localName.remove();
}
}
}
