问题：ThreadLocalMap - entry
static class Entry extends WeakReference<ThreadLocal<?>> {
/** The value associated with this ThreadLocal. */
Object value;
Entry(ThreadLocal<?> k, Object v) {
// 由于key是弱类型引用，value不是，所以容易出现内存溢出的问题，如果不去手动remove的话
super(k);
value = v;
}
}ThreadLocalMap的问题
由于ThreadLocalMap的key是弱引用，而Value是强引用。这就导致了一个问题，ThreadLocal在没有外部对象强引用时，发生GC时弱引用Key会被回收，而Value不会回收，如果创建ThreadLocal的线程一直持续运行，那么这个Entry对象中的value就有可能一直得不到回收，发生内存泄露。
如何避免泄漏
既然Key是弱引用，那么我们要做的事，就是在调用ThreadLocal的get()、set()方法时完成后再调用remove方法，将Entry节点和Map的引用关系移除，这样整个Entry对象在GC Roots分析后就变成不可达了，下次GC的时候就可以被回收。
如果使用ThreadLocal的set方法之后，没有显示的调用remove方法，就有可能发生内存泄露，所以养成良好的编程习惯十分重要，使用完ThreadLocal之后，记得调用remove方法。
ThreadLocal不支持继承性
public class TestThreadLocal {
public static ThreadLocal<String> threadLocal = new ThreadLocal<>();
public static void main(String[] args) {
// 设置main线程应变量
threadLocal.set("hello world");
Thread thread = new Thread(new Runnable() {
@Override
public void run() {
System.out.println("thread:" + threadLocal.get());
}
});
thread.start();
System.out.println("main:" + threadLocal.get());
}
}输出结果：
一：前面也说了是线程本地变量，其他线程不能访问。二：这结果也说明在子线程中访问不到父线程的变量。
解决方案：
// 只需要创建InheritableThreadLocal类的对象就能解决这个问题
public static ThreadLocal<String> threadLocal = new InheritableThreadLocal<>();InheritableThreadLocal 类
为了解决在子线程中需要访问父线程的相关变量的问题，InheritableThreadLocal应运而生。
应用场景
比如子线程需要存放在父线程的用户登录信息、一些中间件需要把统一的id追踪的整个调用链路记录下来
原理分析
// 一、介绍InheritableThreadLocal类中被重写的方法
// 当调用ThreadLocal类的get方法时，get()方法里面调用的是重写后的getMap()方法，
// 然后返回的值也是inheritableThreadLocals
ThreadLocalMap getMap(Thread t) {
return t.inheritableThreadLocals;
}
// 第一次调用set方法时，会初始化创建inheritableThreadLocals变量的实例，而不再是threadLocals
void createMap(Thread t, T firstValue) {
t.inheritableThreadLocals = new ThreadLocalMap(this, firstValue);
}
// 二、重写的代码何时执行
// Thread的构造方法
public Thread(Runnable target) {
init(null, target, "Thread-" + nextThreadNum(), 0);
}
private void init(ThreadGroup g, Runnable target, String name,
long stackSize, AccessControlContext acc,
boolean inheritThreadLocals) {
...
// 当在父线程中创建子线程时，这个会获取父线程的引用（也就是上面的main线程）
Thread parent = currentThread();
...
// 当父线程的inheritableThreadLocals变量不为空时（当在父线程中调用set和get方法操作那个的是inheritableThreadLocals，所以就不会为空）
if (inheritThreadLocals && parent.inheritableThreadLocals != null)
// 看下面方法
this.inheritableThreadLocals =
ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);
/* Stash the specified stack size in case the VM cares */
this.stackSize = stackSize;
/* Set thread ID */
tid = nextThreadID();
}
// 使用父线程的inheritableThreadLocals变量作为参数传给ThreadLocalMap的构造方法
static ThreadLocalMap createInheritedMap(ThreadLocalMap parentMap) {
return new ThreadLocalMap(parentMap);
}ThreadLocalMap类的构造方法
// 这个构造方法是将父线程中inheritableThreadLocals中的值复制到子线程的ThreadLocalMap对象里
private ThreadLocalMap(ThreadLocalMap parentMap) {
Entry[] parentTable = parentMap.table;
int len = parentTable.length;
setThreshold(len);
table = new Entry[len];
for (int j = 0; j < len; j++) {
Entry e = parentTable[j];
if (e != null) {
@SuppressWarnings("unchecked")
ThreadLocal<Object> key = (ThreadLocal<Object>) e.get();
if (key != null) {
// 这里调用了InheritableThreadLocal里面重写的childValue方法
Object value = key.childValue(e.value);
Entry c = new Entry(key, value);
int h = key.threadLocalHashCode & (len - 1);
while (table[h] != null)
h = nextIndex(h, len);
table[h] = c;
size++;
}
}
}
}总结
InheritableThreadLocal类中重写了geMap()、createMap()、childValue()方法，当线程通过InheritableThreadLocal类的实例set或get操作变量时，就会创建当前线程的inheritableThreadLocals变量，当父线程创建子线程时，就会把父线程中inheritableThreadLocals的值复制给子线程
参考资料
Java并发编程之美（书籍）
版权属于：未央花
本文链接：http://www.pslanys.com/archives/152.html
转载时须注明出处及本声明