ThreadLocal源码分析

我们不从把ThreadLocal源码直接复制到这里进行挨个分析，我们把使用列出来，找出突破点进行分析

public static final ThreadLocal<String> mThreadLocal = new ThreadLocal<String>() ;
    public static void main(String[] args) {
        //主线程设置为false
        mThreadLocal.set("小明") ;
        new Thread("therad--1"){
            public void run() {
                mThreadLocal.set("小红") ;
                System.out.println("thread--1:" + mThreadLocal.get());
            };
        }.start() ;
        new Thread("therad--2"){
            public void run() {
                mThreadLocal.set("小花") ;
                System.out.println("thread--2:" + mThreadLocal.get());
            };
        }.start() ;
        System.out.println("主线程:" + mThreadLocal.get());
    }

执行结果大家肯定知道，各个线程有自己各个线程的打印。但是ThreadLocal这个类型的变量却是全局的，如果ThreadLocal与普通变量一样，那么久违背了多线程访问共享资源原理,所以从set方法作为切入点

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;//ThreadLocal.ThreadLocalMap threadLocals = null;这个变量是Thread内所有
}

也就是说对应Thread内部有一个ThreadLocal.ThreadLocalMap，设置值是拿到当前线程，然后设置对应的值入map中

private void set(ThreadLocal<?> key, Object value) {
    Entry[] tab = table;
    int len = tab.length;
    int i = key.threadLocalHashCode & (len-1);//得到对应下标，每一个线程都有自己的key，这个i是一定的
    //遍历下表然后挨个进行查找对比
    for (Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        ThreadLocal<?> k = e.get();
        //覆盖式操作
        if (k == key) {
            e.value = value;
            return;
        }
        //当key为空也就是之前并没有这个元素
        if (k == null) {
            replaceStaleEntry(key, value, i);
            return;
        }
    }
    //生成这个元素&添加到列表中
    tab[i] = new Entry(key, value);
    int sz = ++size;
    //如果超过对应size则进行rehash操作
    if (!cleanSomeSlots(i, sz) && sz >= threshold)
        rehash();
}

这里需要注意初始容量是16：
private static final int INITIAL_CAPACITY = 16;具体操作是set的时候如果没有创建ThreadLocalMap则进行创建构造中设置值

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);//16
}

这个时候我们就将值放入了对应集合

public T get() {
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);//得到对应线程内的对应map
    if (map != null) {
        ThreadLocalMap.Entry e = map.getEntry(this);//得到Entry
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    return setInitialValue();//当没有找到值也就是说没有设置值，然后要找值就进行初始化
}

ThreadLocal源码分析

推荐阅读更多精彩内容