Java的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-1
thread-4
thread-4 after remove: null
thread-2
thread-3
thread-2 after remove: null
thread-1 after remove: null
thread-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.ThreadLocalMap threadLocals = null;
*/

// ThreadLocalMap中的内部类，存放key，value
static 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为 null
private 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;
        }
    }
}