ThreadLocal的使用
ThreadLocal用于多線程情況下,為每個(gè)線程分配一個(gè)只屬于該線程的全局變量,實(shí)現(xiàn)線程間的數(shù)據(jù)隔離凸主。ThreadLocal使用上主要是set和set兩個(gè)方法。
ThreadLocal<Integer> threadLocal1 = new ThreadLocal<>();
threadLocal1.set(1);
ThreadLocal<Integer> threadLocal2 = new ThreadLocal<>();
threadLocal2.set(2);
Integer num1 = threadLocal1.get();
Integer num2 = threadLocal2.get();
ThreadLocal實(shí)現(xiàn)
ThreadLocal大致實(shí)現(xiàn)原理是在每個(gè)線程上綁定一個(gè)類Map的結(jié)構(gòu)党远,Map的Key是ThreadLocal變量,value是ThreadLocal中維持的對(duì)象住涉。
ThreadLocal的set方法源碼如下:
public void set(T value) {
//獲取當(dāng)前線程
Thread t = Thread.currentThread();
//獲取與當(dāng)前線程綁定的類Map(線性探測(cè)法實(shí)現(xiàn)的hash表)
ThreadLocalMap map = getMap(t);
//如果存hash表的數(shù)組已經(jīng)存在麸锉,ThreadLocal變量為key,ThreadLocal變量維持的對(duì)象為value舆声,存入Map
if (map != null)
map.set(this, value);
//如果存hash表的數(shù)組未初始化,初始化并以ThreadLocal變量為key,ThreadLocal變量維持的對(duì)象為value存入Map
else
createMap(t, value);
}
其中媳握,getMap()方法:
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
threadLocals是線程的成員變量碱屁,類型為ThreadLocal.ThreadLocalMap,可見(jiàn)用于保存ThreadLocal-value的ThreadLocalMap是ThreadLocal的內(nèi)部類蛾找。
createMap方法如下:
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
該方法new了一個(gè)ThreadLocalMap娩脾,并存入了第一個(gè)Entry。ThreadLocalMap的構(gòu)造方法如下:
ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
//初始化默認(rèn)長(zhǎng)度的entry數(shù)組
table = new Entry[INITIAL_CAPACITY];
//計(jì)算entry應(yīng)該放到哪里
int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
table[i] = new Entry(firstKey, firstValue);
size = 1;
//設(shè)置擴(kuò)容閾值
setThreshold(INITIAL_CAPACITY)
}
ThreadLocalMap內(nèi)部有一個(gè)Entry數(shù)組打毛,在這個(gè)數(shù)組上通過(guò)線性探測(cè)法構(gòu)造了一個(gè)hash表柿赊。
這里的Entry對(duì)TreadLocal對(duì)象的引用方式是弱引用,所以當(dāng)程序中ThreadLocal變量生命周期結(jié)束后幻枉,Entry對(duì)ThreadLocal對(duì)象的引用不會(huì)影響ThreadLocal的回收碰声。
static class Entry extends WeakReference<ThreadLocal<?>> {
Object value;
Entry(ThreadLocal<?> k, Object v) {
super(k);
value = v;
}
}
回到ThreadLocalMap的set方法:
private void set(ThreadLocal<?> key, Object value) {
Entry[] tab = table;
int len = tab.length;
int i = key.threadLocalHashCode & (len-1);
//從i位置,線性探測(cè)可用的位置
for (Entry e = tab[i]; e != null; e = tab[i = nextIndex(i, len)]) {
ThreadLocal<?> k = e.get();
//如果這個(gè)位置有entry熬甫,且entry的key就是當(dāng)前的ThreadLocal胰挑,則用當(dāng)前value覆蓋entry的value
if (k == key) {
e.value = value;
return;
}
//如果entry不為空,但entry的key為空椿肩,說(shuō)明需要整理了
if (k == null) {
replaceStaleEntry(key, value, i);
return;
}
}
//非空entry瞻颂,都不能存儲(chǔ),找到第一個(gè)空的entry郑象,存入
tab[i] = new Entry(key, value);
int sz = ++size;
//滿足rehash條件時(shí)贡这,rehash
if (!cleanSomeSlots(i, sz) && sz >= threshold)
rehash()
}
rehash和整理的邏輯太復(fù)雜,改日再看吧厂榛!