在处理多下线程并发安全的方法中，最常用的方法就是使用锁，通过锁来控制多个线程对临界区的访问。但是不管什么锁，乐观锁，悲观锁，都会在并发冲突的时候对性能产生一定的影响。所以就有了ThreadLocal，一个ThreadLocal变量只能被当前自身线程访问，别的线程无法访问，就避免了线程竞争。
1. 基本使用
创建一个ThreadLocal对象，指定类型为整数；
private ThreadLocal<Integer> localInt = new ThreadLocal<>();
获取和设置变量值；
public int setAndGet(){
localInt.set(8);
return localInt.get();
}
上述代码由于是在ThreadLocal对象内部设置值，意味着不可能通过其他线程为他初始化，为了弥补这一点，ThreadLocal提供了一个withInitial（）方法初始化所有线程的ThreadLocal的值。
private ThreadLocal<Integer> localInt = ThreadLocal.withInitial(() -> 6);
2. 实现原理
public T get() {
//获得当前线程
Thread t = Thread.currentThread();
//每个线程 都有一个自己的ThreadLocalMap，
//ThreadLocalMap里就保存着所有的ThreadLocal变量
ThreadLocalMap map = getMap(t);
if (map != null) {
//ThreadLocalMap的key就是当前ThreadLocal对象实例，
//多个ThreadLocal变量都是放在这个map中的
ThreadLocalMap.Entry e = map.getEntry(this);
if (e != null) {
@SuppressWarnings("unchecked")
//从map里取出来的值就是我们需要的这个ThreadLocal变量
T result = (T)e.value;
return result;
}
}
// 如果map没有初始化，那么在这里初始化一下
return setInitialValue();
}
可以看到ThreadLocal变量是保存在每个线程的map中的；
ThreadLocalMap是一个比较特殊的Map，它的每个Entry的key都是一个弱引用：
static class Entry extends WeakReference<ThreadLocal<?>> {
/** The value associated with this ThreadLocal. */
Object value;
//key就是一个弱引用
Entry(ThreadLocal<?> k, Object v) {
super(k);
value = v;
}
}
这样设计的好处是，如果这个对像不再被其他对象使用时，可以自动回收这个ThreadLocal对象，避免造成内存泄漏（需要注意的是Entry中的value依然是强引用）。
3. 内存泄漏问题
由上面可知，value依然是一个强引用，无法做到自动回收，原因是value的引用链条导致的：
只有当Thread被回收的时候，value才会有被回收的机会，否则，只要线程不退出，value总会有一个强引用。对于线程池来说，大部分线程一直存在系统的整个生命周期内，这样就会导致内存泄漏。处理的方法是在ThreadLocalMap进行set、get、remove的时候，都进行清理。
private Entry getEntry(ThreadLocal<?> key) {
int i = key.threadLocalHashCode & (table.length - 1);
Entry e = table[i];
if (e != null && e.get() == key)
//如果找到key，直接返回
return e;
else
//如果找不到，就会尝试清理，如果你总是访问存在的key，那么这个清理永远不会进来
return getEntryAfterMiss(key, i, e);
}
private Entry getEntryAfterMiss(ThreadLocal<?> key, int i, Entry e) {
Entry[] tab = table;
int len = tab.length;
while (e != null) {
// 整个e是entry ，也就是一个弱引用
ThreadLocal<?> k = e.get();
//如果找到了，就返回
if (k == key)
return e;
if (k == null)
//如果key为null，说明弱引用已经被回收了
//那么就要在这里回收里面的value了
expungeStaleEntry(i);
else
//如果key不是要找的那个，那说明有hash冲突，这里是处理冲突，找下一个entry
i = nextIndex(i, len);
e = tab[i];
}
return null;
}
真正用来回收value的是expungeStaleEntry()方法，在remove()和set()方法中，都会直接或者间接调用到这个方法进行value的清理：
从这里可以看到，ThreadLocal为了避免内存泄露，也算是花了一番大心思。不仅使用了弱引用维护key，还会在每个操作上检查key是否被回收，进而再回收value。
但是从中也可以看到，ThreadLocal并不能100%保证不发生内存泄漏。
比如，get()方法总是访问固定几个一直存在的ThreadLocal，那么清理动作就不会执行，如果你没有机会调用set()和remove()，那么这个内存泄漏依然会发生。
因此，一个良好的习惯依然是：当你不需要这个ThreadLocal变量时，主动调用remove()，这样对整个系统是有好处的。
4. ThreadLocalMap中的Hash冲突处理
HashMap使用链地址法解决hash冲突，而ThreadLocalMap使用的是简单的线性探测法，如果发生元素冲突，那么就使用下一槽位存放。
ThreadLocalMap.jpg
5. 可以被继承的ThreadLocal——InheritableThreadLocal
在实际开发过程中，我们可能会遇到这么一种场景。主线程开了一个子线程，但是我们希望在子线程中可以访问主线程中的ThreadLocal对象，也就是说有些数据需要进行父子线程间的传递。比如像这样：
public static void main(String[] args) {
ThreadLocal threadLocal = new ThreadLocal();
IntStream.range(0,10).forEach(i -> {
//每个线程的序列号，希望在子线程中能够拿到
threadLocal.set(i);
//这里来了一个子线程，我们希望可以访问上面的threadLocal
new Thread(() -> {
System.out.println(Thread.currentThread().getName() + ":" + threadLocal.get());
}).start();
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
});
}
结果：
Thread-0:null
Thread-1:null
Thread-2:null
Thread-3:null
因为在子线程中，是没有threadLocal的。如果我们希望子线可以看到父线程的ThreadLocal，那么就可以使用InheritableThreadLocal。顾名思义，这就是一个支持线程间父子继承的ThreadLocal，将上述代码中的threadLocal使用InheritableThreadLocal：
执行后结果为：
Thread-0:0
Thread-1:1
Thread-2:2
Thread-3:3
Thread-4:4
可以看到，每个线程都可以访问到从父进程传递过来的一个数据。虽然InheritableThreadLocal看起来挺方便的，但是依然要注意以下几点：
- 变量的传递是发生在线程创建的时候，如果不是新建线程，而是用了线程池里的线程，就不灵了
- 变量的赋值就是从主线程的map复制到子线程，它们的value是同一个对象，如果这个对象本身不是线程安全的，那么就会有线程安全问题
网友评论