天天看點

ThreadLocal應用及源碼分析

基本使用

ThreadLocal 的作用是:提供線程内的局部變量,不同的線程之間不會互相幹擾,這種變量線上程的生命周期内起作用,減少同一個線程内多個函數或元件之間一些公共變量傳遞的複雜度,降低耦合性。

方法聲明 描述
ThreadLocal() 建立ThreadLocal對象
public void set( T value) 設定目前線程綁定的局部變量
public T get() 擷取目前線程綁定的局部變量
public void remove() 移除目前線程綁定的局部變量

簡單使用:

public class MyDemo {
    private String content;

    private String getContent() {
        return content;
    }

    private void setContent(String content) {
        this.content = content;
    }

    public static void main(String[] args) {
        MyDemo demo = new MyDemo();
        for (int i = 0; i < 5; i++) {
            Thread thread = new Thread(new Runnable() {
                @Override
                public void run() {
                    demo.setContent(Thread.currentThread().getName() + "的資料");
                    System.out.println("-----------------------");
               System.out.println(Thread.currentThread().getName() + "--->" + demo.getContent());
                }
            });
            thread.setName("線程" + i);
            thread.start();
        }
    }
}      
ThreadLocal應用及源碼分析
public class MyDemo {

    private static ThreadLocal<String> tl = new ThreadLocal<>();

    private String content;

    private String getContent() {
        return tl.get();
    }

    private void setContent(String content) {
         tl.set(content);
    }

    public static void main(String[] args) {
        MyDemo demo = new MyDemo();
        for (int i = 0; i < 5; i++) {
            Thread thread = new Thread(new Runnable() {
                @Override
                public void run() {
                    demo.setContent(Thread.currentThread().getName() + "的資料");
                    System.out.println("-----------------------");
                    System.out.println(Thread.currentThread().getName() + "--->" + demo.getContent());
                }
            });
            thread.setName("線程" + i);
            thread.start();
        }
    }
}      
ThreadLocal應用及源碼分析

這樣可以很好的解決多線程之間資料隔離的問題,用synchronized加鎖也可以實作,但synchronized側重的是多個線程之間通路資源的同步性,而ThreadLocal側重的是每個線程之間的資料隔離。

synchronized ThreadLocal
原理 同步機制采用'以時間換空間'的方式, 隻提供了一份變量,讓不同的線程排隊通路 ThreadLocal采用'以空間換時間'的方式, 為每一個線程都提供了一份變量的副本,進而實作同時通路而相不幹擾
側重點 多個線程之間通路資源的同步性 多線程中讓每個線程之間的資料互相隔離

應用場景

涉及到資料傳遞和線程隔離的場景,可以考慮用ThreadLocal來解決:轉賬案例,涉及兩個DML操作:一個轉出,一個轉入。這些操作是需要具備原子性的。是以這裡就需要操作事務,來保證轉出和轉入操作具備原子性。開啟事務的注意兩點:

  • 為了保證所有的操作在一個事務中, 使用的連接配接必須是同一個: service層開啟事務的connection需要跟dao層通路資料庫的connection保持一緻。
  • 線程并發情況下, 每個線程隻能操作各自的 connection。

用ThreadLocal的解決方案:在擷取Connection連接配接的JdbcUtils工具類加入ThreadLocal,代碼如下:

public class JdbcUtils {
    //ThreadLocal對象 : 将connection綁定在目前線程中
    private static final ThreadLocal<Connection> tl = new ThreadLocal();

    // c3p0 資料庫連接配接池對象屬性
    private static final ComboPooledDataSource ds = new ComboPooledDataSource();

    // 擷取連接配接
    public static Connection getConnection() throws SQLException {
        //取出目前線程綁定的connection對象
        Connection conn = tl.get();
        if (conn == null) {
            //如果沒有,則從連接配接池中取出
            conn = ds.getConnection();
            //再将connection對象綁定到目前線程中
            tl.set(conn);
        }
        return conn;
    }

    //釋放資源
    public static void release(AutoCloseable... ios) {
        for (AutoCloseable io : ios) {
            if (io != null) {
                try {
                    io.close();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
    }

    public static void commitAndClose() {
        try {
            Connection conn = getConnection();
            //送出事務
            conn.commit();
            //解除綁定
            tl.remove();
            //釋放連接配接
            conn.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    public static void rollbackAndClose() {
        try {
            Connection conn = getConnection();
            //復原事務
            conn.rollback();
            //解除綁定
            tl.remove();
            //釋放連接配接
            conn.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}      

可以看出使用ThreadLocal的好處:

  1. 傳遞資料 :儲存每個線程綁定的資料,在需要的地方可以直接擷取, 避免參數直接傳遞帶來的代碼耦合問題
  2. 線程隔離 :各線程之間的資料互相隔離卻又具備并發性,避免同步方式帶來的性能損失

ThreadLocal的内部結構

jdk8以前:

ThreadLocal應用及源碼分析

jdk8之前使用ThreadLocal來維護一個ThreadLocalMap,以線程作為key

jdk8以後:

ThreadLocal應用及源碼分析

jdk8之後使用Thread來維護一個ThreadLocalMap,以ThreadLocal作為key

這樣涉及的好處:

(1) 每個Map存儲的Entry數量就會變少,因為jdk8之前的存儲數量由Thread的數量決定,現在是由ThreadLocal的數量決定。

(2) 當Thread銷毀之後,對應的ThreadLocalMap也會随之銷毀,能減少記憶體的使用。

ThreadLocal核心方法的源碼

protected T initialValue() 傳回目前線程局部變量的初始值

get()

/**
     * 傳回目前線程中儲存ThreadLocal的值
     * 如果目前線程沒有此ThreadLocal變量,
     * 則它會通過調用{@link #initialValue} 方法進行初始化值
     *
     * @return 傳回目前線程對應此ThreadLocal的值
     */
public T get() {
    // 擷取目前線程對象
    Thread t = Thread.currentThread();
    // 擷取此線程對象中維護的ThreadLocalMap對象
    ThreadLocalMap map = getMap(t);
    // 如果此map存在
    if (map != null) {
        // 以目前的ThreadLocal 為 key,調用getEntry擷取對應的存儲實體e
        ThreadLocalMap.Entry e = map.getEntry(this);
        // 找到對應的存儲實體 e 
        if (e != null) {
            @SuppressWarnings("unchecked")
            // 擷取存儲實體 e 對應的 value值
            // 即為我們想要的目前線程對應此ThreadLocal的值
            T result = (T)e.value;
            return result;
        }
    }
    // 如果map不存在,則證明此線程沒有維護的ThreadLocalMap對象
    // 調用setInitialValue進行初始化
    return setInitialValue();
}

/**
     * set的變樣實作,用于初始化值initialValue,
     * 用于代替防止使用者重寫set()方法
     *
     * @return the initial value 初始化後的值
     */
private T setInitialValue() {
    // 調用initialValue擷取初始化的值
    T value = initialValue();
    // 擷取目前線程對象
    Thread t = Thread.currentThread();
    // 擷取此線程對象中維護的ThreadLocalMap對象
    ThreadLocalMap map = getMap(t);
    // 如果此map存在
    if (map != null)
        // 存在則調用map.set設定此實體entry
        map.set(this, value);
    else
        // 1)目前線程Thread 不存在ThreadLocalMap對象
        // 2)則調用createMap進行ThreadLocalMap對象的初始化
        // 3)并将此實體entry作為第一個值存放至ThreadLocalMap中
        createMap(t, value);
    // 傳回設定的值value
    return value;
}

/**
     * 擷取目前線程Thread對應維護的ThreadLocalMap 
     * 
     * @param  t the current thread 目前線程
     * @return the map 對應維護的ThreadLocalMap 
     */
ThreadLocalMap getMap(Thread t) {
    return t.threadLocals;
}
/**
     *建立目前線程Thread對應維護的ThreadLocalMap 
     *
     * @param t 目前線程
     * @param firstValue 存放到map中第一個entry的值
     */
void createMap(Thread t, T firstValue) {
    //這裡的this是調用此方法的threadLocal
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}      

首先調用Thread.currentThread()方法擷取目前線程對象,然後根據目前線程擷取維護的ThreadLocalMap對象;如果擷取的Map不為空,則在Map中以ThreadLocal的引用作為key,調用getEntry擷取對應的存儲實體,如果Entry不為空,擷取對應的 value值。如果Map為空或者Entry為空,則調用setInitialValue()方法。setInitialValue()方法裡,調用initialValue()方法擷取初始化值value,然後判斷目前線程是否有ThreadLocalMap,map存在,調用set設定Entry;map不存在則調用createMap()進行ThreadLocalMap對象的初始化,并将此entry作為第一個值存放至ThreadLocalMap中。

set()

/**
     * 設定目前線程對應的ThreadLocal的值
     *
     * @param value 将要儲存在目前線程對應的ThreadLocal的值
     */
public void set(T value) {
    // 擷取目前線程對象
    Thread t = Thread.currentThread();
    // 擷取此線程對象中維護的ThreadLocalMap對象
    ThreadLocalMap map = getMap(t);
    // 如果此map存在
    if (map != null)
        // 存在則調用map.set設定此實體entry
        map.set(this, value);
    else
        // 1)目前線程Thread 不存在ThreadLocalMap對象
        // 2)則調用createMap進行ThreadLocalMap對象的初始化
        // 3)并将此實體entry作為第一個值存放至ThreadLocalMap中
        createMap(t, value);
}      

A. 首先擷取目前線程,并根據目前線程擷取一個ThreadLocalMap

B. 如果擷取的Map不為空,則将參數設定到Map中(目前ThreadLocal的引用作為key)

C. 如果Map為空,則調用createMap給該線程建立 Map,并設定初始值

remove()

/**
     * 删除目前線程中儲存的ThreadLocal對應的實體entry
     */
public void remove() {
    // 擷取目前線程對象中維護的ThreadLocalMap對象
    ThreadLocalMap m = getMap(Thread.currentThread());
    // 如果此map存在
    if (m != null)
        // 存在則調用map.remove
        // 以目前ThreadLocal為key删除對應的實體entry
        m.remove(this);
}      

B. 如果擷取的Map不為空,則移除目前ThreadLocal對象對應的entry

initialValue()

protected T initialValue() {
    return null;
}      

(1) 這個方法是一個延遲調用方法,在set方法還未調用而先調用了get方法時才執行,并且僅執行1次。

(2)這個方法直接傳回一個null。

(3)如果想要一個除null之外的初始值,可以重寫此方法。(備注:該方法是一個protected的方法,顯然是為了讓子類覆寫而設計的)

ThreadLocalMap

ThreadLocalMap是ThreadLocal的内部類,沒有實作Map接口,用獨立的方式實作了Map的功能,其内部的Entry也是獨立實作。

ThreadLocal應用及源碼分析

1、成員變量

/**
     * 初始容量 —— 必須是2的整次幂
     */
    private static final int INITIAL_CAPACITY = 16;

    /**
     * 存放資料的table
     * 同樣,數組長度必須是2的幂。
     */
    private Entry[] table;

    /**
     * 數組裡面entrys的個數,可以用于判斷table目前使用量是否超過負載因子。
     */
    private int size = 0;

    /**
     * 進行擴容的門檻值,表使用量大于它的時候進行擴容。
     */
    private int threshold; // Default to 0
    
    /**
     * 門檻值設定為長度的2/3
     */
    private void setThreshold(int len) {
        threshold = len * 2 / 3;
    }      

2、Entry

static class Entry extends WeakReference<ThreadLocal> {
    /** The value associated with this ThreadLocal. */
    Object value;

    Entry(ThreadLocal k, Object v) {
        super(k);
        value = v;
    }
}      

在ThreadLocalMap中,也是用Entry來儲存K-V結構資料的。但是Entry中key隻能是ThreadLocal對象,這點被Entry的構造方法已經限定死了;

另外,Entry繼承WeakReference,使用弱引用,可以将ThreadLocal對象的生命周期和線程生命周期解綁,持有對ThreadLocal的弱引用,可以使得ThreadLocal在沒有其他強引用的時候被回收掉,這樣可以避免因為線程得不到銷毀導緻ThreadLocal對象無法被回收

3、hash沖突的解決

ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
    //初始化table
    table = new ThreadLocal.ThreadLocalMap.Entry[INITIAL_CAPACITY]; //16
    //計算索引
    int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
    //設定值
    table[i] = new ThreadLocal.ThreadLocalMap.Entry(firstKey, firstValue);
    size = 1;
    //設定門檻值
    setThreshold(INITIAL_CAPACITY);
}      

& (INITIAL_CAPACITY - 1),這是取模的一種方式,對于2的幂取模,用此代替%(2^n),這也就是為啥容量必須為2的幂

firstKey.threadLocalHashCode:

private final int threadLocalHashCode = nextHashCode();
    
private static int nextHashCode() {
    return nextHashCode.getAndAdd(HASH_INCREMENT);
}
private static AtomicInteger nextHashCode =  new AtomicInteger();

private static final int HASH_INCREMENT = 0x61c88647;      

這裡定義了一個AtomicInteger類型,每次擷取目前值并加上HASH_INCREMENT,HASH_INCREMENT = 0x61c88647,這個值是32位整型上限2^32-1乘以黃金分割比例0.618....的值2654435769,用有符号整型表示就是-1640531527,去掉符号後16進制表示為0x61c88647,目的就是為了讓哈希碼能均勻的分布在2的n次方的數組Entry[] table中。

      • 線性探測法:
 /**
     * 擷取環形數組的下一個索引
     */
    private static int nextIndex(int i, int len) {
        return ((i + 1 < len) ? i + 1 : 0);
    }

    /**
     * 擷取環形數組的上一個索引
     */
    private static int prevIndex(int i, int len) {
        return ((i - 1 >= 0) ? i - 1 : len - 1);
    }      
private void set(ThreadLocal<?> key, Object value) {
    ThreadLocal.ThreadLocalMap.Entry[] tab = table;
    int len = tab.length;
    //計算索引
    int i = key.threadLocalHashCode & (len-1);
    /**
         * 根據擷取到的索引進行循環,如果目前索引上的table[i]不為空,在沒有return的情況下,
         * 就使用nextIndex()擷取下一個(線性探測法)。
         */
    for (ThreadLocal.ThreadLocalMap.Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        ThreadLocal<?> k = e.get();
        //table[i]上key不為空,并且和目前key相同,更新value
        if (k == key) {
            e.value = value;
            return;
        }
        /**
             * table[i]上的key為空,說明被回收了
             * 這個時候table[i]可以重新使用,用新的key-value将其替換,并删除其他無效的entry
             */
        if (k == null) {
            replaceStaleEntry(key, value, i);
            return;
        }
    }      
上一篇: 做漢堡
下一篇: 作業三