threadLocal源码解析

简单使用

public class ThreadLocalTest {
    static ThreadLocal<String> threadLocal = new ThreadLocal(){
        //初始化方法，线程首次调用的时候如果没有初始化方法，返回null
        @Override
        protected Object initialValue() {
            return "initialValue";
        }
    };
    public static void main(String[] args) throws InterruptedException {
        new Thread(()->{
            //获取该线程的私有的变量  initialValue
            String s = threadLocal.get();
            System.out.println("t1: " + s);
            //设置值，再次同一个线程获取就是设置的值
            threadLocal.set("huge");
            System.out.println("t2" + threadLocal.get());
        },"t1").start();
        Thread.sleep(1000);
        new Thread(()->{
            //另一个线程获取，虽然有其他线程设置过了，但是这是线程私有的
            String s = threadLocal.get();
            System.out.println("t2: " + s);
        },"t2").start();
    }
}

源码解析

get方法

get方法首次调用

1. 先获取当前线程
2. 根据当前线程获取，当前线程中有一个ThreadLocal.ThreadLocalMap变量

class Thread{
	ThreadLocal.ThreadLocalMap threadLocals = null;
}

3. 获取到变量为null，就调用threadlocal的初始化方法，并创建对应的map，然后把初始化的值返回

public T get() {
		// 获取当前线程
        Thread t = Thread.currentThread();
        //获取map
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        //为空就调用初始化方法，并将值返回
        return setInitialValue();
    }

初始化中创建map的方法

1. 为该线程的变量创建ThreadLocalMap 并赋值
2. new ThreadLocalMap(this, firstValue) 这个方法是在ThreadLocal中的，所以this就是用户定义的threadLoacl对象
3. ThreadLocalMap的创建，里面用创建了一个Entry对象，有一个key就是this，和一个value对象
4. 这个Entry 对象又是内部对象，他继承了弱引用，所以this对象又被弱引用指着

void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

	ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
            table = new Entry[INITIAL_CAPACITY];
            int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
            table[i] = new Entry(firstKey, firstValue);
            size = 1;
            setThreshold(INITIAL_CAPACITY);
        }

	static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;
			
            Entry(ThreadLocal<?> k, Object v) {
            //k 对象被一个弱引用指着
                super(k);
                value = v;
            }
        }

最后的内存布局

为什么要用软引用

如果将threadlocal == null，那么第一个强引用指向断开，但是Entry里面还有一个引用指向，如果这个也是强引用的话，线程不死，ThreadLocal那块内存永远也得不到释放，所以使用软引用

内存泄露问题

如果将threadlocal == null，那么经过垃圾回收之后key也为null，那么就没有办法获取到value对象，所以内容就泄露了，所以要使用remove方法