【并发编程】ThreadLocal 初识

ThreadLocal

本地线程，意思就是当前ThreadLocal中填充的变量属于当前线程，对于其他线程是隔离的。

• 在进行对象跨层传递时使用，可以避免多次传递
• 不同线程间数据隔离
• 管理数据库链接

private static ThreadLocal<T> local = new ThreadLocal<>();
// ThreadLocal存在一个内部类
static class ThreadLocalMap {
    Entry(ThreadLocal<?> k, Object v)
}
// Thread中存在一个ThreadLocalMap的引用
ThreadLocal.ThreadLocalMap threadLocals = null;
// 主要的方法
set()、get()、remove()、initialValue()

SET 方法

默认情况下Thread中的ThreadLocal.ThreadLocalMap引用是空的，当调用ThreadLocal 类的 set或get方法时才创建它们

public void set(T value) {
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}

ThreadLocalMap getMap(Thread t) {
    return t.threadLocals;
}

// 调用set方法会初始化
void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

首先获取当前线程，然后调用getMap获取ThreadLocalMap对象，那么这个对象是哪里来的呢？t.threadLocals是从Thread类（维护了一个 ThreadLocal.ThreadLocalMap 引用）来的，然后把ThreadLocal作为key，value就是我们设置的对象。也就是ThreadLocal设置的是当前线程自己的map对象，所以对其他线程显然是屏蔽的。

public static void main(String[] args) {
    ExecutorService executorService = Executors.newFixedThreadPool(10);
    ThreadLocal<String> threadLocal = new ThreadLocal<>();
    ThreadLocal<String> threadLocal2 = new ThreadLocal<>();
    for (int i = 0; i < 10; i++) {
        executorService.execute(() -> {
            threadLocal.set(Thread.currentThread().getName());
            threadLocal2.set(Thread.currentThread().getName());
        });
    }

    for (int i = 0; i < 10; i++) {
        executorService.execute(() -> {
            System.out.println(Thread.currentThread().getName() + " : " + threadLocal.get());
            System.out.println(Thread.currentThread().getName() + " : " + threadLocal2.get());
        });
    }
    executorService.shutdownNow();
}

ThreadLocal会存在内存泄漏的问题

ThreadLocalMap 中使用的 key 为 ThreadLocal 的弱引用，而 value 是强引用。所以在key没有被外部强引用的情况下，GC会将key清除，就会出现key为null但是强引用了value，导致value永远无法被GC回收，这时候就可能产生内存泄漏。

所以ThreadLocal在调用set()、get()、remove() 方法时会清理掉key为null的记录。

在使用完ThreadLocal后最好手动remove()进行清除