Java ThreadLocal類使用詳解

package cuit.pymjl.thradlocal;

/**
 * @author Pymjl
 * @version 1.0
 * @date 2022/7/1 10:56
 **/
public class MainTest {
    static ThreadLocal<String> threadLocal = new ThreadLocal<>();
​
    static void print(String str) {
        //打印當前線程中本地內存中本地變量的值
        System.out.println(str + " :" + threadLocal.get());
        //清除本地內存中的本地變量
        threadLocal.remove();
    }
​
    public static void main(String[] args) {
        Thread t1 = new Thread(new Runnable() {
            @Override
            public void run() {
                //設置線程1中本地變量的值
                threadLocal.set("thread1 local variable");
                //調用打印方法
                print("thread1");
                //打印本地變量
                System.out.println("after remove : " + threadLocal.get());
            }
        });
​
        Thread t2 = new Thread(new Runnable() {
            @Override
            public void run() {
                //設置線程1中本地變量的值
                threadLocal.set("thread2 local variable");
                //調用打印方法
                print("thread2");
                //打印本地變量
                System.out.println("after remove : " + threadLocal.get());
            }
        });
​
        t1.start();
        t2.start();
    }
}

    public void set(T value) {
        // 1.獲取當前線程（調用者線程）
        Thread t = Thread.currentThread();
        // 2.以當前線程作為key值，去查找對應的線程變量，找到對應的map
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            // 3.如果map不為null，則直接添加元素
            map.set(this, value);
        } else {
            // 4.否則就先創建map，再添加元素
            createMap(t, value);
        }
    }

    void createMap(Thread t, T firstValue) {
        /**
         * 這裡是創建一個ThreadLocalMap，以當前調用線程的實例對象為key，初始值為value
         * 然後放入當前線程的Therad.threadLocals屬性裡面
         */
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

    ThreadLocalMap getMap(Thread t) {
        //這裡就是直接獲取調用線程的成員屬性threadlocals
        return t.threadLocals;
    }

    public T get() {
        // 1.獲取當前線程
        Thread t = Thread.currentThread();
        // 2.獲取當前線程的threadlocals，即ThreadLocalMap
        ThreadLocalMap map = getMap(t);
        // 3.如果map不為null，則直接返回對應的值
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        // 4.否則，則進行初始化
        return setInitialValue();
    }

private T setInitialValue() {
    //初始化屬性，其實就是null
    T value = initialValue();
    //獲取當前線程
    Thread t = Thread.currentThread();
    //通過當前線程獲取ThreadLocalMap
    ThreadLocalMap map = getMap(t);
    //如果map不為null，則直接添加元素
    if (map != null) {
        map.set(this, value);
    } else {
        //否則就創建，然後將創建好的map放入當前線程的屬性threadlocals
        createMap(t, value);
    }
        //將當前ThreadLocal實例註冊進TerminatingThreadLocal類裡面
    if (this instanceof TerminatingThreadLocal) {
        TerminatingThreadLocal.register((TerminatingThreadLocal<?>) this);
    }
    return value;
}

/**
 * A thread-local variable that is notified when a thread terminates and
 * it has been initialized in the terminating thread (even if it was
 * initialized with a null value).
 * 一個線程局部變量，
 * 當一個線程終止並且它已經在終止線程中被初始化時被通知（即使它被初始化為一個空值）。
 */

     public void remove() {
         //如果當前線程的threadLocals 變量不為空， 則刪除當前線程中指定ThreadLocal 實例的本地變量。
         ThreadLocalMap m = getMap(Thread.currentThread());
         if (m != null) {
             m.remove(this);
         }
     }