ThreadLoacl分析

一、使用场景

如果一段代码中的数据必须与另一段代码共享，且这些共享数据的可见范围在同一个线程之内，就可以使用 java.lang.ThreadLocal来解决。

二、实现原理及其源码

每个Thread里面其实都有一个 ThreadLocal.ThreadLocalMap 里面存储着ThreadLocal在当前Thread运行时存储的value。ThreadLocal的get()、set()方法都基于此实现。

1.set()

public void set(T value) {
	// 获取当前线程
    Thread t = Thread.currentThread();
    // 获取当前线程的 threadLocals
    ThreadLocalMap map = getMap(t);
    // 插入（key=当前ThreadLocal, value=当前线程存储的值）
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}

2.get()

public T get() {
	// 获取当前线程
    Thread t = Thread.currentThread();
    // 获取当前线程中的 threadLocals
    ThreadLocalMap map = getMap(t);
    /**
    * ThreadLocalMap getMap(Thread t) {
    *     return t.threadLocals;
    * }
    /*
    if (map != null) {
    	// 通过当前 key(this/ThreadLoacl) 获取对应的Entry
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    return setInitialValue();
}

三、主要事项

ThreadLocal 从理论上讲并不是用来解决多线程并发问题的，因为根本不存在多线程竞争。

在一些场景 (尤其是使用线程池) 下，由于ThreadLocal.ThreadLocalMap 的底层数据结构导致 ThreadLocal 有内存泄漏的情况，应该尽可能在每次使用 ThreadLocal 后手动调用 remove()，以避免出现 ThreadLocal 经典的内存泄漏甚至是造成自身业务混乱的风险。

remove()方法

public void remove() {
    ThreadLocalMap m = getMap(Thread.currentThread());
    if (m != null)
        m.remove(this);
}