JAVA并发编程系列(12)ThreadLocal就是这么简单|建议收藏

//T,支持创建任意类型的线程本地变量
ThreadLocal<T> threadLocal = new ThreadLocal<>();

//1、设置线程本地变量值
threadLocal.set(value);
//2、读取变量值
T value = threadLocal.get();
//3、线程执行结束前，调用该方法，清空变量，避免内存泄漏
threadLocal.remove();

package lading.java.mutithread;
/**
 * 模拟主线程持有的线程本地变量ThreadLocal，其他线程无法读取。
 * 此外，每个线程无法修改其他线程的本地变量，但是可以设置别的线程定义的本地变量
 */
public class Demo014ThreadLocal {
    //1、主线程定义的一个线程本地变量
    public static ThreadLocal<String> mainThreadLocalVar = new ThreadLocal<>();
    public static void main(String[] args) {
        //2、主线程设置本地线程变量值为：I am main Thread
        mainThreadLocalVar.set("I am main Thread");
        System.out.println(Thread.currentThread().getName() + "设置了本地变量：" + mainThreadLocalVar.get());
        //3、新建一个线程【老六】
        new Thread(() -> {
            //4、去读取主线程定义初始化的mainThreadLocalVar，看是否能读取
            System.out.println(Thread.currentThread().getName() + "读取主线程定义的本地变量，值为：" + mainThreadLocalVar.get());
            if (mainThreadLocalVar.get() == null) {
                System.out.println(Thread.currentThread().getName() + "确实没读到主线程定义的本地变量！");
            }
            //5、老六尝试去在主线程的本地变量新增一个变量值【I am old Six】
            mainThreadLocalVar.set("I am old Six");
            //6、看能否读取成功
            System.out.println(Thread.currentThread().getName() + "设置了自己的本地变量，值为：" + mainThreadLocalVar.get());
        }, "老六").start();
        //7、主线程休眠2s，等老六线程设置本地变量，主线程再读自己的本地变量看是否被修改
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        //8、老六线程也设置了自己的本地变量值，是新增还是覆盖主线程的变量？
        System.out.println(Thread.currentThread().getName() + "在老六设置了本地变量后，自己的本地变量值为：" + mainThreadLocalVar.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);
    }

//这个线程本地变量第一次被线程设置值
    void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }
    
    //new ThreadLocalMap(this, firstValue)的源码如下
     ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
            //按初始化大小，初始化数组
            table = new Entry[INITIAL_CAPACITY];
            //计算数据应该存放的hash索引位置
            int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
            //在Entity table数组的i位置，存这个线程本地变量
            table[i] = new Entry(firstKey, firstValue);
            //数组大小改成1
            size = 1;
            setThreshold(INITIAL_CAPACITY);
        }

//这几个都是强引用
        ThreadLocal<String> threadLocal = new ThreadLocal<>();
        int a = 44;
        Object hh = new Object();
        
        //用WeakReference泛型类定义的就是弱引用
        WeakReference<String> weakObject = new WeakReference<>("aaa");

//ThreadLocal，在set方法里，有加这个 cleanSomeSlots(i, sz)方法，就是清除那些key为空的无效变量值
 if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();