TheadLocal源码解析

TheadLocal详解

概述

这是JDK提供的api类,该类提供了线程局部变量,它独立于变量的初始化副本,为每个通过get/set方法访问某个变量提供基于自己线程的独立变量。

源码解析

整体上ThreadLocal的使用就是通过get/set方法来使用：

get:

1. 获取当前线程
2. 获取当前线程ThreadLocalMap
3. 如果不为空,获取当前map的值
4. 如果为空,初始化新的ThreadLocalMap

set:

1. 获取当前线程
2. 获取当前的ThreadLocalMap
3. 为空，初始化新的map并赋值
4. 不为空，重新赋值,key为当前ThreadLocal实例

public T get() {
    //获取当前线程
    Thread t = Thread.currentThread();
    // 获取当前线程的ThreadLocalMap
    ThreadLocalMap map = getMap(t);
    if (map != null) {
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null)
            return (T)e.value;
    }
    //初始化
    return setInitialValue();
}

private T setInitialValue() {
    //初始化value=null
    T value = initialValue();
    //获取当前线程
    Thread t = Thread.currentThread();
    // 获取当前线程的ThreadLocalMap
    ThreadLocalMap map = getMap(t);
    if (map != null)
        //指向空值
       map.set(this, value);
    else
        //初始化一个新的ThreadLocalMap
       createMap(t, value);
    return value;
}
void createMap(Thread t, T firstValue) {
     t.threadLocals = new ThreadLocalMap(this, firstValue);
}

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

内存泄漏

很多文章都说ThreadLocal会导致内存泄漏,却没有说为什么会内存泄漏，导致内存泄漏的原因是什么,我们结合源码，来分析一下这个问题,首先先看一下当一个线程使用ThreadLocal的时候堆栈的分配情况:

通过图中可以看出,在一个线程创建的时候,对于线程本身的ThreadLocalMap是强引用的，而对于ThreadLocalMap的key是弱引用的,结合上文,可以看出这个key实际上就是ThreadLocal,那么系统gc的时候，ThreadLocal必然会被回收，这样ThreadLocalMap里就会出现key为null的Entry，如果当前线程一直没有结束或者挂起的时候，就会出现这样一条强引用链:Thread Reference –>Thread –>ThreadLocalMap –>Entry–>value,key为null的value就一直没办法回收。