ThreadLocal原理

发布时间 2023-12-13 16:22:33作者: 爱喝茶的安迪

ThreadLocal主要起到线程隔离作用,使得每个线程拥有自己独立的一份数据,经过threadLocal处理的数据是线程独享的,不与其它线程分享或者干扰,因此能起到线程之间数据隔离的作用。

ThreadLocal的几个核心方法:

方法声明描述
public void set(T value) 设置当前线程绑定的局部变量
public T get() 获取当前线程绑定的局部变量
Public void remove() 移除当前线程绑定的局部变量

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

该方法首先获取当前线程,并获取当前线程下维护的ThreadLocalMap元素并将数据存入到ThreadLocalMap,其中key为ThreadLocal对象本身,value就是我们想要存入的数据。

(如果没有ThreadLocalMap,就生成一个ThreadLocalMap对象,并将其与当前线程绑定,这样就是说每个线程维护了一份属于自己的ThreadLocalMap,使用的时候将数据保存在ThreadLocalMap中,这就是为什么ThreadLocal能够做到线程之间数据隔离的原因。本质就是因为每个线程通过ThreadLocal维护了一份属于自己的ThreadLocalMap)

2: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();
}

 

获取当前线程绑定的ThreadLocalMap元素,并获取其中的数据。(key为ThreadLocal对象本身)

3:remove()方法

 

public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null) {
m.remove(this);
}
}

 

获取当前线程绑定的ThreadLocalMap,将数据remove掉。(key为ThreadLocal对象本身)

本质:ThreadLocal是一个工具类,Thread类下维护了一个threadlocals变量,ThreadLocal将ThreadLocalMap赋值给Thread类下维护了一个threadlocals,从而实现了线程与ThreadLocalMap的绑定。每个线程独自拥有一个属于自己的ThreadLocalMap,进而实现了线程之间的数据隔离。

需要注意的是在数据使用完毕后,不要忘了调用remove方法,清除数据。

若依框架下ThreadLocal使用的例子: