ThreadLocal的簡單使用

ThreadLocal的簡單使用

• ThreadLocal在線程內部調用set(value)方法存儲值。

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

在set方法內部，我們獲取到當前線程對象，通過ThreadLocal成員方法getMap(t)根據當前線程對象獲取到當前線程的的成員屬性變量threadLocals（變量類型是ThreadLocalMap，在createMap(t,value)中初始化）這是key-value存儲容器，我們調用這個threadLocals來實現保存和修改數據的操作。保存數據有兩種方式，一個通過ThreadLocalMap的成員方法set(this, value),通過ThreadLocal成員方法 createMap(t, value)實現。

• ThreadLocal的成員方法createMap(線程對象,value)

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

在createMap內部我們直接調用ThreadLocalMap的構造函數傳入當前ThreadLocal對象和要保存的數據值創建該對象，並把該對象的引用傳遞給當前線程的屬性變量threadLocals。

• 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);
        }

在ThreadLocalMap(threadLocal,firstValue)構造函數內部，我們初始化了一個Entry對象（Entry是ThreadLocalMap的靜態內部類，繼承了WeakReference對象），這個對象存儲了傳入的ThreadLocal對象和數據對象。我們把這個對象加入一個初始化長度爲16的Entry類型的數組表中。

• ThreadLocalMap的成員方法set(this, value)

private void set(ThreadLocal<?> key, Object value) {
            Entry[] tab = table;
            int len = tab.length;
            int i = key.threadLocalHashCode & (len-1);

            for (Entry e = tab[i];
                 e != null;
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();

                if (k == key) {
                    e.value = value;
                    return;
                }

                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }

            tab[i] = new Entry(key, value);
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
        }

在set(threadLocal, value)的方法內部，我們會先遍歷table表判斷是否存在存儲了該ThreadLocal對象的entry存在，如果存在則替換數據值，不存在則新建entry對象，加入到table表中。

• ThreadLocal在線程內部調用get()方法獲取值。

public T get() {
        Thread t = Thread.currentThread();
        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();
    }

在get方法內部，我們首先獲取到當前線程對象，再通過當前線程對象獲取到它的屬性變量threadLocals，調用它的getEntry(threadLocal)獲取到Entry對象，通過Entry對象我們可以直接獲取到值。

• 總結
  ThreadLocal爲每一個線程都單獨維護了一個ThreadLocalMap對象，其中一個線程通過ThreadLocal存儲或者修改他的數據對象時，不會影響到其他線程。