一文搞懂 ThreadLocal 原理

public static void main(String[] args) {   IntStream.range(1, 5).forEach(i -> new Thread(() -> {       // 设置线程中本地变量的值       threadLocal.set("thread-" + i);       // 打印当前线程中本地内存中本地变量的值       System.out.println(threadLocal.get());       // 清除本地内存中的本地变量       threadLocal.remove();       // 打印本地变量       System.out.println("thread-" + i + " after remove: " + threadLocal.get());   }).start());}/*thread-1thread-4thread-4 after remove: nullthread-2thread-3thread-2 after remove: nullthread-1 after remove: nullthread-3 after remove: null*/

public void set(T value) {    // 获取当前线程    Thread t = Thread.currentThread();    // 通过当前线程获取线程中的ThreadLocal.ThreadLocalMap对象    ThreadLocalMap map = getMap(t);    if (map != null)        // map不为空，则直接赋值        map.set(this, value);    else        // map为空，则创建一个ThreadLocalMap对象        createMap(t, value);}// 根据提供的线程对象，和指定的值，创建一个ThreadLocalMap对象void createMap(Thread t, T firstValue) {    t.threadLocals = new ThreadLocalMap(this, firstValue);}// threadLocals是Thread类的一个属性ThreadLocalMap getMap(Thread t) {    return t.threadLocals;}
/*Thread 类 182行 // ThreadLocal values pertaining to this thread. This map is maintained by the ThreadLocal class. 与该线程有关的ThreadLocal值。这个映射由ThreadLocal类维护    ThreadLocal.ThreadLocalMap threadLocals = null;*/

// ThreadLocalMap中的内部类，存放key，valuestatic class Entry extends WeakReference<ThreadLocal<?>> {    // 与此ThreadLocal关联的值    Object value; // k：ThreadLocal的引用，被传递给WeakReference的构造方法    Entry(ThreadLocal<?> k, Object v) {        super(k);        value = v;    }}
public T get() {    // 获取当前线程    Thread t = Thread.currentThread();    // 通过当前线程获取线程中的ThreadLocal.ThreadLocalMap对象    ThreadLocalMap map = getMap(t);    if (map != null) {        // map不为空，通过this(当前对象，即ThreadLocal对象)获取Entry对象        ThreadLocalMap.Entry e = map.getEntry(this);        if (e != null) {            @SuppressWarnings("unchecked")            T result = (T)e.value;            // Entry不为空，则直接返回Entry中的value值            return result;        }    }    // 如果map或Entry为空，则返回初始值-null    return setInitialValue();}// 设置初始值，初始化ThreadLocalMap对象，并设置value为 nullprivate T setInitialValue() {    // 初始化值，此方法返回 null    T value = initialValue();    Thread t = Thread.currentThread();    ThreadLocalMap map = getMap(t);    if (map != null)        map.set(this, value);    else        createMap(t, value);    return value;}

public void remove() {    // 通过当前线程获取线程中的ThreadLocal.ThreadLocalMap对象    ThreadLocalMap m = getMap(Thread.currentThread());    if (m != null)        // 移除对象        m.remove(this);}// 根据key，删除对应的所有值private void remove(ThreadLocal<?> key) {    Entry[] tab = table;    int len = tab.length;    // 获取key对应的 Entry[] 下标    int i = key.threadLocalHashCode & (len-1);    for (Entry e = tab[i];         e != null;         // 获取下一个Entry对象         e = tab[i = nextIndex(i, len)]) {        if (e.get() == key) {            e.clear();            // 通过重新哈希位于staleSlot和下一个null插槽之间的任何可能冲突的条目，来清除陈旧的条目。这还会清除尾随null之前遇到的所有其他过时的条目，防止出现内存泄漏问题            expungeStaleEntry(i);            return;        }    }}